def start_client(proto, server): manager = tendril.get_manager(proto) manager.start() tend = manager.connect(server, EchoApplicationClient) return manager, tend.application
def test_connect_rejected(self, mock_UDPTendril, mock_track_tendril, mock_connect): acceptor = mock.Mock(side_effect=application.RejectConnection()) manager = udp.UDPTendrilManager() tend = manager.connect(('127.0.0.1', 8080), acceptor) mock_connect.assert_called_once_with(('127.0.0.1', 8080), acceptor, None) mock_UDPTendril.assert_called_once_with(manager, ('0.0.0.0', 8880), ('127.0.0.1', 8080)) acceptor.assert_called_once_with(mock_UDPTendril.return_value) self.assertFalse(mock_track_tendril.called) self.assertEqual(tend, None)
def test_connect(self, mock_UDPTendril, mock_track_tendril, mock_connect): acceptor = mock.Mock() manager = udp.UDPTendrilManager() tend = manager.connect(('127.0.0.1', 8080), acceptor) mock_connect.assert_called_once_with(('127.0.0.1', 8080), acceptor, None) mock_UDPTendril.assert_called_once_with(manager, ('0.0.0.0', 8880), ('127.0.0.1', 8080)) acceptor.assert_called_once_with(mock_UDPTendril.return_value) mock_track_tendril.assert_called_once_with( mock_UDPTendril.return_value) self.assertEqual(id(tend), id(mock_UDPTendril.return_value))
def test_connect_failure(self, mock_TCPTendril, mock_track_tendril, mock_connect, mock_socket): acceptor = mock.Mock(side_effect=TestException()) manager = tcp.TCPTendrilManager() with self.assertRaises(TestException): tend = manager.connect(('127.0.0.1', 8080), acceptor) mock_connect.assert_called_once_with(('127.0.0.1', 8080), acceptor, None) mock_socket.assert_called_once_with(socket.AF_INET, socket.SOCK_STREAM) mock_socket.return_value.assert_has_calls([ mock.call.bind(('', 0)), mock.call.connect(('127.0.0.1', 8080)), mock.call.close(), ]) mock_TCPTendril.assert_called_once_with(manager, mock_socket.return_value) acceptor.assert_called_once_with(mock_TCPTendril.return_value) self.assertFalse(mock_track_tendril.called) self.assertFalse(mock_TCPTendril.return_value._start.called)
def test_connect_basic(self, mock_TCPTendril, mock_track_tendril, mock_connect, mock_socket): acceptor = mock.Mock() manager = tcp.TCPTendrilManager() tend = manager.connect(('127.0.0.1', 8080), acceptor) mock_connect.assert_called_once_with(('127.0.0.1', 8080), acceptor, None) mock_socket.assert_called_once_with(socket.AF_INET, socket.SOCK_STREAM) mock_socket.return_value.assert_has_calls([ mock.call.bind(('', 0)), mock.call.connect(('127.0.0.1', 8080)), ]) mock_TCPTendril.assert_called_once_with(manager, mock_socket.return_value) acceptor.assert_called_once_with(mock_TCPTendril.return_value) mock_track_tendril.assert_called_once_with( mock_TCPTendril.return_value) mock_TCPTendril.return_value._start.assert_called_once_with() self.assertEqual(id(tend), id(mock_TCPTendril.return_value))
def test_connect_reject_connection(self, mock_TCPTendril, mock_track_tendril, mock_connect, mock_socket): acceptor = mock.Mock(side_effect=application.RejectConnection) manager = tcp.TCPTendrilManager() tend = manager.connect(('127.0.0.1', 8080), acceptor) self.assertEqual(tend, None) mock_connect.assert_called_once_with(('127.0.0.1', 8080), acceptor, None) mock_socket.assert_called_once_with(socket.AF_INET, socket.SOCK_STREAM) mock_socket.return_value.assert_has_calls([ mock.call.bind(('', 0)), mock.call.connect(('127.0.0.1', 8080)), mock.call.close(), ]) mock_TCPTendril.assert_called_once_with(manager, mock_socket.return_value) acceptor.assert_called_once_with(mock_TCPTendril.return_value) self.assertFalse(mock_track_tendril.called) self.assertFalse(mock_TCPTendril.return_value._start.called)