def test_nl_shutdown(self, mock_join): mock_config = make_mock_config(8082, "HTTPPlugin", "HTTPPlugin") mock_socket = Mock() mock_socket.shutdown = Mock() mock_socket.detach = Mock() mock_socket.close = Mock() listener = NetworkListener("", mock_config, None) listener._session_socket = mock_socket listener._running = True with patch("platform.system", return_value="Darwin"): listener.shutdown() self.assertFalse(listener._running) self.assertTrue(mock_socket.close.called) self.assertTrue(mock_join.called) listener.running = True with patch("platform.system", return_value="Linux"): listener.shutdown() self.assertFalse(listener._running) self.assertTrue(mock_socket.shutdown.called) self.assertEqual(2, mock_join.call_count) listener._session_socket = None listener.running = True with patch("platform.system", return_value="Linux") as mock_system: listener.shutdown() self.assertFalse(listener._running) self.assertFalse(mock_system.called) self.assertEqual(3, mock_join.call_count)
def test_plugins_enabled(self, mock_start_listening): mock_config = make_mock_config(8082, "HTTPPlugin", "HTTPPlugin") listener = NetworkListener("", mock_config, None) listener.start() while listener.connection_count == 0: pass listener._session_socket = None listener.shutdown() self.assertTrue(mock_start_listening.called)