mirror of https://github.com/avast/PurpleDome
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
import unittest
|
|
from unittest.mock import patch
|
|
from app.metasploit import Metasploit
|
|
from app.attack_log import AttackLog
|
|
from pymetasploit3.msfrpc import MsfRpcClient
|
|
import requests
|
|
from app.exceptions import ServerError
|
|
import time
|
|
|
|
# https://docs.python.org/3/library/unittest.html
|
|
|
|
|
|
class FakeAttacker():
|
|
|
|
def __init__(self):
|
|
pass
|
|
|
|
def remote_run(self, cmd, disown):
|
|
pass
|
|
|
|
def get_ip(self):
|
|
return "66.55.44.33"
|
|
|
|
|
|
class TestMetasploit(unittest.TestCase):
|
|
|
|
def setUp(self) -> None:
|
|
with patch.object(time, "sleep") as _:
|
|
self.attack_logger = AttackLog(0)
|
|
|
|
@unittest.skip("temporary skip. Needs to be adopted")
|
|
def test_basic_init(self):
|
|
with patch.object(time, "sleep") as _:
|
|
m = Metasploit("FooBar", self.attack_logger)
|
|
self.assertEqual(m.password, "FooBar")
|
|
self.assertEqual(m.attack_logger, self.attack_logger)
|
|
|
|
@unittest.skip("temporary skip. Needs to be adopted")
|
|
def test_msfrpcd_cmd(self):
|
|
attacker = FakeAttacker()
|
|
with patch.object(time, "sleep") as _:
|
|
m = Metasploit("FooBar", self.attack_logger, attacker=attacker, username="Pennywise")
|
|
self.assertEqual(m.__msfrpcd_cmd__(), "killall msfrpcd; nohup msfrpcd -P FooBar -U Pennywise -S &")
|
|
|
|
@unittest.skip("temporary skip. Needs to be adopted")
|
|
def test_get_client_simple(self):
|
|
attacker = FakeAttacker()
|
|
with patch.object(time, "sleep") as _:
|
|
m = Metasploit("FooBar", self.attack_logger, attacker=attacker, username="Pennywise")
|
|
m.client = "Foo"
|
|
self.assertEqual(m.get_client(), "Foo")
|
|
|
|
@unittest.skip("temporary skip. Needs to be adopted")
|
|
def test_get_client_success(self):
|
|
attacker = FakeAttacker()
|
|
with patch.object(time, "sleep") as _:
|
|
m = Metasploit("FooBar", self.attack_logger, attacker=attacker, username="Pennywise")
|
|
with patch.object(MsfRpcClient, "__init__", return_value=None) as mock_method:
|
|
m.get_client()
|
|
mock_method.assert_called_once_with("FooBar", attacker=attacker, username="Pennywise", server="66.55.44.33")
|
|
|
|
@unittest.skip("temporary skip. Needs to be adopted")
|
|
def test_get_client_retries(self):
|
|
attacker = FakeAttacker()
|
|
with patch.object(time, "sleep") as _:
|
|
m = Metasploit("FooBar", self.attack_logger, attacker=attacker, username="Pennywise")
|
|
with self.assertRaises(ServerError):
|
|
with patch.object(MsfRpcClient, "__init__", side_effect=requests.exceptions.ConnectionError()) as mock_method:
|
|
m.get_client()
|
|
mock_method.assert_called_with("FooBar", attacker=attacker, username="Pennywise", server="66.55.44.33")
|