Add missing package import in PortalRegistryTest.java (#862)

* Add unregistry for subAndUnsubService() method * Update PortalRegistryTest.java * Add unregistry & unavailable code * Add import ignore * Update PortalRegistryTest.java * Update ConsulRegistryTest.java * Update PortalRegistryTest.java
networknt · Dec 11, 2020 · 513b66d · 513b66d
1 parent e1e8486
commit 513b66d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/consul/src/test/java/com/networknt/consul/ConsulRegistryTest.java b/consul/src/test/java/com/networknt/consul/ConsulRegistryTest.java
@@ -141,10 +141,10 @@ public void discoverService() throws Exception {
         urls = registry.discover(serviceUrl);
         Assert.assertTrue(urls.contains(serviceUrl));
 
-        // unavailable
+        // unavailable & unregister
         registry.doUnavailable(null);
         Thread.sleep(sleepTime);
-        Assert.assertFalse(client.isWorking(serviceid));
+        registry.doUnregister(serviceUrl);
     }
 
     private Boolean containsNotifyListener(URL serviceUrl, URL clientUrl, NotifyListener listener) {

diff --git a/portal-registry/src/test/java/com/networknt/portal/registry/PortalRegistryTest.java b/portal-registry/src/test/java/com/networknt/portal/registry/PortalRegistryTest.java
@@ -25,6 +25,7 @@
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
+import org.junit.Ignore;
 import org.junit.Test;
 
 import java.util.Iterator;
@@ -126,10 +127,10 @@ public void discoverService() throws Exception {
         }
         Assert.assertTrue(urls.contains(serviceUrl));
 
-        // unavailable
+        // unavailable & unregister
         registry.doUnavailable(null);
         Thread.sleep(sleepTime);
-        Assert.assertFalse(client.isWorking(serviceid));
+        registry.doUnregister(serviceUrl);
     }
 
     @Test