diff --git a/config/mitmf.conf b/config/mitmf.conf index 40c105a..6832dce 100644 --- a/config/mitmf.conf +++ b/config/mitmf.conf @@ -3,7 +3,7 @@ # [MITMf] - + # Required BeEF and Metasploit options [[BeEF]] host = 127.0.0.1 diff --git a/tests/basic_tests.py b/tests/basic_tests.py index 94506d7..4b11843 100644 --- a/tests/basic_tests.py +++ b/tests/basic_tests.py @@ -1,11 +1,23 @@ import unittest -from commands import getstatusoutput +import logging class BasicTests(unittest.TestCase): - def test_exec(self): - status, res = getstatusoutput('sudo python mitmf.py --help') - self.assertEqual(0, status, msg=res) + def test_configfile(self): + from configobj import ConfigObj + config = ConfigObj('config/mitmf.conf') + + def test_logger(self): + from core.logger import logger + logger.log_level = logging.DEBUG + formatter = logging.Formatter("%(asctime)s [unittest] %(message)s", datefmt="%Y-%m-%d %H:%M:%S") + log = logger().setup_logger("unittest", formatter) + + def test_DNSChef(self): + from core.logger import logger + logger.log_level = logging.DEBUG + from core.servers.DNS import DNSChef + DNSChef().start() if __name__ == '__main__': unittest.main() \ No newline at end of file