update tests and cli errors

This commit is contained in:
Akash Mahanty
2021-04-02 10:28:08 +05:30
parent 8d2cc112c1
commit 6755d4d015
2 changed files with 3 additions and 1 deletions

View File

@@ -29,7 +29,7 @@ def test_save():
get=None,
)
reply = cli.args_handler(args)
assert "could happen because either your waybackpy" in str(reply)
assert "could happen because either your waybackpy" or "cannot be archived by wayback machine as it is a redirect" in str(reply)
def test_json():

View File

@@ -29,6 +29,8 @@ def _save(obj):
version=__version__, header=header
)
)
if "URL cannot be archived by wayback machine as it is a redirect" in e:
return ("URL cannot be archived by wayback machine as it is a redirect")
raise WaybackError(err)