forked from lorien/user_agent
/
test.py
executable file
·90 lines (70 loc) · 3.37 KB
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#!/usr/bin/env python
from unittest import TestCase
import unittest
import re
from user_agent import (generate_user_agent, generate_navigator,
UserAgentRuntimeError, UserAgentInvalidRequirements)
class UserAgentTestCase(TestCase):
def test_it(self):
ua = generate_user_agent()
self.assertTrue(len(ua) > 0)
def test_platform_option(self):
for x in range(100):
ua = generate_user_agent(platform='linux')
self.assertTrue('linux' in ua.lower())
ua = generate_user_agent(platform='win')
self.assertTrue('windows' in ua.lower())
ua = generate_user_agent(platform='mac')
self.assertTrue('mac' in ua.lower())
self.assertRaises(UserAgentRuntimeError,
generate_user_agent,
platform=11)
def test_navigator_option(self):
for x in range(100):
ua = generate_user_agent(navigator='firefox')
self.assertTrue('firefox' in ua.lower())
ua = generate_user_agent(navigator='chrome')
self.assertTrue('chrome' in ua.lower())
ua = generate_user_agent(navigator='ie')
self.assertTrue('msie' in ua.lower() or 'rv:11' in ua.lower())
def test_navigator_option_tuple(self):
for x in range(100):
ua = generate_user_agent(navigator=('chrome',))
ua = generate_user_agent(navigator=('chrome', 'firefox'))
ua = generate_user_agent(navigator=('chrome', 'firefox', 'ie'))
def test_platform_option_tuple(self):
for x in range(100):
ua = generate_user_agent(platform=('win', 'linux'))
ua = generate_user_agent(platform=('win', 'linux', 'mac'))
ua = generate_user_agent(platform=('win',))
ua = generate_user_agent(platform=('linux',))
ua = generate_user_agent(platform=('mac',))
def test_platform_navigator_option(self):
for x in range(100):
ua = generate_user_agent(platform='win', navigator='firefox')
self.assertTrue('firefox' in ua.lower())
self.assertTrue('windows' in ua.lower())
ua = generate_user_agent(platform='win', navigator='chrome')
self.assertTrue('chrome' in ua.lower())
self.assertTrue('windows' in ua.lower())
ua = generate_user_agent(platform='win', navigator='ie')
self.assertTrue('msie' in ua.lower() or 'rv:11' in ua.lower())
self.assertTrue('windows' in ua.lower())
def test_platform_linux(self):
for x in range(100):
ua = generate_user_agent(platform='linux')
self.assertTrue(ua.startswith('Mozilla/5.0 (X11;'))
def test_mac_chrome(self):
for x in range(100):
ua = generate_user_agent(platform='mac', navigator='chrome')
self.assertTrue(re.search(r'OS X \d+_\d+(_\d+\b|\b)', ua))
def test_impossible_combination(self):
for x in range(100):
self.assertRaises(UserAgentInvalidRequirements,
generate_user_agent,
platform='linux', navigator='ie')
self.assertRaises(UserAgentInvalidRequirements,
generate_user_agent,
platform='mac', navigator='ie')
if __name__ == '__main__':
unittest.main()