1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21 package info.mikethomas.jfold;
22
23 import info.mikethomas.jfold.info.InfoItem;
24 import info.mikethomas.jfold.options.Options;
25 import info.mikethomas.jfold.simulation.SimulationInfo;
26 import info.mikethomas.jfold.slot.Slot;
27 import info.mikethomas.jfold.slot.SlotOptions;
28 import info.mikethomas.jfold.unit.Unit;
29 import info.mikethomas.jfold.util.Command;
30
31 import java.io.IOException;
32 import java.util.List;
33
34 import lombok.extern.slf4j.XSlf4j;
35
36 import org.apache.commons.io.FileUtils;
37
38
39
40
41
42
43
44 @XSlf4j
45 public class MockConnection extends ClientConnection implements Connection {
46
47
48
49
50
51
52 public MockConnection() throws IOException {
53
54 super(null, 0);
55 }
56
57
58 @Override
59 protected String sendCommand(Command command, List<String> args) {
60 switch (command) {
61 case INFO:
62 if (args.contains("Folding@home Client") && args.contains("Website")) {
63 return "http://folding.stanford.edu/";
64 } else {
65 return getJson(InfoItem.class);
66 }
67
68 case NUM_SLOTS:
69 return "1";
70
71 case OPTIONS:
72 return getJson(Options.class);
73
74 case PPD:
75 return "0";
76
77 case QUEUE_INFO:
78 return getJson(Unit.class);
79
80 case SIMULATION_INFO:
81 return getJson(SimulationInfo.class);
82
83 case SLOT_INFO:
84 return getJson(Slot.class);
85
86 case SLOT_OPTIONS:
87 return getJson(SlotOptions.class);
88
89 case UPTIME:
90 return "9d 3h 32m 33s";
91
92 default:
93 return null;
94 }
95 }
96
97 private String getJson(Class clazz) {
98 try {
99 var filename = "Example" + clazz.getSimpleName() + ".json";
100 var url = clazz.getResource(filename);
101 var file = FileUtils.toFile(url);
102 var json = FileUtils.readFileToString(file, ENCODING);
103 log.info(json);
104 return json;
105 } catch (IOException e) {
106 throw new UnsupportedOperationException(e);
107 }
108 }
109 }