|
1
|
|
|
'''test mining domain''' |
|
2
|
|
|
import unittest |
|
3
|
|
|
import datetime |
|
4
|
|
|
import domain.minerstatistics |
|
5
|
|
|
from domain.mining import Miner, MinerStatus, MinerInfo, MinerApiCall |
|
6
|
|
|
|
|
7
|
|
|
class TestMiner(unittest.TestCase): |
|
8
|
|
|
def test_miner_is_not_disabled(self): |
|
9
|
|
|
miner = Miner("test", '', '', '', '', '', '', '', '') |
|
10
|
|
|
self.assertFalse(miner.is_disabled()) |
|
11
|
|
|
|
|
12
|
|
|
def test_miner_is_disabled(self): |
|
13
|
|
|
miner = Miner("#test", '', '', '', '', '', '', '', '') |
|
14
|
|
|
self.assertTrue(miner.is_disabled()) |
|
15
|
|
|
|
|
16
|
|
|
def test_miner_key_is_id(self): |
|
17
|
|
|
miner = Miner("test", minerid='unittest') |
|
18
|
|
|
self.assertTrue(miner.key() == 'unittest') |
|
19
|
|
|
|
|
20
|
|
|
def test_miner_key_not_unknown(self): |
|
21
|
|
|
miner = Miner('test', minerid='unknown', ipaddress='unittest', port='1234') |
|
22
|
|
|
self.assertTrue(miner.key() == 'unittest:1234') |
|
23
|
|
|
|
|
24
|
|
|
def test_miner_key_is_networkid(self): |
|
25
|
|
|
miner = Miner("test", networkid={'unittest1', 'unittest2'}) |
|
26
|
|
|
self.assertTrue(miner.key() == str({'unittest1', 'unittest2'})) |
|
27
|
|
|
|
|
28
|
|
|
def test_key_notempty_net(self): |
|
29
|
|
|
miner = Miner("test", networkid={}) |
|
30
|
|
|
self.assertFalse(miner.key() == str({})) |
|
31
|
|
|
|
|
32
|
|
|
def test_miner_key_is_ip(self): |
|
33
|
|
|
miner = Miner("test", ipaddress='unittest', port='1234') |
|
34
|
|
|
self.assertTrue(miner.key() == 'unittest:1234') |
|
35
|
|
|
|
|
36
|
|
|
def test_miner_key_is_name(self): |
|
37
|
|
|
miner = Miner("test") |
|
38
|
|
|
self.assertTrue(miner.key() == 'test') |
|
39
|
|
|
|
|
40
|
|
|
def test_miner_formattime_now(self): |
|
41
|
|
|
miner = Miner('test') |
|
42
|
|
|
miner.lastmonitor = datetime.datetime.utcnow() |
|
43
|
|
|
self.assertTrue(miner.formattime(miner.lastmonitor)) |
|
44
|
|
|
|
|
45
|
|
|
def test_miner_formattime_string(self): |
|
46
|
|
|
miner = Miner('test') |
|
47
|
|
|
miner.lastmonitor = '2018-04-11T01:28:45+00:00' |
|
48
|
|
|
self.assertTrue(miner.formattime(miner.lastmonitor)) |
|
49
|
|
|
|
|
50
|
|
|
def test_miner_uses_custom(self): |
|
51
|
|
|
miner = Miner('test') |
|
52
|
|
|
miner.ftpport = '99' |
|
53
|
|
|
miner.set_ftp_port('22') |
|
54
|
|
|
self.assertTrue(miner.ftpport == '99') |
|
55
|
|
|
|
|
56
|
|
|
def test_miner_uses_default(self): |
|
57
|
|
|
miner = Miner('test') |
|
58
|
|
|
miner.ftpport = '' |
|
59
|
|
|
miner.set_ftp_port('22') |
|
60
|
|
|
self.assertTrue(miner.ftpport == '22') |
|
61
|
|
|
|
|
62
|
|
|
def test_miner_status_set_last_update(self): |
|
63
|
|
|
miner = Miner('test') |
|
64
|
|
|
self.assertTrue(not miner.laststatuschanged) |
|
65
|
|
|
miner.status = MinerStatus.Offline |
|
66
|
|
|
self.assertTrue(miner.laststatuschanged) |
|
67
|
|
|
|
|
68
|
|
|
def test_miner_status_change_keeps_original(self): |
|
69
|
|
|
miner = Miner('test') |
|
70
|
|
|
miner.status = MinerStatus.Offline |
|
71
|
|
|
self.assertTrue(miner.laststatuschanged) |
|
72
|
|
|
originalstatuschangetime = miner.laststatuschanged |
|
73
|
|
|
miner.status = MinerStatus.Offline |
|
74
|
|
|
self.assertTrue(miner.laststatuschanged == originalstatuschangetime) |
|
75
|
|
|
|
|
76
|
|
|
def test_miner_status_no_you_cant(self): |
|
77
|
|
|
miner = Miner('test') |
|
78
|
|
|
def set_status_test(): |
|
79
|
|
|
miner.status = 'you can be anyting' |
|
80
|
|
|
self.assertRaises(ValueError, set_status_test) |
|
81
|
|
|
|
|
82
|
|
|
def test_miner_reset_offline_count(self): |
|
83
|
|
|
miner = Miner('test') |
|
84
|
|
|
self.assertTrue(miner.offlinecount == 0) |
|
85
|
|
|
self.assertTrue(miner.is_send_offline_alert()) |
|
86
|
|
|
miner.offline_now() |
|
87
|
|
|
miner.offline_now() |
|
88
|
|
|
miner.offline_now() |
|
89
|
|
|
miner.offline_now() |
|
90
|
|
|
self.assertFalse(miner.is_send_offline_alert()) |
|
91
|
|
|
miner.online_now() |
|
92
|
|
|
self.assertTrue(miner.offlinecount == 0) |
|
93
|
|
|
self.assertTrue(miner.is_send_offline_alert()) |
|
94
|
|
|
|
|
95
|
|
|
def test_miner_setminerinfo(self): |
|
96
|
|
|
miner = Miner('test') |
|
97
|
|
|
minerinfo = MinerInfo('testminertype', miner.minerid) |
|
98
|
|
|
miner.setminerinfo(minerinfo) |
|
99
|
|
|
self.assertTrue(miner.miner_type == minerinfo.miner_type) |
|
100
|
|
|
|
|
101
|
|
|
def test_miner_shouldmonitor(self): |
|
102
|
|
|
miner = Miner('test') |
|
103
|
|
|
miner.lastmonitor = "" |
|
104
|
|
|
self.assertTrue(miner.should_monitor()) |
|
105
|
|
|
|
|
106
|
|
|
def test_miner_elapsed_format(self): |
|
107
|
|
|
miner = Miner('test') |
|
108
|
|
|
miner.minerstats = domain.minerstatistics.MinerStatistics(miner) |
|
109
|
|
|
self.assertTrue(miner.minerstats.format_elapsed() == "0:0:0:0") |
|
110
|
|
|
|
|
111
|
|
|
def test_miner_monitored_nothing_none(self): |
|
112
|
|
|
miner = Miner('test') |
|
113
|
|
|
miner.monitored(None, None, None, None) |
|
114
|
|
|
self.assertTrue(miner.lastmonitor is None) |
|
115
|
|
|
|
|
116
|
|
|
def test_miner_monitored_not_nothing_ismonitored(self): |
|
117
|
|
|
miner = Miner('test') |
|
118
|
|
|
stats = domain.minerstatistics.MinerStatistics(miner) |
|
119
|
|
|
miner.monitored(stats, pool=None, info=None, sec=None) |
|
120
|
|
|
self.assertFalse(miner.lastmonitor is None) |
|
121
|
|
|
|
|
122
|
|
|
def test_miner_monitored_timer(self): |
|
123
|
|
|
miner = Miner('test') |
|
124
|
|
|
stats = domain.minerstatistics.MinerStatistics(miner) |
|
125
|
|
|
apicall = MinerApiCall(miner) |
|
126
|
|
|
apicall.start() |
|
127
|
|
|
apicall.stop() |
|
128
|
|
|
miner.monitored(stats, pool=None, info=None, sec=apicall.elapsed()) |
|
129
|
|
|
self.assertFalse(miner.lastmonitor is None) |
|
130
|
|
|
|
|
131
|
|
|
def test_miner_update(self): |
|
132
|
|
|
miner = Miner('test') |
|
133
|
|
|
miner.ipaddress = 'ip1' |
|
134
|
|
|
miner.location = 'location1' |
|
135
|
|
|
miner.in_service_date = datetime.datetime.now().replace(second=0, microsecond=0) |
|
136
|
|
|
minerupdate = Miner('test') |
|
137
|
|
|
minerupdate.ipaddress = 'ip2' |
|
138
|
|
|
minerupdate.location = 'location2' |
|
139
|
|
|
minerupdate.in_service_date = datetime.datetime.now() |
|
140
|
|
|
miner.updatefrom(minerupdate) |
|
141
|
|
|
self.assertTrue(miner.ipaddress == minerupdate.ipaddress) |
|
142
|
|
|
self.assertTrue(miner.location == minerupdate.location) |
|
143
|
|
|
self.assertTrue(miner.in_service_date == minerupdate.in_service_date) |
|
144
|
|
|
|
|
145
|
|
|
def test_miner_is_unknown(self): |
|
146
|
|
|
miner = Miner('test') |
|
147
|
|
|
miner.minerid = "unknown" |
|
148
|
|
|
self.assertTrue(miner.is_unknown) |
|
149
|
|
|
|
|
150
|
|
|
def test_miner_is_known(self): |
|
151
|
|
|
miner = Miner('test') |
|
152
|
|
|
miner.minerid = "123.123.123.123" |
|
153
|
|
|
self.assertFalse(miner.is_unknown) |
|
154
|
|
|
|
|
155
|
|
|
def test_miner_no_update(self): |
|
156
|
|
|
miner = Miner('test') |
|
157
|
|
|
miner.ipaddress = 'ip1' |
|
158
|
|
|
minerupdate = Miner('test') |
|
159
|
|
|
minerupdate.ipaddress = None |
|
160
|
|
|
miner.updatefrom(minerupdate) |
|
161
|
|
|
self.assertTrue(miner.ipaddress != minerupdate.ipaddress) |
|
162
|
|
|
|
|
163
|
|
|
def test_miner_update_port(self): |
|
164
|
|
|
miner = Miner('test') |
|
165
|
|
|
miner.port = 'port1' |
|
166
|
|
|
minerupdate = Miner('test') |
|
167
|
|
|
minerupdate.port = 'port2' |
|
168
|
|
|
miner.updatefrom(minerupdate) |
|
169
|
|
|
self.assertTrue(miner.port == minerupdate.port) |
|
170
|
|
|
|
|
171
|
|
|
def test_miner_create(self): |
|
172
|
|
|
values = [] |
|
173
|
|
|
values.append({"name":"UnitTest"}) |
|
174
|
|
|
values.append({"minerid": "1"}) |
|
175
|
|
|
values.append({"ipaddress": "123.123.123.123"}) |
|
176
|
|
|
values.append({"port": "987"}) |
|
177
|
|
|
values.append({"location": "rack"}) |
|
178
|
|
|
values.append({"in_service_date": datetime.date(2018,1, 1)}) |
|
179
|
|
|
miner = Miner.create(values) |
|
180
|
|
|
self.assertTrue(miner.name == "UnitTest") |
|
181
|
|
|
self.assertTrue(miner.minerid == "1") |
|
182
|
|
|
self.assertTrue(miner.ipaddress == "123.123.123.123") |
|
183
|
|
|
self.assertTrue(miner.port == "987") |
|
184
|
|
|
self.assertTrue(miner.location == "rack") |
|
185
|
|
|
self.assertTrue(miner.in_service_date == datetime.date(2018,1, 1) ) |
|
186
|
|
|
|
|
187
|
|
|
def test_miner_key_original(self): |
|
188
|
|
|
miner = Miner('test') |
|
189
|
|
|
miner.minerid = "unittest" |
|
190
|
|
|
self.assertTrue(miner.is_key_updated) |
|
191
|
|
|
|
|
192
|
|
|
if __name__ == '__main__': |
|
193
|
|
|
unittest.main() |
|
194
|
|
|
|
dfoderick /
fullcycle
