mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-07-04 16:03:03 +02:00
30 lines
1007 B
Python
30 lines
1007 B
Python
|
"""Tests for brother_printer.__main__ module."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import importlib
|
||
|
|
import types
|
||
|
|
from unittest.mock import MagicMock, patch
|
||
|
|
|
||
|
|
|
||
|
|
class TestMain:
|
||
|
|
def test_main_called(self) -> None:
|
||
|
|
"""Test that __main__ calls main()."""
|
||
|
|
mock_main = MagicMock()
|
||
|
|
# Create a fake brother_printer.check_brother_printer module
|
||
|
|
fake_module = types.ModuleType("brother_printer.check_brother_printer")
|
||
|
|
vars(fake_module)["main"] = mock_main
|
||
|
|
with patch.dict(
|
||
|
|
"sys.modules",
|
||
|
|
{
|
||
|
|
"brother_printer": types.ModuleType("brother_printer"),
|
||
|
|
"brother_printer.check_brother_printer": fake_module,
|
||
|
|
},
|
||
|
|
):
|
||
|
|
# Remove cached __main__ module so it gets re-imported
|
||
|
|
import sys
|
||
|
|
|
||
|
|
sys.modules.pop("python_pkg.brother_printer.__main__", None)
|
||
|
|
importlib.import_module("python_pkg.brother_printer.__main__")
|
||
|
|
mock_main.assert_called_once()
|