Benjamin_Loison/waybackpy
1
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

make mypy happy

Browse Source
This commit is contained in:
Akash Mahanty
2022-02-08 17:17:43 +05:30
parent 2489e97868
commit 1803803112
1 changed files with 20 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
tests/test_cli.py
View File

@@ -1,10 +1,11 @@
import requests import requests
from click.testing import CliRunner from click.testing import CliRunner
from waybackpy.cli import __version__, main from waybackpy.cli import main
from waybackpy import __version__
def test_oldest(): def test_oldest() -> None:
runner = CliRunner() runner = CliRunner()
result = runner.invoke(main, ["--url", " https://github.com ", "--oldest"]) result = runner.invoke(main, ["--url", " https://github.com ", "--oldest"])
assert result.exit_code == 0 assert result.exit_code == 0
@@ -15,7 +16,7 @@ def test_oldest():
) )
def test_near(): def test_near() -> None:
runner = CliRunner() runner = CliRunner()
result = runner.invoke( result = runner.invoke(
main, main,
@@ -24,13 +25,13 @@ def test_near():
" https://facebook.com ", " https://facebook.com ",
"--near", "--near",
"--year", "--year",
2010, "2010",
"--month", "--month",
5, "5",
"--day", "--day",
10, "10",
"--hour", "--hour",
6, "6",
], ],
) )
assert result.exit_code == 0 assert result.exit_code == 0
@@ -41,7 +42,7 @@ def test_near():
) )
def test_json(): def test_json() -> None:
runner = CliRunner() runner = CliRunner()
result = runner.invoke( result = runner.invoke(
main, main,
@@ -50,13 +51,13 @@ def test_json():
" https://apple.com ", " https://apple.com ",
"--near", "--near",
"--year", "--year",
2010, "2010",
"--month", "--month",
2, "2",
"--day", "--day",
8, "8",
"--hour", "--hour",
12, "12",
"--json", "--json",
], ],
) )
@@ -74,7 +75,7 @@ amp": "20100208125854"}}, "timestamp":"""
) )
def test_newest(): def test_newest() -> None:
runner = CliRunner() runner = CliRunner()
result = runner.invoke(main, ["--url", " https://microsoft.com ", "--newest"]) result = runner.invoke(main, ["--url", " https://microsoft.com ", "--newest"])
assert result.exit_code == 0 assert result.exit_code == 0
@@ -84,7 +85,7 @@ def test_newest():
) )
def test_cdx(): def test_cdx() -> None:
runner = CliRunner() runner = CliRunner()
result = runner.invoke( result = runner.invoke(
main, main,
@@ -100,7 +101,7 @@ def test_cdx():
assert result.output.count("\n") > 3000 assert result.output.count("\n") > 3000
def test_save(): def test_save() -> None:
runner = CliRunner() runner = CliRunner()
result = runner.invoke( result = runner.invoke(
main, main,
@@ -118,14 +119,14 @@ def test_save():
assert result.output.find("://news.ycombinator.com") != -1 assert result.output.find("://news.ycombinator.com") != -1
def test_version(): def test_version() -> None:
runner = CliRunner() runner = CliRunner()
result = runner.invoke(main, ["--version"]) result = runner.invoke(main, ["--version"])
assert result.exit_code == 0 assert result.exit_code == 0
assert result.output == f"waybackpy version {__version__}\n" assert result.output == f"waybackpy version {__version__}\n"
def test_license(): def test_license() -> None:
runner = CliRunner() runner = CliRunner()
result = runner.invoke(main, ["--license"]) result = runner.invoke(main, ["--license"])
assert result.exit_code == 0 assert result.exit_code == 0
@@ -138,7 +139,7 @@ def test_license():
) )
def test_only_url(): def test_only_url() -> None:
runner = CliRunner() runner = CliRunner()
result = runner.invoke(main, ["--url", "https://google.com"]) result = runner.invoke(main, ["--url", "https://google.com"])
assert result.exit_code == 0 assert result.exit_code == 0
@@ -149,7 +150,7 @@ def test_only_url():
) )
def test_known_url(): def test_known_url() -> None:
# with file generator enabled # with file generator enabled
runner = CliRunner() runner = CliRunner()
result = runner.invoke( result = runner.invoke(