archive
/
PurpleDome
mirror of https://github.com/avast/PurpleDome
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Adopted machine unit tests

Browse Source
pull/44/head
Thorsten Sick 3 years ago
parent 4721bc986c
commit cae79c5393
1 changed files with 159 additions and 91 deletions
Show Stats Download Patch File Download Diff File

250
tests/test_machinecontrol.py
Unescape Escape View File

@ -6,6 +6,7 @@ from app.exceptions import ConfigurationError
from app.config import MachineConfig
from unittest.mock import patch
from app.attack_log import AttackLog
from app.config_verifier import Attacker, Target
# https://docs.python.org/3/library/unittest.html
@ -16,13 +17,23 @@ class TestMachineControl(unittest.TestCase):
self.attack_logger = AttackLog(0)
def test_get_os_linux_machine(self):
m = Machine(DotMap({"root": "systems/attacker1",
"os": "linux",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3"}), self.attack_logger)
conf = { # "root": "systems/attacker1",
"os": "linux",
"vm_name": "foo_bar",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"machinepath": "target3",
"nicknames": [],
"sensors": [],
"paw": "ignoreme",
"name": "Foobar",
"group": "some_group",
}
m = Machine(Target(**conf), self.attack_logger)
self.assertEqual(m.get_os(), "linux")
def test_get_os_linux_machine_with_config_class(self):
@ -37,48 +48,78 @@ class TestMachineControl(unittest.TestCase):
self.assertEqual(m.get_os(), "linux")
def test_get_paw_good(self):
m = Machine(DotMap({"root": "systems/attacker1",
"os": "linux",
"paw": "testme",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3"}), self.attack_logger)
conf = { # "root": "systems/attacker1",
"os": "linux",
"vm_name": "foo_bar",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"machinepath": "target3",
"nicknames": [],
"sensors": [],
"paw": "testme",
"name": "Foobar",
"group": "some_group",
}
m = Machine(Target(**conf), self.attack_logger)
self.assertEqual(m.get_paw(), "testme")
def test_get_paw_missing(self):
m = Machine(DotMap({"root": "systems/attacker1",
"os": "linux",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3"
}), self.attack_logger)
self.assertEqual(m.get_paw(), None)
conf = {# "root": "systems/attacker1",
"os": "linux",
"vm_name": "foo_bar",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"machinepath": "target3",
"nicknames": [],
"sensors": [],
"name": "Foobar",
"group": "some_group",
}
with self.assertRaisesRegex(TypeError, 'paw'):
m = Machine(Target(**conf), self.attack_logger)
def test_get_group_good(self):
m = Machine(DotMap({"root": "systems/attacker1",
"os": "linux",
"group": "testme",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3"}), self.attack_logger)
conf = {# "root": "systems/attacker1",
"os": "linux",
"vm_name": "foo_bar",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"machinepath": "target3",
"nicknames": [],
"sensors": [],
"name": "Foobar",
"paw": "some_paw",
"group": "testme"
}
m = Machine(Target(**conf), self.attack_logger)
self.assertEqual(m.get_group(), "testme")
def test_get_group_missing(self):
m = Machine(DotMap({"root": "systems/attacker1",
"os": "linux",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3"
}), self.attack_logger)
self.assertEqual(m.get_group(), None)
conf = {# "root": "systems/attacker1",
"os": "linux",
"vm_name": "foo_bar",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"machinepath": "target3",
"nicknames": [],
"sensors": [],
"name": "Foobar",
"paw": "some_paw",
}
with self.assertRaisesRegex(TypeError, 'group'):
m = Machine(Target(**conf), self.attack_logger)
def test_vagrantfilepath_missing(self):
with self.assertRaises(ConfigurationError):
@ -102,14 +143,18 @@ class TestMachineControl(unittest.TestCase):
}), self.attack_logger)
def test_vagrantfile_existing(self):
m = Machine(DotMap({"root": "systems/attacker1",
"os": "linux",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3"
}), self.attack_logger)
conf = {# "root": "systems/attacker1",
"os": "linux",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"name": "test_attacker",
"nicknames": ["a","b"],
"machinepath": "attacker1"
}
m = Machine(Attacker(**conf), self.attack_logger)
self.assertIsNotNone(m)
# test: auto generated, dir missing
@ -138,33 +183,45 @@ class TestMachineControl(unittest.TestCase):
}), self.attack_logger)
# test auto generated, dir there (external/internal dirs must work !)
def test_auto_generated_machinepath_with_good_config(self):
m = Machine(DotMap({"root": "systems/attacker1",
"os": "linux",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3"
}), self.attack_logger)
vagrantfilepath = os.path.abspath("systems")
ext = os.path.join(vagrantfilepath, "target3")
internal = os.path.join("/vagrant/", "target3")
def test_missing_machinepath_with_good_config_eeception(self):
conf = {# "root": "systems/attacker1",
"os": "linux",
"vm_name": "foo_bar",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"nicknames": [],
"sensors": [],
"name": "Foobar",
"paw": "some_paw",
"group": "some_group",
}
self.assertEqual(m.abs_machinepath_external, ext)
self.assertEqual(m.abs_machinepath_internal, internal)
with self.assertRaisesRegex(TypeError, "machinepath"):
m = Machine(Target(**conf), self.attack_logger)
# test: manual config, dir there (external/internal dirs must work !)
def test_configured_machinepath_with_good_config(self):
m = Machine(DotMap({"root": "systems/attacker1",
"os": "linux",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "missing",
"machinepath": "target3"
}), self.attack_logger)
conf = {# "root": "systems/attacker1",
"os": "linux",
"vm_name": "foo_bar",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"machinepath": "target3",
"nicknames": [],
"sensors": [],
"name": "Foobar",
"paw": "some_paw",
"group": "some_group",
}
m = Machine(Target(**conf), self.attack_logger)
vagrantfilepath = os.path.abspath("systems")
ext = os.path.join(vagrantfilepath, "target3")
internal = os.path.join("/vagrant/", "target3")
@ -183,15 +240,21 @@ class TestMachineControl(unittest.TestCase):
# Create caldera start command and verify it
def test_get_linux_caldera_start_cmd(self):
m = Machine(DotMap({"root": "systems/attacker1",
"os": "linux",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"group": "testgroup",
"paw": "testpaw"}), self.attack_logger)
conf = {# "root": "systems/attacker1",
"os": "linux",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"group": "testgroup",
"paw": "testpaw",
"name": "test_attacker",
"nicknames": ["a","b"],
"machinepath": "target3",
"sensors": []
}
m = Machine(Target(**conf), self.attack_logger)
m.set_caldera_server("http://www.test.test")
with patch.object(m.vm_manager, "get_playground", return_value="/vagrant/target3"):
cmd = m.create_start_caldera_client_cmd()
@ -199,16 +262,21 @@ class TestMachineControl(unittest.TestCase):
# Create caldera start command and verify it (windows)
def test_get_windows_caldera_start_cmd(self):
m = Machine(DotMap({"root": "systems/attacker1",
"os": "windows",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"group": "testgroup",
"paw": "testpaw",
"machinepath": "target3"}), self.attack_logger)
conf = {# "root": "systems/attacker1",
"os": "windows",
"vm_controller": {
"vm_type": "vagrant",
"vagrantfilepath": "systems",
},
"vm_name": "target3",
"group": "testgroup",
"paw": "testpaw",
"name": "test_attacker",
"nicknames": ["a","b"],
"machinepath": "target3",
"sensors": []
}
m = Machine(Target(**conf), self.attack_logger)
m.set_caldera_server("www.test.test")
cmd = m.create_start_caldera_client_cmd()
self.maxDiff = None

Write Preview
Loading…
Cancel
Save