From 496baf7f2ff218826f66a3597f05fade4d30b554 Mon Sep 17 00:00:00 2001
From: Pravin Agre <agrepravin@gmail.com>
Date: Sun, 18 Jun 2017 22:07:55 +0530
Subject: [PATCH 1/3] Changes

---
 test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test.py b/test.py
index 1da1fd27..7fc30911 100644
--- a/test.py
+++ b/test.py
@@ -3,7 +3,7 @@
 from stem.connection import connect
 
 if __name__ == '__main__':
-  controller = connect("agrepravin")
+  controller = connect("Narayanan123")
 
   if not controller:
     sys.exit(1)  # unable to get a connection

From 65735ac8477642c3d5cc848f7b20a1734e50c901 Mon Sep 17 00:00:00 2001
From: Pravin Agre <agrepravin@gmail.com>
Date: Sun, 18 Jun 2017 22:33:58 +0530
Subject: [PATCH 2/3] added test 2nd

---
 modules/__pycache__/getemails.cpython-35.pyc  | Bin 802 -> 802 bytes
 .../__pycache__/getweblinks.cpython-35.pyc    | Bin 628 -> 758 bytes
 modules/getemails.py                          |   1 -
 modules/getweblinks.py                        |  37 +++++++++++-------
 tests/test_getemails.py                       |   4 +-
 tests/test_getweblinks.py                     |  24 ++++++++++++
 6 files changed, 47 insertions(+), 19 deletions(-)
 create mode 100644 tests/test_getweblinks.py

