This is an automated email from the ASF dual-hosted git repository.
zhangliang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git
The following commit(s) were added to refs/heads/master by this push:
new 78e32982a38 Add test cases on ClusterMetaDataManagerPersistService
(#32856)
78e32982a38 is described below
commit 78e32982a3894627a30469ac85d029c5204a94ae
Author: Liang Zhang <[email protected]>
AuthorDate: Fri Sep 13 23:06:53 2024 +0800
Add test cases on ClusterMetaDataManagerPersistService (#32856)
---
.../ClusterMetaDataManagerPersistServiceTest.java | 120 +++++++++++++++++++++
1 file changed, 120 insertions(+)
diff --git
a/mode/type/cluster/core/src/test/java/org/apache/shardingsphere/mode/manager/cluster/persist/ClusterMetaDataManagerPersistServiceTest.java
b/mode/type/cluster/core/src/test/java/org/apache/shardingsphere/mode/manager/cluster/persist/ClusterMetaDataManagerPersistServiceTest.java
new file mode 100644
index 00000000000..2a9fdd6931e
--- /dev/null
+++
b/mode/type/cluster/core/src/test/java/org/apache/shardingsphere/mode/manager/cluster/persist/ClusterMetaDataManagerPersistServiceTest.java
@@ -0,0 +1,120 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.shardingsphere.mode.manager.cluster.persist;
+
+import lombok.SneakyThrows;
+import
org.apache.shardingsphere.infra.metadata.database.schema.model.ShardingSphereSchema;
+import
org.apache.shardingsphere.infra.metadata.database.schema.pojo.AlterSchemaPOJO;
+import org.apache.shardingsphere.metadata.persist.MetaDataPersistService;
+import org.apache.shardingsphere.mode.metadata.MetaDataContextManager;
+import
org.apache.shardingsphere.mode.persist.service.ListenerAssistedPersistService;
+import org.apache.shardingsphere.mode.spi.PersistRepository;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.Answers;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+
+import java.lang.reflect.Field;
+import java.util.Collections;
+
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+@ExtendWith(MockitoExtension.class)
+class ClusterMetaDataManagerPersistServiceTest {
+
+ private ClusterMetaDataManagerPersistService metaDataManagerPersistService;
+
+ @Mock(answer = Answers.RETURNS_DEEP_STUBS)
+ private MetaDataPersistService metaDataPersistService;
+
+ @Mock
+ private ListenerAssistedPersistService listenerAssistedPersistService;
+
+ @Mock(answer = Answers.RETURNS_DEEP_STUBS)
+ private MetaDataContextManager metaDataContextManager;
+
+ @BeforeEach
+ void setUp() {
+ metaDataManagerPersistService = new
ClusterMetaDataManagerPersistService(mock(PersistRepository.class),
metaDataContextManager);
+ setField("metaDataPersistService", metaDataPersistService);
+ setField("listenerAssistedPersistService",
listenerAssistedPersistService);
+ }
+
+ @SneakyThrows(ReflectiveOperationException.class)
+ void setField(final String filedName, final Object fieldValue) {
+ Field field =
metaDataManagerPersistService.getClass().getDeclaredField(filedName);
+ field.setAccessible(true);
+ field.set(metaDataManagerPersistService, fieldValue);
+ field.setAccessible(false);
+ }
+
+ @Test
+ void assertCreateDatabase() {
+ metaDataManagerPersistService.createDatabase("foo_db");
+
verify(metaDataPersistService.getDatabaseMetaDataService()).addDatabase("foo_db");
+
verify(listenerAssistedPersistService).persistDatabaseNameListenerAssisted(any());
+ }
+
+ @Test
+ void assertDropDatabase() {
+
when(metaDataContextManager.getMetaDataContexts().get().getMetaData().getDatabase("foo_db").getName()).thenReturn("foo_db");
+ metaDataManagerPersistService.dropDatabase("foo_db");
+
verify(listenerAssistedPersistService).persistDatabaseNameListenerAssisted(any());
+
verify(metaDataPersistService.getDatabaseMetaDataService()).dropDatabase("foo_db");
+ }
+
+ @Test
+ void assertCreateSchema() {
+ metaDataManagerPersistService.createSchema("foo_db", "foo_schema");
+
verify(metaDataPersistService.getDatabaseMetaDataService()).addSchema("foo_db",
"foo_schema");
+ }
+
+ @Test
+ void assertAlterNotEmptySchema() {
+ assertAlterSchema(mock(ShardingSphereSchema.class));
+ verify(metaDataPersistService.getDatabaseMetaDataService(),
times(0)).addSchema("foo_db", "bar_schema");
+ }
+
+ @Test
+ void assertAlterEmptySchema() {
+ ShardingSphereSchema schema = mock(ShardingSphereSchema.class);
+ when(schema.isEmpty()).thenReturn(true);
+ assertAlterSchema(schema);
+
verify(metaDataPersistService.getDatabaseMetaDataService()).addSchema("foo_db",
"bar_schema");
+ }
+
+ private void assertAlterSchema(final ShardingSphereSchema schema) {
+
when(metaDataContextManager.getMetaDataContexts().get().getMetaData().getDatabase("foo_db").getSchema("foo_schema")).thenReturn(schema);
+ metaDataManagerPersistService.alterSchema(new
AlterSchemaPOJO("foo_db", "foo_schema", "bar_schema",
Collections.singleton("foo_ds")));
+
verify(metaDataPersistService.getDatabaseMetaDataService().getTableMetaDataPersistService()).persist("foo_db",
"bar_schema", Collections.emptyMap());
+
verify(metaDataPersistService.getDatabaseMetaDataService().getViewMetaDataPersistService()).persist("foo_db",
"bar_schema", Collections.emptyMap());
+
verify(metaDataPersistService.getDatabaseMetaDataService()).dropSchema("foo_db",
"foo_schema");
+ }
+
+ @Test
+ void assertDropSchema() {
+ metaDataManagerPersistService.dropSchema("foo_db",
Collections.singleton("foo_schema"));
+
verify(metaDataPersistService.getDatabaseMetaDataService()).dropSchema("foo_db",
"foo_schema");
+ }
+}
