unit test for SimpleDataStore (apache#1797)

rolandhe · May 15, 2018 · 322e8ff · 322e8ff
1 parent 4845b64
commit 322e8ff
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/dubbo-common/src/test/java/com/alibaba/dubbo/common/store/support/SimpleDataStoreTest.java b/dubbo-common/src/test/java/com/alibaba/dubbo/common/store/support/SimpleDataStoreTest.java
@@ -18,14 +18,18 @@
 
 import org.junit.Test;
 
+import java.util.Map;
+
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotEquals;
 import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
 
 public class SimpleDataStoreTest {
-    SimpleDataStore dataStore = new SimpleDataStore();
+    private SimpleDataStore dataStore = new SimpleDataStore();
 
     @Test
-    public void testPut_Get() throws Exception {
+    public void testPutGet() throws Exception {
         assertNull(dataStore.get("xxx", "yyy"));
 
         dataStore.put("name", "key", "1");
@@ -42,4 +46,15 @@ public void testRemove() throws Exception {
         dataStore.remove("name", "key");
         assertNull(dataStore.get("name", "key"));
     }
+
+    @Test
+    public void testGetComponent() throws Exception {
+        Map<String, Object> map = dataStore.get("component");
+        assertTrue(map != null && map.isEmpty());
+        dataStore.put("component", "key", "value");
+        map = dataStore.get("component");
+        assertTrue(map != null && map.size() == 1);
+        dataStore.remove("component", "key");
+        assertNotEquals(map, dataStore.get("component"));
+    }
 }