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)
