Support PaxosService in ConnectionManager.

1 files changed, 9 insertions, 1 deletions

diff --git a/same/src/main/java/com/orbekk/same/TestConnectionManager.java b/same/src/main/java/com/orbekk/same/TestConnectionManager.java
index 4e26894..3439a78 100644
--- a/same/src/main/java/com/orbekk/same/TestConnectionManager.java
+++ b/same/src/main/java/com/orbekk/same/TestConnectionManager.java
@@ -3,6 +3,8 @@ package com.orbekk.same;
 import java.util.Map;
 import java.util.HashMap;
 
+import com.orbekk.paxos.PaxosService;
+
 /**
  * This class is used in test.
  */
@@ -10,7 +12,9 @@ public class TestConnectionManager implements ConnectionManager {
     public Map<String, ClientService> clientMap =
         new HashMap<String, ClientService>();
     public Map<String, MasterService> masterMap =
-        new HashMap<String, MasterService>();
+            new HashMap<String, MasterService>();
+    public Map<String, PaxosService> paxosMap =
+            new HashMap<String, PaxosService>();
 
     public TestConnectionManager() {
     }
@@ -22,4 +26,8 @@ public class TestConnectionManager implements ConnectionManager {
     public MasterService getMaster(String url) {
         return masterMap.get(url);
     }
+    
+    public PaxosService getPaxos(String url) {
+        return paxosMap.get(url);
+    }
 }