From db5b65c463ae6ad68e4be73ab1e5e3a12cec4a38 Mon Sep 17 00:00:00 2001
From: byt3bl33d3r <byt3bl33d3r@gmail.com>
Date: Sun, 23 Aug 2015 20:11:27 +0200
Subject: [PATCH] trying some new tests out

---
 config/mitmf.conf    |  2 +-
 tests/basic_tests.py | 20 ++++++++++++++++----
 2 files changed, 17 insertions(+), 5 deletions(-)

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