def test_deinitialize(self) -> None:
"""Tests that deinitialize deinitializes Supervisor."""
config = TestConfiguration.load_config("sample_config.yml")
launcher = Launcher(config)
# Create mock OK response.
response = Response()
response.status_code = 200
# Setup init mocks.
launcher._supervisor.initialize = MagicMock(
return_value=None) # type: ignore
for client in launcher.clients:
client.public_api.stats = MagicMock(return_value=response)
# Initialize launcher.
launcher.initialize()
# Setup deinit mock.
launcher._supervisor.deinitialize = MagicMock(
return_value=None) # type: ignore
# Deinitialize
launcher.deinitialize()
launcher._supervisor.deinitialize.assert_called() # type: ignore
def test_initialize(self) -> None:
"""Tests that on initialize launcher initializes Supervisor and verifies clients."""
config = TestConfiguration.load_config("sample_config.yml")
launcher = Launcher(config)
# Create mock OK response.
response = Response()
response.status_code = 200
# Setup mocks.
launcher._supervisor.initialize = MagicMock(
return_value=None) # type: ignore
for client in launcher.clients:
client.public_api.stats = MagicMock(return_value=response)
# Initialize launcher.
launcher.initialize()
# Check that expected methods are called
launcher._supervisor.initialize.assert_called() # type: ignore
for client in launcher.clients:
client.public_api.stats.assert_called()
