Add test cases on ShadowRuleConfigurationChecker (#33521)

* Add ShadowDataSourceMappingsFinder * Add ShadowDataSourceMappingsFinder * Add ShadowDataSourceMappingsFinder * Add ShadowDataSourceMappingsFinder * Add test cases on ShadowRuleConfigurationChecker
apache · Nov 3, 2024 · 14a2d75 · 14a2d75
1 parent 9e193e7
commit 14a2d75
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/...est/java/org/apache/shardingsphere/shadow/checker/ShadowRuleConfigurationCheckerTest.java b/...est/java/org/apache/shardingsphere/shadow/checker/ShadowRuleConfigurationCheckerTest.java
@@ -27,6 +27,7 @@
 import org.apache.shardingsphere.shadow.exception.metadata.MissingRequiredProductionDataSourceException;
 import org.apache.shardingsphere.shadow.exception.metadata.MissingRequiredShadowDataSourceException;
 import org.apache.shardingsphere.shadow.exception.metadata.NotImplementHintShadowAlgorithmException;
+import org.apache.shardingsphere.shadow.exception.metadata.ShadowDataSourceMappingNotFoundException;
 import org.apache.shardingsphere.test.fixture.jdbc.MockedDataSource;
 import org.apache.shardingsphere.test.util.PropertiesBuilder;
 import org.apache.shardingsphere.test.util.PropertiesBuilder.Property;
@@ -88,6 +89,21 @@ private ShadowRuleConfiguration createRuleConfigurationWithInvalidDefaultShadowA
         return result;
     }
 
+    @Test
+    void assertCheckWithInvalidShadowTableDataSourcesReferences() {
+        assertThrows(ShadowDataSourceMappingNotFoundException.class,
+                () -> ruleConfigChecker.check("foo_db", createRuleConfigurationWithInvalidShadowTableDataSourcesReferences(), createDataSourceMap(), Collections.emptyList()));
+    }
+
+    private ShadowRuleConfiguration createRuleConfigurationWithInvalidShadowTableDataSourcesReferences() {
+        ShadowRuleConfiguration result = new ShadowRuleConfiguration();
+        result.setShadowAlgorithms(Collections.singletonMap("foo-algo", new AlgorithmConfiguration("SQL_HINT", new Properties())));
+        result.setDefaultShadowAlgorithmName("foo-algo");
+        result.setDataSources(Collections.singleton(new ShadowDataSourceConfiguration("foo_ds", "prod_ds", "shadow_ds")));
+        result.setTables(Collections.singletonMap("foo_tbl", new ShadowTableConfiguration(Collections.singletonList("bar_ds"), new LinkedList<>(Collections.singleton("foo-algo")))));
+        return result;
+    }
+
     @Test
     void assertCheckWithInvalidShadowTableAlgorithmsReferences() {
         assertThrows(MissingRequiredAlgorithmException.class,