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()