p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Fix test

This commit is contained in:
itay 2019-06-11 16:42:55 +03:00
parent 4d962feafe
commit 22815b7e02
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
monkey/infection_monkey/telemetry/attack/test_victim_host_telem.py
View File

@ -13,8 +13,17 @@ class TestVictimHostTelem(TestCase):
telem = VictimHostTelem(technique, status, machine) telem = VictimHostTelem(technique, status, machine)
self.assertEqual(telem.technique, technique)
self.assertEqual(telem.status, status)
self.assertEqual(telem.telem_type, 'attack') self.assertEqual(telem.telem_type, 'attack')
self.assertEqual(telem.machine['domain_name'], machine.domain_name)
self.assertEqual(telem.machine['ip_addr'], machine.ip_addr) expected_data = {
'machine': {
'domain_name': machine.domain_name,
'ip_addr': machine.ip_addr
},
'status': status.value,
'technique': technique
}
actual_data = telem.get_data()
self.assertEqual(actual_data, expected_data)