def start_client(proto, server):
manager = tendril.get_manager(proto)
manager.start()
tend = manager.connect(server, EchoApplicationClient)
return manager, tend.application
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_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_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)
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)
