import socket import urllib from functools import partial from unittest import TestCase import models import fixtures TCP_ADDR = ('127.0.0.1', 8283) UDP_ADDR = ('127.0.0.1', 8283) HTTP_URL = 'http://127.0.0.1:8080' nm_req = partial( models.create_nm_req, fixtures.device_mac, fixtures.device_name, fixtures.sensors, ) json_req = None # TODO class TestTCP(TestCase): def setUp(self) -> None: self.sock = socket.socket() self.sock.connect(TCP_ADDR) def test_nm(self) -> None: self.sock.sendall(nm_req().encode('utf-8')) self._wait_for_ok() def test_json(self) -> None: self.sock.sendall(b'') # TODO self._wait_for_ok() def _wait_for_ok(self) -> None: data = self.sock.recv(1024) self.assertEqual(data, b'OK')