class TestClient(unittest.TestCase): def setUp(self): self.server = socket(AF_INET, SOCK_STREAM) address = ('',1234) self.server.bind(address) self.server.listen() self.client = ChatClient("Mauricio","localhost",1234) self.client.init_socket() def tearDown(self): self.client.close_socket() self.server.close() def test_str_connected(self): self.assertEqual(self.client.__str__(),"Mauricio -> localhost:1234 [CONNECTED]") def test_str_disconnected(self): self.client.close_socket() self.assertEqual(self.client.__str__(),"Mauricio -> localhost:1234 [DISCONNECTED]") def test_get_username(self): self.assertIs(self.client.get_username(),self.client.username) def test_get_host(self): self.assertIs(self.client.get_host(),self.client.host) def test_get_port(self): self.assertIs(self.client.get_port(),self.client.port) def test_get_address(self): self.assertIs(self.client.get_address(),self.client.address) def test_get_socket(self): self.assertIs(self.client.get_socket(),self.client.client_socket) def test_get_status(self): self.assertEqual(self.client.get_status(),self.client.status) def test_set_username(self): username = "******" self.client.set_username(username) self.assertIs(username,self.client.username) def test_set_host(self): host = "127.0.0.8" self.client.set_host(host) self.assertIs(host,self.client.host) def test_set_port(self): port = 33000 self.client.set_port(port) self.assertIs(port,self.client.port) def test_set_address(self): address = ("127.0.2.0",12345) self.client.set_address(address) self.assertIs(address,self.client.address) def test_set_socket(self): self.client.close_socket() new_socket = socket(AF_INET, SOCK_STREAM) address = ("127.0.0.1",1234) new_socket.connect(address) self.client.set_socket(new_socket) self.assertIs(new_socket,self.client.client_socket) def test_set_socket_None(self): socket = None self.client.set_socket(socket) self.assertIs(socket,self.client.client_socket) def test_set_status(self): self.client.set_status("BUSY") self.assertEqual(self.client.status,"BUSY")