Using user agent file as default - fake-useragent package seems curre…

…ntly off (#54)
pgaref · Oct 19, 2019 · dd2cf26 · dd2cf26
1 parent 5322364
commit dd2cf26
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/http_request_randomizer/requests/proxy/requestProxy.py b/http_request_randomizer/requests/proxy/requestProxy.py
@@ -34,7 +34,7 @@ def __init__(self, web_proxy_list=[], sustain=False, timeout=5, protocol=Protoco
         self.logger = logging.getLogger()
         self.logger.addHandler(handler)
         self.logger.setLevel(0)
-        self.userAgent = UserAgentManager()
+        self.userAgent = UserAgentManager(file=os.path.join(os.path.dirname(__file__), '../data/user_agents.txt'))
 
         #####
         # Each of the classes below implements a specific URL Parser

diff --git a/http_request_randomizer/requests/useragent/userAgent.py b/http_request_randomizer/requests/useragent/userAgent.py
@@ -5,6 +5,7 @@
 
 logger = logging.getLogger(__name__)
 
+
 class UserAgentManager:
     def __init__(self, fallback=None, file=None):
         self.agent_file = file

diff --git a/tests/test_useragent.py b/tests/test_useragent.py
@@ -32,8 +32,11 @@ def test_last_user_agent(self):
         self.assertIsNone(self.uafake.get_last_user_agent())
 
     def test_random_user_agent(self):
-        self.assertNotEqual(self.uafile.get_random_user_agent(), self.uafile.get_random_user_agent())
-        self.assertNotEqual(self.uafake.get_random_user_agent(), self.uafake.get_random_user_agent())
+        count = 0
+        for i in range(1, 101):
+            if self.uafile.get_random_user_agent() == self.uafile.get_random_user_agent():
+                count = count + 1
+        self.assertNotEqual(count, i)
 
 
 if __name__ == '__main__':