Add more test cases on MaskRule (#33446)

apache · Oct 28, 2024 · ec3138d · ec3138d
1 parent 9a6e570
commit ec3138d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/features/mask/core/src/test/java/org/apache/shardingsphere/mask/rule/MaskRuleTest.java b/features/mask/core/src/test/java/org/apache/shardingsphere/mask/rule/MaskRuleTest.java
@@ -27,8 +27,11 @@
 import java.util.Collections;
 import java.util.Properties;
 
+import static org.hamcrest.CoreMatchers.is;
+import static org.hamcrest.MatcherAssert.assertThat;
 import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.mockito.Mockito.mock;
 
 class MaskRuleTest {
 
@@ -55,4 +58,11 @@ void assertFindMaskTableWhenTableNameExists() {
     void assertFindMaskTableWhenTableNameDoesNotExist() {
         assertFalse(maskRule.findMaskTable("non_existent_table").isPresent());
     }
+
+    @Test
+    void assertUpdateConfiguration() {
+        MaskRuleConfiguration toBeUpdatedRuleConfig = mock(MaskRuleConfiguration.class);
+        maskRule.updateConfiguration(toBeUpdatedRuleConfig);
+        assertThat(maskRule.getConfiguration(), is(toBeUpdatedRuleConfig));
+    }
 }