2 | 
3 | pyshorteners
4 |
5 |
6 | 7 | 8 |
9 | 
10 | 
11 | 
12 |
\r\n\t\t
\r\n \r\n"""
14 | mock_url = osdb.api_url
15 |
16 |
17 | @responses.activate
18 | def test_tinyurl_short_method():
19 | # mock responses
20 | responses.add(responses.POST, mock_url, body=body)
21 |
22 | shorten_result = osdb.short(expanded)
23 | assert shorten_result == shorten
24 |
25 |
26 | @responses.activate
27 | def test_tinyurl_short_bad_response():
28 | # mock responses
29 | responses.add(responses.POST, mock_url, body=body, status=400)
30 |
31 | with pytest.raises(ShorteningErrorException):
32 | osdb.short(expanded)
33 |
--------------------------------------------------------------------------------
/tests/test_owly.py:
--------------------------------------------------------------------------------
1 | import json
2 | from urllib.parse import urlencode
3 |
4 | from pyshorteners import Shortener
5 | from pyshorteners.exceptions import ShorteningErrorException, ExpandingErrorException
6 |
7 | import responses
8 | import pytest
9 |
10 | s = Shortener(api_key="TEST_KEY")
11 | shorten = "http://ow.ly/test"
12 | expanded = "http://www.test.com"
13 | owly = s.owly
14 |
15 |
16 | @responses.activate
17 | def test_owly_short_method():
18 | # mock responses
19 | params = urlencode({"apiKey": "TEST_KEY", "longUrl": expanded})
20 | body = json.dumps({"results": {"shortUrl": shorten}})
21 | mock_url = f"{owly.api_url}shorten?{params}"
22 | responses.add(responses.GET, mock_url, body=body, match_querystring=True)
23 |
24 | shorten_result = owly.short(expanded)
25 |
26 | assert shorten_result == shorten
27 |
28 |
29 | @responses.activate
30 | def test_owly_short_method_bad_response():
31 | # mock responses
32 | params = urlencode({"apiKey": "TEST_KEY", "longUrl": expanded})
33 | body = "{'rerrsults': {'shortUrl': shorten}}"
34 | mock_url = f"{owly.api_url}shorten?{params}"
35 | responses.add(responses.GET, mock_url, body=body, match_querystring=True)
36 |
37 | with pytest.raises(json.decoder.JSONDecodeError):
38 | owly.short(expanded)
39 |
40 |
41 | @responses.activate
42 | def test_owly_short_method_bad_response_status():
43 | # mock responses
44 | params = urlencode({"apiKey": "TEST_KEY", "longUrl": expanded})
45 | body = "{'rerrsults': {'shortUrl': shorten}}"
46 | mock_url = f"{owly.api_url}shorten?{params}"
47 | responses.add(
48 | responses.GET, mock_url, body=body, status=400, match_querystring=True
49 | )
50 |
51 | with pytest.raises(ShorteningErrorException):
52 | owly.short(expanded)
53 |
54 |
55 | @responses.activate
56 | def test_owly_expand_method():
57 | # mock responses
58 | params = urlencode({"apiKey": "TEST_KEY", "shortUrl": shorten})
59 | body = json.dumps({"results": {"longUrl": expanded}})
60 | mock_url = f"{owly.api_url}expand?{params}"
61 | responses.add(responses.GET, mock_url, body=body, match_querystring=True)
62 |
63 | expanded_result = owly.expand(shorten)
64 |
65 | assert expanded_result == expanded
66 |
67 |
68 | @responses.activate
69 | def test_owly_expand_method_bad_response():
70 | # mock responses
71 | params = urlencode({"apiKey": "TEST_KEY", "shortUrl": shorten})
72 | body = "{'results': {'longUrl': ''}}"
73 | mock_url = f"{owly.api_url}expand?{params}"
74 | responses.add(responses.GET, mock_url, body=body, match_querystring=True)
75 |
76 | with pytest.raises(json.decoder.JSONDecodeError):
77 | owly.expand(shorten)
78 |
79 |
80 | @responses.activate
81 | def test_owly_expand_method_bad_status_code():
82 | # mock responses
83 | params = urlencode({"apiKey": "TEST_KEY", "shortUrl": shorten})
84 | body = "{'results': {'longUrl': ''}}"
85 | mock_url = f"{owly.api_url}expand?{params}"
86 | responses.add(
87 | responses.GET, mock_url, body=body, status=400, match_querystring=True
88 | )
89 |
90 | with pytest.raises(ExpandingErrorException):
91 | owly.expand(shorten)
92 |
--------------------------------------------------------------------------------
/tests/test_post.py:
--------------------------------------------------------------------------------
1 | from urllib.parse import urlencode
2 |
3 | from pyshorteners import Shortener
4 | from pyshorteners.exceptions import ShorteningErrorException
5 |
6 | import responses
7 | import pytest
8 |
9 | s = Shortener(api_key="TEST")
10 | post = s.post
11 | shorten = "http://po.st/test"
12 | expanded = "http://www.test.com"
13 | params = urlencode({"apiKey": "TEST", "longUrl": expanded, "format": "txt"})
14 | mock_url = f"{post.api_url}?{params}"
15 |
16 |
17 | @responses.activate
18 | def test_post_short_method():
19 | # mock responses
20 | response = {"short_url": shorten}
21 | responses.add(responses.GET, mock_url, json=response, match_querystring=True)
22 |
23 | shorten_result = post.short(expanded)
24 | assert shorten_result == shorten
25 |
26 |
27 | @responses.activate
28 | def test_post_short_bad_response():
29 | # mock responses
30 | responses.add(
31 | responses.GET, mock_url, body=shorten, status=400, match_querystring=True
32 | )
33 |
34 | with pytest.raises(ShorteningErrorException):
35 | post.short(expanded)
36 |
--------------------------------------------------------------------------------
/tests/test_qpsru.py:
--------------------------------------------------------------------------------
1 | from urllib.parse import urlencode
2 |
3 | from pyshorteners import Shortener
4 | from pyshorteners.exceptions import ShorteningErrorException
5 |
6 | import responses
7 | import pytest
8 |
9 | s = Shortener()
10 | shorten = "http://qps.ru/test"
11 | expanded = "http://www.test.com"
12 | qp = s.qpsru
13 |
14 |
15 | @responses.activate
16 | def test_qpsru_short_method():
17 | # mock responses
18 | params = urlencode({"url": expanded})
19 | mock_url = f"{qp.api_url}?{params}"
20 | responses.add(responses.GET, mock_url, body=shorten, match_querystring=True)
21 |
22 | shorten_result = qp.short(expanded)
23 |
24 | assert shorten_result == shorten
25 |
26 |
27 | @responses.activate
28 | def test_qpsru_short_method_bad_response():
29 | # mock responses
30 | params = urlencode({"url": expanded})
31 | mock_url = f"{qp.api_url}?{params}"
32 | responses.add(
33 | responses.GET, mock_url, body=shorten, status=400, match_querystring=True
34 | )
35 |
36 | with pytest.raises(ShorteningErrorException):
37 | qp.short(expanded)
38 |
--------------------------------------------------------------------------------
/tests/test_shortcm.py:
--------------------------------------------------------------------------------
1 | #!/usr/bin/env python
2 | # encoding: utf-8
3 |
4 | import responses
5 | import pytest
6 |
7 | from pyshorteners import Shortener
8 | from pyshorteners.exceptions import ShorteningErrorException, ExpandingErrorException
9 |
10 |
11 | token = "TEST_TOKEN"
12 | domain = "short.cm"
13 | path = "test"
14 | expanded = "http://www.test.com"
15 | shorten = f"http://{domain}/{path}"
16 | s = Shortener(api_key=token, domain=domain)
17 | shortcm = s.shortcm
18 |
19 |
20 | @responses.activate
21 | def test_shortcm_short_method():
22 | # mock responses
23 | json = {"path": path, "originalURL": expanded, "shortURL": shorten}
24 |
25 | responses.add(responses.POST, shortcm.api_url, json=json)
26 |
27 | shorten_result = shortcm.short(expanded)
28 |
29 | assert shorten_result == shorten
30 |
31 |
32 | @responses.activate
33 | def test_shortcm_short_method_bad_response():
34 | # mock responses
35 | json = {"error": "Test error", "success": False}
36 | responses.add(responses.POST, shortcm.api_url, json=json, status=400)
37 |
38 | with pytest.raises(ShorteningErrorException):
39 | shortcm.short(expanded)
40 |
41 |
42 | @responses.activate
43 | def test_shortcm_expand_method():
44 | # mock responses
45 | json = {"path": path, "shortURL": shorten, "originalURL": expanded}
46 | url = f"{shortcm.api_url}expand"
47 | responses.add(responses.GET, url, json=json)
48 | assert shortcm.expand(shorten) == expanded
49 |
50 |
51 | @responses.activate
52 | def test_shortcm_expand_method_bad_response():
53 | # mock responses
54 | json = {"error": "Test error"}
55 | url = f"{shortcm.api_url}expand"
56 | responses.add(responses.GET, url, json=json, status=400)
57 |
58 | with pytest.raises(ExpandingErrorException):
59 | shortcm.expand(shorten)
60 |
--------------------------------------------------------------------------------
/tests/test_tinycc.py:
--------------------------------------------------------------------------------
1 | from urllib.parse import urlencode
2 |
3 | from pyshorteners import Shortener
4 | from pyshorteners.exceptions import ShorteningErrorException, ExpandingErrorException
5 |
6 | import responses
7 | import pytest
8 |
9 | token = "TEST_TOKEN"
10 | login = "TEST_LOGIN"
11 | api_version = "2.0.3"
12 | shorten = "http://tiny.cc/test"
13 | expanded = "http://www.test.com"
14 | s = Shortener(api_key=token, login=login)
15 | tiny = s.tinycc
16 |
17 |
18 | @responses.activate
19 | def test_tinycc_short_method():
20 | # mock responses
21 | body = {"results": {"short_url": shorten}}
22 | params = urlencode(
23 | dict(
24 | longUrl=expanded,
25 | apiKey=token,
26 | format="json",
27 | c="rest_api",
28 | version=api_version,
29 | login=login,
30 | m="shorten",
31 | )
32 | )
33 |
34 | url = f"{tiny.api_url}?{params}"
35 | responses.add(responses.GET, url, json=body, match_querystring=True)
36 |
37 | shorten_result = tiny.short(expanded)
38 |
39 | assert shorten_result == shorten
40 |
41 |
42 | @responses.activate
43 | def test_tinycc_short_method_bad_response():
44 | # mock responses
45 | body = shorten
46 | params = urlencode(
47 | dict(
48 | longUrl=expanded,
49 | apiKey=token,
50 | format="json",
51 | c="rest_api",
52 | version=api_version,
53 | login=login,
54 | m="shorten",
55 | )
56 | )
57 | url = f"{tiny.api_url}?{params}"
58 | responses.add(responses.GET, url, body=body, status=400, match_querystring=True)
59 |
60 | with pytest.raises(ShorteningErrorException):
61 | tiny.short(expanded)
62 |
63 |
64 | @responses.activate
65 | def test_tinycc_expand_method():
66 | # mock responses
67 | body = {"results": {"longUrl": expanded}}
68 | params = urlencode(
69 | dict(
70 | longUrl=shorten,
71 | apiKey=token,
72 | format="json",
73 | c="rest_api",
74 | version=api_version,
75 | login=login,
76 | m="expand",
77 | )
78 | )
79 | url = f"{tiny.api_url}?{params}"
80 | responses.add(responses.GET, url, json=body, match_querystring=True)
81 | assert tiny.expand(shorten) == expanded
82 |
83 |
84 | @responses.activate
85 | def test_tinycc_expand_method_bad_response():
86 | # mock responses
87 | body = expanded
88 | params = urlencode(
89 | dict(
90 | longUrl=shorten,
91 | apiKey=token,
92 | format="json",
93 | c="rest_api",
94 | version=api_version,
95 | login=login,
96 | m="expand",
97 | )
98 | )
99 | url = f"{tiny.api_url}?{params}"
100 | responses.add(responses.GET, url, body=body, status=400, match_querystring=True)
101 |
102 | with pytest.raises(ExpandingErrorException):
103 | tiny.expand(shorten)
104 |
105 |
106 | @responses.activate
107 | def test_tinycc_total_clicks():
108 | body = {"results": {"clicks": 20}}
109 | params = urlencode(
110 | dict(
111 | shortUrl=shorten,
112 | apiKey=token,
113 | format="json",
114 | c="rest_api",
115 | version=api_version,
116 | login=login,
117 | m="total_visits",
118 | )
119 | )
120 | url = f"{tiny.api_url}?{params}"
121 | responses.add(responses.GET, url, json=body, match_querystring=True)
122 | assert tiny.total_clicks(shorten) == 20
123 |
124 |
125 | @responses.activate
126 | def test_tinycc_total_clicks_bad_response():
127 | clicks_body = {"results": "a"}
128 | params = urlencode(
129 | dict(
130 | c="rest_api",
131 | version=api_version,
132 | format="json",
133 | apiKey=token,
134 | login=login,
135 | m="total_visits",
136 | shortUrl=shorten,
137 | )
138 | )
139 | url = f"{tiny.api_url}?{params}"
140 | responses.add(responses.GET, url, json=clicks_body, match_querystring=True)
141 | assert tiny.total_clicks(shorten) == 0
142 |
--------------------------------------------------------------------------------
/tests/test_tinyurl.py:
--------------------------------------------------------------------------------
1 | from pyshorteners import Shortener
2 | from pyshorteners.exceptions import ShorteningErrorException
3 |
4 | import responses
5 | import pytest
6 |
7 | s = Shortener()
8 | shorten = "http://tinyurl.com/test"
9 | expanded = "http://www.test.com"
10 | tiny = s.tinyurl
11 |
12 |
13 | @responses.activate
14 | def test_tinyurl_short_method():
15 | # mock responses
16 | mock_url = f"{tiny.api_url}?url={expanded}"
17 | responses.add(responses.GET, mock_url, body=shorten, match_querystring=True)
18 |
19 | shorten_result = tiny.short(expanded)
20 | assert shorten_result == shorten
21 |
22 |
23 | @responses.activate
24 | def test_tinyurl_short_bad_response():
25 | # mock responses
26 | mock_url = f"{tiny.api_url}?url={expanded}"
27 | responses.add(
28 | responses.GET, mock_url, body=shorten, status=400, match_querystring=True
29 | )
30 |
31 | with pytest.raises(ShorteningErrorException):
32 | tiny.short(expanded)
33 |
--------------------------------------------------------------------------------
\r\n\t\t\t
\r\n\t