-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest.py
25 lines (20 loc) · 1.24 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
import unittest
from tripadvisor.spiders import tripadvisor
class Test(unittest.TestCase):
def test_review_next_page(self):
url = 'https://www.tripadvisor.cn/Restaurant_Review-g294217-d1028680-Reviews-Aqua_Luna-Hong_Kong.html'
new_url = tripadvisor.review_next_page(url, 12)
expect = 'https://www.tripadvisor.cn/Restaurant_Review-g294217-d1028680-Reviews-or10-Aqua_Luna-Hong_Kong.html'
self.assertEqual(new_url, expect)
url = 'https://www.tripadvisor.cn/Restaurant_Review-g294217-d1028680-Reviews-or10-Aqua_Luna-Hong_Kong.html'
new_url = tripadvisor.review_next_page(url, 12)
self.assertEqual(new_url, None)
url = 'https://www.tripadvisor.cn/Restaurant_Review-g294217-d1028680-Reviews-or20-Aqua_Luna-Hong_Kong.html'
new_url = tripadvisor.review_next_page(url, 30)
self.assertEqual(new_url, None)
url = 'https://www.tripadvisor.cn/Restaurant_Review-g294217-d1028680-Reviews-or20-Aqua_Luna-Hong_Kong.html'
new_url = tripadvisor.review_next_page(url, 35)
expect = 'https://www.tripadvisor.cn/Restaurant_Review-g294217-d1028680-Reviews-or30-Aqua_Luna-Hong_Kong.html'
self.assertEqual(new_url, expect)
if __name__ == '__main__':
unittest.main()