add assertGetKeyGeneratorClassNameWithUUID()

73880b64 · tristaZero · 2edf02b5 · 73880b64
隐藏空白更改
内联并排

Showing with 14 addition and 0 deletion

sharding-core/src/test/java/io/shardingsphere/core/yaml/sharding/YamlKeyGeneratorConfigurationTest.java ...core/yaml/sharding/YamlKeyGeneratorConfigurationTest.java +14 -0

未找到文件。
--- a/sharding-core/src/test/java/io/shardingsphere/core/yaml/sharding/YamlKeyGeneratorConfigurationTest.java
+++ b/sharding-core/src/test/java/io/shardingsphere/core/yaml/sharding/YamlKeyGeneratorConfigurationTest.java
@@ -44,4 +44,18 @@ public class YamlKeyGeneratorConfigurationTest {
        YamlKeyGeneratorConfiguration keyGeneratorConfiguration = new YamlKeyGeneratorConfiguration();
        assertThat(keyGeneratorConfiguration.getKeyGenerator().getClass().getName(), is(SnowflakeKeyGenerator.class.getName()));
    }
+    
+    @Test(expected = IllegalArgumentException.class)
+    public void assertGetKeyGeneratorClassNameWithLeaf() {
+        YamlKeyGeneratorConfiguration keyGeneratorConfiguration = new YamlKeyGeneratorConfiguration();
+        keyGeneratorConfiguration.setType("LEAF");
+        keyGeneratorConfiguration.getKeyGenerator();
+    }
+    
+    @Test(expected = IllegalArgumentException.class)
+    public void assertGetKeyGeneratorClassNameWithUUID() {
+        YamlKeyGeneratorConfiguration keyGeneratorConfiguration = new YamlKeyGeneratorConfiguration();
+        keyGeneratorConfiguration.setType("UUID");
+        keyGeneratorConfiguration.getKeyGenerator();
+    }
 }