forked from rdegges/python-amazonify
-
Notifications
You must be signed in to change notification settings - Fork 1
/
tests.py
53 lines (39 loc) · 3.24 KB
/
tests.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
from unittest import TestCase
from urllib.parse import urlparse
from amazonify import amazonify
class Amazonify(TestCase):
def test_requires_url(self):
self.assertRaises(TypeError, amazonify)
def test_requires_affiliate_tag(self):
self.assertRaises(TypeError, amazonify, 'http://www.amazon.com/PostgreSQL-High-Performance-Gregory-Smith/dp/184951030X/ref=trdrt_tipp_dp_img_GWTB_507846?pf_rd_p=1367759962&pf_rd_s=right-4&pf_rd_t=101&pf_rd_i=507846&pf_rd_m=ATVPDKIKX0DER&pf_rd_r=1216X6HJC7KEWY0X3VD7')
def test_returns_string(self):
self.assertIsInstance(amazonify('http://www.amazon.com/PostgreSQL-High-Performance-Gregory-Smith/dp/184951030X/ref=trdrt_tipp_dp_img_GWTB_507846?pf_rd_p=1367759962&pf_rd_s=right-4&pf_rd_t=101&pf_rd_i=507846&pf_rd_m=ATVPDKIKX0DER&pf_rd_r=1216X6HJC7KEWY0X3VD7', 'rdegges-20'), str)
def test_returns_none_for_invalid_urls(self):
self.assertFalse(amazonify('', 'rdegges-20'))
self.assertFalse(amazonify('http://?hi=there', 'rdegges-20'))
self.assertFalse(amazonify('?hi=there', 'rdegges-20'))
self.assertFalse(amazonify('/yo/yo/', 'rdegges-20'))
def test_adds_affiliate_tag_as_a_querystring(self):
# Test it on a URL that already has a querystring:
test_url = amazonify('http://www.amazon.com/PostgreSQL-High-Performance-Gregory-Smith/dp/184951030X/ref=trdrt_tipp_dp_img_GWTB_507846?pf_rd_p=1367759962&pf_rd_s=right-4&pf_rd_t=101&pf_rd_i=507846&pf_rd_m=ATVPDKIKX0DER&pf_rd_r=1216X6HJC7KEWY0X3VD7', 'rdegges-20')
self.assertNotEqual(-1, urlparse(test_url).query.find('tag=rdegges-20'))
# Test it on a URL that has no querystring:
test_url = amazonify('http://www.amazon.com/PostgreSQL-High-Performance-Gregory-Smith/dp/184951030X/ref=trdrt_tipp_dp_img_GWTB_507846', 'rdegges-20')
self.assertEqual(urlparse(test_url).query, 'tag=rdegges-20')
self.assertTrue(test_url.endswith('tag=rdegges-20'))
def test_doesnt_change_scheme(self):
test_url = amazonify('http://www.amazon.com/PostgreSQL-High-Performance-Gregory-Smith/dp/184951030X/ref=trdrt_tipp_dp_img_GWTB_507846', 'rdegges-20')
self.assertEqual(urlparse(test_url).scheme, 'http')
test_url = amazonify('https://www.amazon.com/PostgreSQL-High-Performance-Gregory-Smith/dp/184951030X/ref=trdrt_tipp_dp_img_GWTB_507846', 'rdegges-20')
self.assertEqual(urlparse(test_url).scheme, 'https')
def test_doesnt_change_netloc(self):
test_url = amazonify('http://www.amazon.com/PostgreSQL-High-Performance-Gregory-Smith/dp/184951030X/ref=trdrt_tipp_dp_img_GWTB_507846', 'rdegges-20')
self.assertEqual(urlparse(test_url).netloc, 'www.amazon.com')
test_url = amazonify('http://www.amazon.co.uk/PostgreSQL-High-Performance-Gregory-Smith/dp/184951030X/ref=trdrt_tipp_dp_img_GWTB_507846', 'rdegges-20')
self.assertEqual(urlparse(test_url).netloc, 'www.amazon.co.uk')
def test_promo_url_has_query_params(self):
# test a URL that contains mandatory query params
test_url = amazonify('http://www.amazon.co.uk/gp/feature.html?docId=1000577623&ie=UTF8', 'rdegges-20')
qs = urlparse(test_url).query
self.assertNotEqual(-1, qs.find('tag=rdegges-20'))
self.assertNotEqual(-1, qs.find('docId=1000577623'))