[Test] Added test for clickhouse migration vNONE -> v2

1 files changed, 67 insertions, 0 deletions

diff --git a/test/functional/cases/210_clickhouse/clickhouse.py b/test/functional/cases/210_clickhouse/clickhouse.py
new file mode 100644
index 000000000..069589fed
--- /dev/null
+++ b/test/functional/cases/210_clickhouse/clickhouse.py
@@ -0,0 +1,67 @@
+import requests
+import json
+from robot.libraries.BuiltIn import BuiltIn
+from robot.api import logger
+
+__client = None
+
+
+class Client:
+    def __init__(self):
+        self.port = 18123
+
+    def get_query_string(self):
+        return "http://localhost:%d/?default_format=JSONEachRow" % (self.port)
+
+    def execute(self, sql):
+        r = requests.post(self.get_query_string(), sql)
+        if r.status_code != 200:
+            raise Exception("Clickhouse request failed: " + r.content)
+        return r
+
+    def query(self, sql):
+        r = self.execute(sql)
+
+        # logger.console("decoding " + r.content)
+        # [logger.console("decoding " + _) for _ in r.content.strip().split("\n")]
+        response = [json.loads(_) for _ in r.content.strip().split("\n")]
+        return response
+
+
+def client():
+    global __client
+    if __client is None:
+        __client = Client()
+    return __client
+
+
+def upload_new_schema(schema_file):
+    with open(schema_file, 'r') as content_file:
+        content = content_file.read()
+
+    queries = content.split(";")
+    for q in queries:
+        if q.strip() == "":
+            continue
+        client().execute(q)  # throws exception on error
+
+
+def insert_data(table_name, filename):
+    with open(filename, 'r') as content_file:
+        content = content_file.read()
+
+    client().execute("insert into %s format Values %s;" % (table_name, content))  # throws exception on error
+
+
+def column_should_exist(table_name, column_name):
+    sql = "select hasColumnInTable('default', '%s', '%s') as is_exist" % (table_name, column_name)
+    r = client().query(sql)
+    if r[0]['is_exist'] != 1:
+        raise Exception("Failed asseting that column '%s' exists in table 'default'.'%s'" % (column_name, table_name))
+
+
+def schema_version_should_be(version):
+    sql = "select max(Version) as version from rspamd_version"
+    r = client().query(sql)
+    if r[0]['version'] != 2:
+        raise Exception("Failed asseting that schema version is '%d'" % version)
\ No newline at end of file