1 files changed, 7 insertions, 4 deletions
diff --git a/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentTest.java b/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentTest.java
index baf77ea..6ab10d6 100644
--- a/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentTest.java
+++ b/src/test/java/pl/edu/mimuw/cloudatlas/agent/AgentTest.java
@@ -11,11 +11,15 @@ import java.lang.Thread;
 
 import java.io.InputStream;
 
+import java.util.Set;
+
 import java.rmi.registry.LocateRegistry;
 import java.rmi.RemoteException;
 import java.rmi.registry.Registry;
 import java.math.BigDecimal;
 
+import pl.edu.mimuw.cloudatlas.api.Api;
+
 public class AgentTest {
     private static Process registryProcess;
     private static Process agentProcess;
@@ -39,14 +43,13 @@ public class AgentTest {
     }
 
     @Test
-    public void testPing() {
+    public void testGetZoneSet() {
 		try {
 			Registry registry = LocateRegistry.getRegistry("localhost");
 			Api api = (Api) registry.lookup("Api");
-			int res = api.ping(10);
-            assertEquals(11, res);
+			Set<String> set = api.getZoneSet();
+            assertEquals(null, set);
 		} catch (Exception e) {
-			System.err.println("FibonacciClient exception:");
 			e.printStackTrace();
             assertTrue(false);
 		}