tests/main.py

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')