41 lines
817 B
Python
41 lines
817 B
Python
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')
|