mirror of
https://github.com/ivan-usov-org/bec.git
synced 2025-04-22 02:20:02 +02:00
33 lines
1.5 KiB
Python
33 lines
1.5 KiB
Python
from unittest import mock
|
|
|
|
from scan_server.cli.launch import main
|
|
|
|
|
|
def test_main():
|
|
with mock.patch("scan_server.cli.launch.argparse.ArgumentParser") as mock_parser:
|
|
with mock.patch("scan_server.cli.launch.ServiceConfig") as mock_config:
|
|
with mock.patch("scan_server.scan_server.ScanServer") as mock_scan_server:
|
|
with mock.patch("scan_server.cli.launch.threading.Event") as mock_event:
|
|
main()
|
|
mock_parser.assert_called_once()
|
|
mock_config.assert_called_once()
|
|
mock_scan_server.assert_called_once()
|
|
mock_event.assert_called_once()
|
|
|
|
|
|
def test_main_shutdown():
|
|
with mock.patch("scan_server.cli.launch.argparse.ArgumentParser") as mock_parser:
|
|
with mock.patch("scan_server.cli.launch.ServiceConfig") as mock_config:
|
|
with mock.patch("scan_server.scan_server.ScanServer") as mock_scan_server:
|
|
with mock.patch("scan_server.cli.launch.threading.Event") as mock_event:
|
|
mock_event.return_value.wait.side_effect = KeyboardInterrupt()
|
|
try:
|
|
main()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
mock_parser.assert_called_once()
|
|
mock_config.assert_called_once()
|
|
mock_scan_server.assert_called_once()
|
|
mock_event.assert_called_once()
|
|
mock_scan_server.return_value.shutdown.assert_called_once()
|