""" Send junk to the collector """
from random import Random
-from socket import getaddrinfo, socket, AF_INET6, SOCK_STREAM
-from unittest import TestCase
+from socket import getaddrinfo, socket, AF_INET6, MSG_DONTWAIT, SOCK_STREAM
+from time import sleep
+from typing import Optional
+import unittest
+from .common import TestWithServers
REPEAT: int = 1000000
-class Fuzz(TestCase):
- def setUp(self) -> None:
+class Fuzz(TestWithServers):
+ def setUp(self, *args: str) -> None:
+ super().setUp("collector")
self.rnd = Random()
for fam, typ, pro, cnm, skadr in getaddrinfo(
"::1",
- 4303,
+ self.conf.getint("collector", "port"),
family=AF_INET6,
type=SOCK_STREAM,
):
self.sock.connect(skadr)
def tearDown(self) -> None:
+ sleep(1) # give collector some time
+ self._send_and_drain(None)
self.sock.close()
+ print("finished fuzzing")
+ super().tearDown()
- def test_fuzz(self) -> None:
+ def _send_and_drain(self, buf: Optional[bytes]) -> None:
+ if buf is not None:
+ self.sock.send(buf)
+ try:
+ self.sock.recv(4096, MSG_DONTWAIT)
+ except BlockingIOError:
+ pass
+
+ def test_stream(self) -> None:
for _ in range(REPEAT):
size = self.rnd.randint(1, 5000)
buf = self.rnd.randbytes(size)
- self.sock.send(buf)
+ self._send_and_drain(buf)
+
+ def test_msgs(self) -> None:
+ for _ in range(REPEAT):
+ size = self.rnd.randint(0, 300)
+ buf = b"xx" + self.rnd.randbytes(size) + b"\r\n"
+ self._send_and_drain(buf)
+
+
+if __name__ == "__main__":
+ unittest.main()