diff --git a/http_request_randomizer/requests/proxy/requestProxy.py b/http_request_randomizer/requests/proxy/requestProxy.py index 64e871a..7419ba9 100644 --- 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 index e419a83..2a9c253 100644 --- 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 index 75b6a96..cca5a55 100644 --- 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__':