Agent: Add AutomatedMaster to master/__init__.py

2021-12-09 14:56:54 -05:00 · 2021-12-09 14:56:54 -05:00 · 86203c8138
parent da8e814b95
commit 86203c8138
2 changed files with 3 additions and 1 deletions
--- a/monkey/infection_monkey/master/init.py
+++ b/monkey/infection_monkey/master/init.py
@ -0,0 +1 @@
+from .automated_master import AutomatedMaster
--- a/monkey/tests/unit_tests/infection_monkey/master/test_automated_master.py
+++ b/monkey/tests/unit_tests/infection_monkey/master/test_automated_master.py
@ -1,4 +1,5 @@
-from infection_monkey.master.automated_master import AutomatedMaster
+from infection_monkey.master import AutomatedMaster
+

 def test_terminate_without_start():
    m = AutomatedMaster(None, None, None)
				`@ -0,0 +1 @@`
				`from .automated_master import AutomatedMaster`