same/src/main/java/com/orbekk/same/TestConnectionManager.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

package com.orbekk.same;

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

import com.orbekk.paxos.PaxosService;
import com.orbekk.same.Services.Directory;

/**
 * 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 Map<String, Services.Directory> directoryMap =
            new HashMap<String, Services.Directory>();

    public TestConnectionManager() {
    }

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

    @Override
    public MasterService getMaster(String url) {
        return masterMap.get(url);
    }

    @Override
    public PaxosService getPaxos(String url) {
        return paxosMap.get(url);
    }

    @Override
    public Directory getDirectory(String location) {
        return directoryMap.get(location);
    }
}