From b1dd80b69cf259d4ad3a26c9cf8f9440e3ace0fd Mon Sep 17 00:00:00 2001 From: Eugene Crosser Date: Wed, 15 Jun 2022 00:04:09 +0200 Subject: [PATCH] test: fuzzing collector --- test/test_fuzz.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 test/test_fuzz.py diff --git a/test/test_fuzz.py b/test/test_fuzz.py new file mode 100644 index 0000000..6dea449 --- /dev/null +++ b/test/test_fuzz.py @@ -0,0 +1,30 @@ +""" Send junk to the collector """ + +from random import Random +from socket import getaddrinfo, socket, AF_INET6, SOCK_STREAM +from unittest import TestCase + +REPEAT: int = 1000000 + + +class Fuzz(TestCase): + def setUp(self) -> None: + self.rnd = Random() + for fam, typ, pro, cnm, skadr in getaddrinfo( + "::1", + 4303, + family=AF_INET6, + type=SOCK_STREAM, + ): + break # Just take the first element + self.sock = socket(AF_INET6, SOCK_STREAM) + self.sock.connect(skadr) + + def tearDown(self) -> None: + self.sock.close() + + def test_fuzz(self) -> None: + for _ in range(REPEAT): + size = self.rnd.randint(1, 5000) + buf = self.rnd.randbytes(size) + self.sock.send(buf) -- 2.39.2