test/test_storage.py

   1 """ Send junk to the collector """
   2
   3 from random import Random
   4 from socket import getaddrinfo, socket, AF_INET, SOCK_STREAM
   5 from sqlite3 import connect
   6 from time import sleep
   7 from typing import Any
   8 import unittest
   9 from .common import send_and_drain, TestWithServers
  10 from gps303.gps303proto import *
  11
  12
  13 class Storage(TestWithServers):
  14     def setUp(self, *args: str, **kwargs: Any) -> None:
  15         super().setUp("collector", "storage", "lookaside", "termconfig")
  16         for fam, typ, pro, cnm, skadr in getaddrinfo(
  17             "127.0.0.1",
  18             self.conf.getint("collector", "port"),
  19             family=AF_INET,
  20             type=SOCK_STREAM,
  21         ):
  22             break  # Just take the first element
  23         self.sock = socket(AF_INET, SOCK_STREAM)
  24         self.sock.connect(skadr)
  25
  26     def tearDown(self) -> None:
  27         sleep(1)  # give collector some time
  28         super().tearDown()
  29
  30     def test_storage(self) -> None:
  31         for buf in (
  32             LOGIN.In(imei="9999123456780000", ver=9).packed,
  33             STATUS.In().packed,
  34             HIBERNATION.In().packed,
  35         ):
  36             send_and_drain(self.sock, b"xx" + buf + b"\r\n")
  37         self.sock.close()
  38         sleep(1)
  39         got = set()
  40         with connect(self.conf.get("storage", "dbfn")) as db:
  41             for is_incoming, packet in db.execute(
  42                 "select is_incoming, packet from events"
  43             ):
  44                 msg = parse_message(packet, is_incoming=is_incoming)
  45                 # print(msg)
  46                 got.add(type(msg))
  47         self.assertEqual(
  48             got, {LOGIN.Out, HIBERNATION.In, LOGIN.In, STATUS.Out, STATUS.In}
  49         )
  50
  51
  52 if __name__ == "__main__":
  53     unittest.main()