def test_send(self):
"""
test send_message() in class SenderSocketClient
:return:
"""
# send something to echo server, it should return the same thing
Config.RETURN_MSG = HoneynetSocketUtil.RET_SUCCESS
client = SenderSocketClient(self.server_addr)
client.connect()
self.assertTrue(client.is_connected)
msg = "something"
Config.RETURN_MSG = ""
client.send_message(msg)
ret_msg = client.sock.recv(Config.BUF_SIZE)
self.assertEqual(msg, ret_msg)
client.disconnect()
def test_connect(self):
"""
test client function connect() in class SenderSocketClient
:return:
"""
# test that the client successfully connects to a server which returns SUCCESS
Config.RETURN_MSG = HoneynetSocketUtil.RET_SUCCESS
client = SenderSocketClient(self.server_addr)
self.assertFalse(client.is_connected)
client.connect()
self.assertTrue(client.is_connected)
client.disconnect()
self.assertFalse(client.is_connected)
# test that the client fails to connect to a server which returns FAIL
Config.RETURN_MSG = HoneynetSocketUtil.RET_FAIL
client = SenderSocketClient(self.server_addr)
self.assertFalse(client.is_connected)
with self.assertRaises(HoneynetSocketExceptions.DenySenderException):
client.connect()
self.assertFalse(client.is_connected)
# test that the client fails to connect to a server which returns unknown message
Config.RETURN_MSG = "This is an unknown message!"
client = SenderSocketClient(self.server_addr)
self.assertFalse(client.is_connected)
with self.assertRaises(HoneynetSocketExceptions.UnknownResponseException):
client.connect()
self.assertFalse(client.is_connected)