def test_proxy_desktop(self):
desktop_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
proxy = Proxy.Interface()
proxy_thread = threading.Thread(target = proxy.start_server)
proxy_thread.start()
#connect
desktop_client.connect(('127.0.0.1',50222))
#gather the init message
who_response = desktop_client.recv(100).decode("ascii")
#should be who are you
self.assertEqual(who_response , "WHO_ARE_YOU")
#claim desktop
desktop_client.send("DESKTOP".encode("ascii"))
#recv response
status_response = desktop_client.recv(100).decode("ascii")
#should be waiting on raspberry pi
self.assertEqual(status_response , "AWAITING_PI")
#close sockets
desktop_client.close()
proxy.running = False
proxy.server.close()
