From 6b3b2e2a7de2f04f406ada5f0ffd2a72b5bdf0ea Mon Sep 17 00:00:00 2001 From: Akash Mahanty Date: Sat, 3 Oct 2020 09:33:50 +0530 Subject: [PATCH] tests for newly added known_urls feature --- tests/test_cli.py | 8 +++++++- tests/test_wrapper.py | 8 ++++++++ 2 files changed, 15 insertions(+), 1 deletion(-) diff --git a/tests/test_cli.py b/tests/test_cli.py index d2fa4ce..91b722d 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -43,6 +43,13 @@ def test_total_archives(): reply = cli.args_handler(args) assert isinstance(reply, int) +def test_known_urls(): + args = argparse.Namespace(user_agent="Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \ + (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9", url="https://akamhy.github.io", total=False, version=False, + oldest=False, save=False, newest=False, near=False, alive=True, subdomain=True, known_urls=True, get=None) + reply = cli.args_handler(args) + assert "github" in reply + def test_near(): args = argparse.Namespace(user_agent="Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \ (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9", url="https://pypi.org/user/akamhy/", total=False, version=False, @@ -50,7 +57,6 @@ def test_near(): reply = cli.args_handler(args) assert "202007" in reply - def test_get(): args = argparse.Namespace(user_agent="Mozilla/5.0 (Macintosh; Intel Mac OS X 10_10_5) AppleWebKit/600.8.9 \ (KHTML, like Gecko) Version/8.0.8 Safari/600.8.9", url="https://pypi.org/user/akamhy/", total=False, version=False, diff --git a/tests/test_wrapper.py b/tests/test_wrapper.py index 6c7eeba..ef00378 100644 --- a/tests/test_wrapper.py +++ b/tests/test_wrapper.py @@ -188,3 +188,11 @@ def test_total_archives(): " https://gaha.e4i3n.m5iai3kip6ied.cima/gahh2718gs/ahkst63t7gad8 ", user_agent ) assert target.total_archives() == 0 + +def test_known_urls(): + + target = waybackpy.Url("akamhy.github.io", user_agent) + assert len(target.known_urls(alive=True, subdomain=True)) > 2 + + target = waybackpy.Url("akamhy.github.io", user_agent) + assert len(target.known_urls()) > 3 \ No newline at end of file