From d1a1cf254680af07139c5bae2001231563a254a3 Mon Sep 17 00:00:00 2001 From: Akash Mahanty Date: Mon, 24 Jan 2022 17:50:36 +0530 Subject: [PATCH] added tests for utils.py at tests/test_utils.py also changed a keyword argument from headers to user_agent for latest_version of utils.py with the usual default vaule. --- tests/test_utils.py | 13 +++++++++++++ waybackpy/utils.py | 3 ++- 2 files changed, 15 insertions(+), 1 deletion(-) create mode 100644 tests/test_utils.py diff --git a/tests/test_utils.py b/tests/test_utils.py new file mode 100644 index 0000000..70259d9 --- /dev/null +++ b/tests/test_utils.py @@ -0,0 +1,13 @@ +from waybackpy.utils import latest_version, DEFAULT_USER_AGENT +from waybackpy.__version__ import __version__ + + +def test_default_user_agent(): + assert ( + DEFAULT_USER_AGENT + == "waybackpy %s - https://github.com/akamhy/waybackpy" % __version__ + ) + + +def test_latest_version(): + assert __version__ == latest_version(package_name="waybackpy") diff --git a/waybackpy/utils.py b/waybackpy/utils.py index 594f608..26cb7cd 100644 --- a/waybackpy/utils.py +++ b/waybackpy/utils.py @@ -4,8 +4,9 @@ from .__version__ import __version__ DEFAULT_USER_AGENT = "waybackpy %s - https://github.com/akamhy/waybackpy" % __version__ -def latest_version(package_name, headers): +def latest_version(package_name, user_agent=DEFAULT_USER_AGENT): request_url = "https://pypi.org/pypi/" + package_name + "/json" + headers = {"User-Agent": user_agent} response = requests.get(request_url, headers=headers) data = response.json() return data["info"]["version"]