diff --git a/modules/__pycache__/getemails.cpython-35.pyc b/modules/__pycache__/getemails.cpython-35.pyc
index d07f35bb9caa91d006af5cacd661f6dc25ee6320..7ed24256bffcff36c49614e06c273e6414dbfa18 100644
GIT binary patch
delta 94
zcmZ3)wup^WjF*?|+eWv@ppBgJOpH;JHJMx(1t#Y)`7^Rk-p8bEDa63QpvhFk2BeBW
e>>>eFLJ-Kk#bJ}1pHiBWYR3rV6*B<|9u@$eN)avq

delta 94
zcmZ3)wup^WjF*?Itj8^~a3iNY6XUJPnoKT?f|K)@{2AFM?_<)o6lP#x&}1rN15!mG
dc99?|Aq3>!;;_lhPbtkwwPOPEia`Vq3jmmG5itM&

diff --git a/modules/__pycache__/getweblinks.cpython-35.pyc b/modules/__pycache__/getweblinks.cpython-35.pyc
index e70d7b13cbbf7713f52906fe8344aeb1cfa03f20..36599bc7c93db55a8142109a03e604c43d91d690 100644
GIT binary patch
literal 758
zcmaJ<F>ljA6#mY39M@_}5l|2y;Wb0+r7RU76rxU$C6yu~RgoKC8Yj-#+?{C?B@5-x
zu))7zX1o;^W;Q0?B~&U4=X>wI_uhBk-IH&x(FlKhIez=g0lwqLpcpU7)MIifO@<uk
z6L65a_(+)9WVh-4BnCP|W)~VhS&i%!nfgXf0~3Z7gLa@@EP*S+1egU^4&+yy11krY
z9b27vR+vid)8|+b&&8x>&!yDL!;;Ft)P+2<H=%IqLtbF%VB*`5Xzbk)X!dj+=_X-O
z4LX3{vP^q0!+lG&6@R_y@_{Y2Pj|5*Y8}hkM(Z<rq|#06_&_~_dJZ~<CB_>|KI(g8
ztxJQ&?H)f<ndT}wE-H~m`_bab{{P3~=^L)cMH-cfOlDkj8D%OGg^r3*lnJF1G33#i
zIy`9cq|(`_%DY8XHbdi5(lq|bnlMw>dyJ>&B{#t+6X_tybK{<KZTzGxxkyc|l$p@R
zRazQ1=c4HwPm!+K8Q5iTt<n!J{<dViJQGtB@>!DQ1B?4E)jmPqp&$Ia6_1M<k4u@n
z&qUlVhSiJ<t>SV{-HE|TAzOux*UL#fE7B_GDyAEegw;|9<=g}mzqWyWJ1Qavw#6PX
zpY5?*Y@5Da$8$pVkcF(lc9`eLd&Frv(r#Ge^<{3jyHqg2v(14o?E6&qS#NuP03V0B
AcmMzZ

delta 390
zcmYjNO-sW-5PiG(C={ZeViklOijtd0FM?>Pw}RwQ6iU!0YCcSCcEutIJ?PQf{sO^Y
zRnUunK|DJfJ#?6PJM-r4%(6e$x4Cc~r@P-9o*Nwa!irIiBMRXXM8E|hL&WhyIVA=H
z-+bzy#?rgYbFVCtvhebx$b@%Ym3N-JjlIYCN+dF#G8&=H`i~Pj#XG^x0%#cKBh*0J
zZg^%GOb*d-)EsHMk*LN%ZRlrY1}ax?r!uoq1*jQP5*r{Ho4Ft#OVXMxp)(88gtWAh
zG~}CFsyp^>E1LIoZ!p<FLqeMUcX#%RF^viQjyhmN?%V2v`5kL%it^!J?eIQx)G<Hn
m>aP;p1W<2$Yp+_TY~=H0Ck&%<8ip-VImE8On)>7}x7jaGw?<O{

diff --git a/modules/getemails.py b/modules/getemails.py
index 3eb4a3c8..232d0504 100644
--- a/modules/getemails.py
+++ b/modules/getemails.py
@@ -1,4 +1,3 @@
-#from bs4 import BeautifulSoup
 import bs4
 
 """Get all emails from the website"""
diff --git a/modules/getweblinks.py b/modules/getweblinks.py
index de16a261..0a907c1b 100644
--- a/modules/getweblinks.py
+++ b/modules/getweblinks.py
@@ -1,17 +1,24 @@
+import bs4
+
+
 """Get all onion links from the website"""
 def getLinks(soup):
-    websites = []
-    for link in soup.find_all('a'):
-        email_link = link.get('href')
-        if email_link != None:
-            if 'http' in email_link:
-                websites.append(email_link)
-        else:
-            pass
-    """Pretty print output as below"""
-    print ('') 
-    print ('Websites Found - '+str(len(websites)))
-    print ('-------------------------------')
-    for web in websites:
-        print (web)
-    return ''       
\ No newline at end of file
+    _soup_instance = bs4.BeautifulSoup
+    if isinstance(type(soup), type(_soup_instance)):
+        websites = []
+        for link in soup.find_all('a'):
+            email_link = link.get('href')
+            if email_link != None:
+                if 'http' in email_link:
+                    websites.append(email_link)
+            else:
+                pass
+        """Pretty print output as below"""
+        print ('') 
+        print ('Websites Found - '+str(len(websites)))
+        print ('-------------------------------')
+        for web in websites:
+            print (web)
+        return ''
+    else:
+        raise('Method parameter is not of instance bs4.BeautifulSoup')
diff --git a/tests/test_getemails.py b/tests/test_getemails.py
index 8b575b08..019427ae 100644
--- a/tests/test_getemails.py
+++ b/tests/test_getemails.py
@@ -13,13 +13,11 @@ class getMailsTestCase(unittest.TestCase):
 	def setUp(self):
 		self.held, sys.stdout = sys.stdout, StringIO()
 	
-	def test_return_emails_list(self):
+	def test_print_emails(self):
 		data = "\nMails Found - 1\n-------------------------------\nadvertise@provaz.eu\n"
 		getemails.getMails(soup)
 		self.assertEqual(sys.stdout.getvalue(),data)
 
-	def tearDown(self):
-                sys.stdout.flush()
 
 if __name__ == '__main__':
 	unittest.main()
diff --git a/tests/test_getweblinks.py b/tests/test_getweblinks.py
new file mode 100644
index 00000000..da133b23
--- /dev/null
+++ b/tests/test_getweblinks.py
@@ -0,0 +1,24 @@
+import sys
+import os
+import unittest
+from io import StringIO
+sys.path.append(os.path.abspath('../modules'))
+import getweblinks
+import pagereader
+
+soup = pagereader.readPage('http://www.whatsmyip.net/')
+
+class getLinksTestCase(unittest.TestCase):
+	
+	def setUp(self):
+		self.held, sys.stdout = sys.stdout, StringIO()
+	
+	def test_print_links(self):
+		data = "\nWebsites Found - 7\n-------------------------------\nhttp://ads.wsrs.net/www/delivery/ck.php?n=MyIP856a6b4\nhttp://ads.wsrs.net/www/delivery/ck.php?n=MyIPbf5d683\nhttp://aff.ironsocket.com/SH7L\nhttp://aff.ironsocket.com/SH7L\nhttp://ads.wsrs.net/www/delivery/ck.php?n=MyIPdb5f512\nhttp://wsrs.net/\nhttp://cmsgear.com/\n"
+
+		getweblinks.getLinks(soup)
+		self.assertEqual(sys.stdout.getvalue(),data)
+
+
+if __name__ == '__main__':
+	unittest.main()

From 3e37eeb72def7e5d6d9cce0baf109748465a8b29 Mon Sep 17 00:00:00 2001
From: Pravin Agre <agrepravin@gmail.com>
Date: Sun, 18 Jun 2017 22:38:04 +0530
Subject: [PATCH 3/3] travis

---
 .travis.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.travis.yml b/.travis.yml
index 53853297..aa820c61 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -11,3 +11,4 @@ install:
   - pip3 install bs4
 script: 
   - python3 test_getemails.py
+  - python3 test_getweblinks.py