p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Fixed UTs

This commit is contained in:
Shay Nehmad 2019-09-16 17:41:26 +03:00
parent 9f98025d33
commit 841e54afc8
1 changed files with 69 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

85
monkey/monkey_island/cc/services/reporting/test_zero_trust_service.py
View File

@ -54,14 +54,14 @@ class TestZeroTrustService(IslandTestCase):
STATUS_FAILED: 0, STATUS_FAILED: 0,
STATUS_VERIFY: 2, STATUS_VERIFY: 2,
STATUS_PASSED: 0, STATUS_PASSED: 0,
STATUS_UNEXECUTED: 0, STATUS_UNEXECUTED: 1,
"pillar": "People" "pillar": "People"
}, },
{ {
STATUS_FAILED: 0, STATUS_FAILED: 0,
STATUS_VERIFY: 2, STATUS_VERIFY: 2,
STATUS_PASSED: 0, STATUS_PASSED: 0,
STATUS_UNEXECUTED: 2, STATUS_UNEXECUTED: 4,
"pillar": "Networks" "pillar": "Networks"
}, },
{ {
@ -82,7 +82,7 @@ class TestZeroTrustService(IslandTestCase):
STATUS_FAILED: 0, STATUS_FAILED: 0,
STATUS_VERIFY: 0, STATUS_VERIFY: 0,
STATUS_PASSED: 0, STATUS_PASSED: 0,
STATUS_UNEXECUTED: 1, STATUS_UNEXECUTED: 3,
"pillar": "Visibility & Analytics" "pillar": "Visibility & Analytics"
}, },
{ {
@ -102,6 +102,8 @@ class TestZeroTrustService(IslandTestCase):
self.fail_if_not_testing_env() self.fail_if_not_testing_env()
self.clean_finding_db() self.clean_finding_db()
self.maxDiff = None
save_example_findings() save_example_findings()
expected = { expected = {
@ -111,14 +113,14 @@ class TestZeroTrustService(IslandTestCase):
"principle": PRINCIPLES[PRINCIPLE_DATA_TRANSIT], "principle": PRINCIPLES[PRINCIPLE_DATA_TRANSIT],
"status": STATUS_FAILED, "status": STATUS_FAILED,
"tests": [ "tests": [
{
"status": STATUS_FAILED,
"test": TESTS_MAP[TEST_DATA_ENDPOINT_HTTP][TEST_EXPLANATION_KEY]
},
{ {
"status": STATUS_UNEXECUTED, "status": STATUS_UNEXECUTED,
"test": TESTS_MAP[TEST_DATA_ENDPOINT_ELASTIC][TEST_EXPLANATION_KEY] "test": TESTS_MAP[TEST_DATA_ENDPOINT_ELASTIC][TEST_EXPLANATION_KEY]
}, },
{
"status": STATUS_FAILED,
"test": TESTS_MAP[TEST_DATA_ENDPOINT_HTTP][TEST_EXPLANATION_KEY]
}
] ]
} }
], ],
@ -127,14 +129,14 @@ class TestZeroTrustService(IslandTestCase):
"principle": PRINCIPLES[PRINCIPLE_ENDPOINT_SECURITY], "principle": PRINCIPLES[PRINCIPLE_ENDPOINT_SECURITY],
"status": STATUS_FAILED, "status": STATUS_FAILED,
"tests": [ "tests": [
{
"status": STATUS_UNEXECUTED,
"test": TESTS_MAP[TEST_MACHINE_EXPLOITED][TEST_EXPLANATION_KEY]
},
{ {
"status": STATUS_FAILED, "status": STATUS_FAILED,
"test": TESTS_MAP[TEST_ENDPOINT_SECURITY_EXISTS][TEST_EXPLANATION_KEY] "test": TESTS_MAP[TEST_ENDPOINT_SECURITY_EXISTS][TEST_EXPLANATION_KEY]
}, },
{
"status": STATUS_UNEXECUTED,
"test": TESTS_MAP[TEST_MACHINE_EXPLOITED][TEST_EXPLANATION_KEY]
}
] ]
} }
], ],
@ -159,6 +161,16 @@ class TestZeroTrustService(IslandTestCase):
} }
] ]
}, },
{
"principle": PRINCIPLES[PRINCIPLE_USERS_MAC_POLICIES],
"status": STATUS_UNEXECUTED,
"tests": [
{
"status": STATUS_UNEXECUTED,
"test": TESTS_MAP[TEST_COMMUNICATE_AS_NEW_USER][TEST_EXPLANATION_KEY]
}
]
},
{ {
"principle": PRINCIPLES[PRINCIPLE_ANALYZE_NETWORK_TRAFFIC], "principle": PRINCIPLES[PRINCIPLE_ANALYZE_NETWORK_TRAFFIC],
"status": STATUS_UNEXECUTED, "status": STATUS_UNEXECUTED,
@ -168,7 +180,17 @@ class TestZeroTrustService(IslandTestCase):
"test": TESTS_MAP[TEST_MALICIOUS_ACTIVITY_TIMELINE][TEST_EXPLANATION_KEY] "test": TESTS_MAP[TEST_MALICIOUS_ACTIVITY_TIMELINE][TEST_EXPLANATION_KEY]
} }
] ]
} },
{
"principle": PRINCIPLES[PRINCIPLE_RESTRICTIVE_NETWORK_POLICIES],
"status": STATUS_UNEXECUTED,
"tests": [
{
"status": STATUS_UNEXECUTED,
"test": TESTS_MAP[TEST_TUNNELING][TEST_EXPLANATION_KEY]
}
]
},
], ],
PEOPLE: [ PEOPLE: [
{ {
@ -180,9 +202,29 @@ class TestZeroTrustService(IslandTestCase):
"test": TESTS_MAP[TEST_SCHEDULED_EXECUTION][TEST_EXPLANATION_KEY] "test": TESTS_MAP[TEST_SCHEDULED_EXECUTION][TEST_EXPLANATION_KEY]
} }
] ]
},
{
"principle": PRINCIPLES[PRINCIPLE_USERS_MAC_POLICIES],
"status": STATUS_UNEXECUTED,
"tests": [
{
"status": STATUS_UNEXECUTED,
"test": TESTS_MAP[TEST_COMMUNICATE_AS_NEW_USER][TEST_EXPLANATION_KEY]
}
]
} }
], ],
"Visibility & Analytics": [ VISIBILITY_ANALYTICS: [
{
"principle": PRINCIPLES[PRINCIPLE_USERS_MAC_POLICIES],
"status": STATUS_UNEXECUTED,
"tests": [
{
"status": STATUS_UNEXECUTED,
"test": TESTS_MAP[TEST_COMMUNICATE_AS_NEW_USER][TEST_EXPLANATION_KEY]
}
]
},
{ {
"principle": PRINCIPLES[PRINCIPLE_ANALYZE_NETWORK_TRAFFIC], "principle": PRINCIPLES[PRINCIPLE_ANALYZE_NETWORK_TRAFFIC],
"status": STATUS_UNEXECUTED, "status": STATUS_UNEXECUTED,
@ -192,12 +234,23 @@ class TestZeroTrustService(IslandTestCase):
"test": TESTS_MAP[TEST_MALICIOUS_ACTIVITY_TIMELINE][TEST_EXPLANATION_KEY] "test": TESTS_MAP[TEST_MALICIOUS_ACTIVITY_TIMELINE][TEST_EXPLANATION_KEY]
} }
] ]
} },
{
"principle": PRINCIPLES[PRINCIPLE_RESTRICTIVE_NETWORK_POLICIES],
"status": STATUS_UNEXECUTED,
"tests": [
{
"status": STATUS_UNEXECUTED,
"test": TESTS_MAP[TEST_TUNNELING][TEST_EXPLANATION_KEY]
}
]
},
], ],
"Workloads": [] WORKLOADS: []
} }
self.assertEquals(ZeroTrustService.get_principles_status(), expected) result = ZeroTrustService.get_principles_status()
self.assertEquals(result, expected)
def test_get_pillars_to_statuses(self): def test_get_pillars_to_statuses(self):
self.fail_if_not_testing_env() self.fail_if_not_testing_env()