package com.orbekk.same;

import java.util.Map;
import java.util.HashMap;

import com.orbekk.paxos.PaxosService;

/**
 * This class is used in test.
 */
public class TestConnectionManager implements ConnectionManager {
    public Map<String, ClientService> clientMap =
        new HashMap<String, ClientService>();
    public Map<String, MasterService> masterMap =
            new HashMap<String, MasterService>();
    public Map<String, PaxosService> paxosMap =
            new HashMap<String, PaxosService>();

    public TestConnectionManager() {
    }

    public ClientService getClient(String url) {
        return clientMap.get(url);
    }

    public MasterService getMaster(String url) {
        return masterMap.get(url);
    }
    
    public PaxosService getPaxos(String url) {
        return paxosMap.get(url);
    }
}