Added tests for Cloner.add_schema (#260)

Handled the case when there is no proper host provided and also added tests for the cases when host provided has length less than 4 which indicates no proper domain was provided.
Refs #187

snare/cloner.py

@@ -21,7 +21,7 @@ def __init__(self, root, max_depth, css_validate):
         self.root, self.error_page = self.add_scheme(root)
         self.max_depth = max_depth
         self.moved_root = None
-        if len(self.root.host) < 4:
+        if (self.root.host is None) or (len(self.root.host) < 4):
             sys.exit('invalid target {}'.format(self.root.host))
         self.target_path = '/opt/snare/pages/{}'.format(self.root.host)
         if not os.path.exists(self.target_path):

snare/tests/test_cloner_add_scheme.py

@@ -38,3 +38,13 @@ def test_no_scheme(self):
 
     def tearDown(self):
         shutil.rmtree(self.main_page_path)
+
+    def test_no_host(self):
+        self.url = 'http:/'
+        with self.assertRaises(SystemExit):
+            Cloner(self.url, self.max_depth, self.css_validate)
+
+    def test_limited_length_host(self):
+        self.url = 'http://aaa'
+        with self.assertRaises(SystemExit):
+            Cloner(self.url, self.max_depth, self.css_validate)