Fixed testcases

Author: PSNAppz

modules/getemails.py

@@ -1,3 +1,6 @@
+import sys
+import os
+sys.path.append(os.path.abspath('../'))
 from modules.bcolors import bcolors
 import bs4
 

modules/getweblinks.py

@@ -1,5 +1,8 @@
+import sys
+import os
+sys.path.append(os.path.abspath('../'))
 import urllib.request 
-import modules.bcolors
+from modules.bcolors import bcolors
 import bs4
 
 """Get all onion links from the website"""

tests/__init__.py

@@ -4,6 +4,7 @@
 from io import StringIO
 sys.path.append(os.path.abspath('../modules'))
 import getemails
+from bcolors import bcolors
 import pagereader
 
 soup = pagereader.readPage('http://www.whatsmyip.net/')
@@ -14,10 +15,10 @@ def setUp(self):
 		self.held, sys.stdout = sys.stdout, StringIO()
 
 	def test_print_emails(self):
-		data = "\nMails Found - 1\n-------------------------------\nadvertise@provaz.eu\n"
+		data = "\n"+bcolors.OKGREEN+"Mails Found - "+bcolors.ENDC+"1\n-------------------------------\nadvertise@provaz.eu\n"
 		getemails.getMails(soup)
 		self.assertEqual(sys.stdout.getvalue(),data)
-
+		
 
 if __name__ == '__main__':
 	unittest.main()

tests/test_getemails.py

@@ -4,6 +4,7 @@
 from io import StringIO
 sys.path.append(os.path.abspath('../modules'))
 import getweblinks
+from bcolors import bcolors
 import pagereader
 
 soup = pagereader.readPage('http://www.whatsmyip.net/')