1 files changed, 8 insertions, 0 deletions
diff --git a/same/src/main/java/com/orbekk/same/TestConnectionManager.java b/same/src/main/java/com/orbekk/same/TestConnectionManager.java
index 25a3ee6..6f5e5d8 100644
--- a/same/src/main/java/com/orbekk/same/TestConnectionManager.java
+++ b/same/src/main/java/com/orbekk/same/TestConnectionManager.java
@@ -4,6 +4,7 @@ import java.util.Map;
 import java.util.HashMap;
 
 import com.orbekk.paxos.PaxosService;
+import com.orbekk.same.discovery.DirectoryService;
 
 /**
  * This class is used in test.
@@ -15,6 +16,8 @@ public class TestConnectionManager implements ConnectionManager {
             new HashMap<String, MasterService>();
     public Map<String, PaxosService> paxosMap =
             new HashMap<String, PaxosService>();
+    public Map<String, DirectoryService> directoryMap =
+            new HashMap<String, DirectoryService>();
 
     public TestConnectionManager() {
     }
@@ -33,4 +36,9 @@ public class TestConnectionManager implements ConnectionManager {
     public PaxosService getPaxos(String url) {
         return paxosMap.get(url);
     }
+
+    @Override
+    public DirectoryService getDirectory(String url) {
+        return directoryMap.get(url);
+    }
 }