1 files changed, 95 insertions, 0 deletions
diff --git a/fuzzer.py b/fuzzer.py
new file mode 100755
index 0000000..b1daf9b
--- /dev/null
+++ b/fuzzer.py
@@ -0,0 +1,95 @@
+#!/usr/bin/env python3
+import functools
+import random
+from multiprocessing.pool import ThreadPool as Pool
+
+import requests
+from pyinfluxtools import *
+from influxdb import InfluxDBClient
+
+settings = {
+    'host': 'localhost',
+    'port': 8086,
+    'username': 'root',
+    'password': 'root',
+    'database': 'test',
+}
+
+write_url = ("http://{host}:{port}/write?db={database}" +
+             "&username={username}&password={password}"
+             ).format(**settings)
+
+client = InfluxDBClient(settings['host'], settings['port'],
+                        settings['username'], settings['password'],
+                        settings['database'])
+
+
+class Generator:
+    all = lambda x: bytes(chr(random.randint(0x00, 0xff)), 'latin-1')
+    _printables = list(map(chr, list(range(0x20, 0x7E))))
+    printable = lambda x: random.choice(Generator._printables)
+    numericText = lambda x: chr(random.randint(ord("0"), ord("9")))
+    text = lambda x: chr(random.choice(
+        list(range(ord("a"), ord("z"))) +
+        list(range(ord("A"), ord("Z")))))
+
+
+class Filter:
+    regex = lambda r: re.compile(r).match
+    alpha = re.compile("^[a-z]+$").match
+    pass_all = lambda _: True
+    @staticmethod
+    def except_chars(chars):
+        regex = re.compile("^[^"+chars+"]*$")
+        return regex.match
+            
+
+
+def generate(length_min, length_max, filter, generator):
+    length = random.randint(length_min, length_max)
+    while True:
+        text = functools.reduce(lambda a,b: a+b,map(generator, range(length)))
+        if filter(text):
+            return text
+
+
+def run(*a):
+    while True:
+        test1 = generate(3, 6, 
+                Filter.regex("^[^\{\"]"), # influxdb doesn't allow measurements starting with { or "
+                Generator.printable)
+        test2 = generate(3, 6, Filter.pass_all, Generator.printable)
+        test3 = generate(3, 6, Filter.pass_all, Generator.printable)
+        test4 = generate(3, 6, Filter.pass_all, Generator.printable)
+        w = Write(test1,  # key
+                  {test2: test3},  # tags
+                  {"value": test4})  # values
+        line = str(w)
+        print(line)
+        try:
+            r = requests.post(write_url, line)
+            assert r.status_code == 204, \
+                "Data:\nline={}\nwrite={}\nr.status_code={}\nr.content={}".format(
+                    line, repr(w), r.status_code, r.text)
+        except Exception as e:
+            print(e)
+            assert False, "Data\nline={}\nwrite={}".format(str(w), repr(w))
+
+        measurement = Write.escape_value(w.key)
+        query = "SELECT * FROM {measurement} WHERE time >= now() - 1s".format(**locals())
+        result = list(client.query(query))
+        DEBUGINFO = "DEBUG:\nquery={query}\nresult={result}\nline={line}".format(**locals())
+        assert len(result) == 1, DEBUGINFO
+        assert result[0][0]['value'] == test4, DEBUGINFO + "\nvalue=" + result[0][0]['value'] + "\ntest4=" + test4
+
+
+N_PROC = 1
+with Pool(processes=N_PROC) as pool:
+    try:
+        for res in pool.imap_unordered(run, [None] * N_PROC):
+            pool.terminate()
+            print("Exit")
+            raise SystemExit(0)
+    except Exception as e:
+        pool.terminate()
+        raise e