def testSendFunctionality(self):
print("\n---------- testSendFunctionality ----------")
#: Get basic setup.
client = chatroomClient()
client.client.connect(('localhost', 12345))
client.silent = True
#: Run the test
client.send("TEST")
time.sleep(.01)
#: Check results.
self.assertTrue(any("TEST" in msg[0] for msg in self.server.messages))
#: Cleanup.
client.quit(False)
def testListenFunctionality(self):
print("\n---------- testListenFunctionality ----------")
#: Get basic setup.
client = chatroomClient()
client.join('localhost', 12345, silent=True)
time.sleep(.05) #: Give the server time to update.
server_cnx = self.server.clientlist[0][0]
#: Run the tests
server_cnx.send("close Server Test".encode())
time.sleep(.1) #: Give the server time to update.
#: Check results
#: If a socket's fileno() method returns -1, then
#: the socket has been disconnected.
self.assertEqual(client.client.fileno(), -1)
def testSendAllFunctionality(self):
print("---------- testSendAllFunctionality ----------")
spare_client = chatroomClient()
self.server.start(no_console=True)
self.client.join('localhost', 12345, silent=True)
spare_client.join('localhost', 12345, silent=True)
time.sleep(0.01) #: Give server time to catch up.
self.server.send_all('test')
time.sleep(1) #: Allow time for both clients to recieve the message.
#: 4 messages need to be sent at a .25s delay
#: Cleanup
spare_client.quit(False)
#: Ensure that the message is recieved by both.
self.assertEqual(self.client.most_recent_message, "(server): test")
self.assertEqual(spare_client.most_recent_message, "(server): test")
def testQuitFunctionality(self):
"""
Test to see if the quit function will successfully
sever the connection to the server.
"""
print("\n---------- testQuitFunctionality ----------")
#: Get basic setup
client = chatroomClient()
client.client.connect(('localhost', 12345))
#: Run the tests
client.quit(False)
time.sleep(.01) #: Allow time for the server to update.
#: Check results:
#: Ensure that the quit message was sent successfully.
self.assertEqual(len(self.server.clientlist), 0)
#: Ensure that the quit command sucessfully closed the socket.
self.assertEqual(client.client.fileno(), -1)
def testJoinFunctionality(self):
"""
This will test to see if a client can
successfully connect to the server and
be registered as connected.
"""
print("\n---------- testJoinFunctionality ----------")
#: Get basic setup
client = chatroomClient()
#: Run the tests.
client.join('localhost', 12345, silent=True)
time.sleep(.01) #: Give the server time to update.
#: Check results.
self.assertTrue(len(self.server.clientlist) == 1)
#: Clean up
client.joined = False
client.client.close()
def setUp(self):
print("\n")
self.server = chatroomServer('localhost', 12345)
self.client = chatroomClient()