Return to Mock from Any

Author: staticdev

tests/test_main.py

@@ -1,50 +1,49 @@
 """Test cases for the __main__ module."""
-from typing import Any
 from unittest.mock import Mock
 
+import click.testing
 import pytest
-from click.testing import CliRunner
 from pytest_mock import MockFixture
 
 from toml_validator import __main__
 
 
 @pytest.fixture
-def runner() -> CliRunner:
+def runner() -> click.testing.CliRunner:
     """Fixture for invoking command-line interfaces."""
-    return CliRunner()
+    return click.testing.CliRunner()
 
 
 @pytest.fixture
-def mock_validation_validate_extension(mocker: MockFixture) -> Any:
+def mock_validation_validate_extension(mocker: MockFixture) -> Mock:
     """Fixture for mocking validation.validate_extension."""
     return mocker.patch("toml_validator.validation.validate_extension")
 
 
 @pytest.fixture
-def mock_validation_validate_toml_no_error(mocker: MockFixture) -> Any:
+def mock_validation_validate_toml_no_error(mocker: MockFixture) -> Mock:
     """Fixture for mocking validation.validate_toml with no errors."""
     mock = mocker.patch("toml_validator.validation.validate_toml")
     mock.return_value = ""
     return mock
 
 
 @pytest.fixture
-def mock_validation_validate_toml_with_error(mocker: MockFixture) -> Any:
+def mock_validation_validate_toml_with_error(mocker: MockFixture) -> Mock:
     """Fixture for mocking validation.validate_toml with error."""
     mock = mocker.patch("toml_validator.validation.validate_toml")
     mock.return_value = "|some error description|"
     return mock
 
 
-def test_main_without_argument(runner: CliRunner) -> None:
+def test_main_without_argument(runner: click.testing.CliRunner) -> None:
     """It exits with a status code of 2."""
     result = runner.invoke(__main__.main)
     assert result.exit_code == 2
 
 
 def test_main_with_argument_success(
-    runner: CliRunner,
+    runner: click.testing.CliRunner,
     mock_validation_validate_extension: Mock,
     mock_validation_validate_toml_no_error: Mock,
 ) -> None:
@@ -61,7 +60,7 @@ def test_main_with_argument_success(
 
 
 def test_main_with_argument_fail(
-    runner: CliRunner,
+    runner: click.testing.CliRunner,
     mock_validation_validate_extension: Mock,
     mock_validation_validate_toml_with_error: Mock,
 ) -> None:
@@ -78,7 +77,9 @@ def test_main_with_argument_fail(
 
 
 @pytest.mark.e2e
-def test_main_without_arguments_in_production_env(runner: CliRunner) -> None:
+def test_main_without_arguments_in_production_env(
+    runner: click.testing.CliRunner,
+) -> None:
     """It exits with a status code of 2 (e2e)."""
     result = runner.invoke(__main__.main)
     assert result.exit_code == 2

tests/test_validation.py

@@ -1,5 +1,4 @@
 """Test cases for the validation module."""
-from typing import Any
 from unittest.mock import Mock
 
 import pytest
@@ -10,27 +9,27 @@
 
 
 @pytest.fixture
-def mock_tomlkit_parse(mocker: MockFixture) -> Any:
+def mock_tomlkit_parse(mocker: MockFixture) -> Mock:
     """Fixture for mocking tomlkit.parse."""
     return mocker.patch("tomlkit.parse")
 
 
 @pytest.fixture
-def mock_tomlkit_parse_exception(mocker: MockFixture) -> Any:
+def mock_tomlkit_parse_exception(mocker: MockFixture) -> Mock:
     """Fixture for mocking tomlkit.parse."""
     mock = mocker.patch("tomlkit.parse")
     mock.side_effect = TOMLKitError("|some tomlkit error|")
     return mock
 
 
 @pytest.fixture
-def mock_open_valid_file(mocker: MockFixture) -> Any:
+def mock_open_valid_file(mocker: MockFixture) -> Mock:
     """Fixture for mocking build-in open for valid TOML file."""
     return mocker.patch("builtins.open", mocker.mock_open(read_data="[x]\na = 3"))
 
 
 @pytest.fixture
-def mock_open_invalid_file(mocker: MockFixture) -> Any:
+def mock_open_invalid_file(mocker: MockFixture) -> Mock:
     """Fixture for mocking build-in open for valid TOML file."""
     return mocker.patch(
         "builtins.open", mocker.mock_open(read_data="[x]\na = 3\n[x]\na = 3")