[Test] Fix CH migration tests

diff --git a/test/functional/cases/210_clickhouse/clickhouse.py b/test/functional/cases/210_clickhouse/clickhouse.py
index c2d0b5fa53f919d71c99553cb7717de79449fa55..e30e7832047985eac8e07b837dec678af5eb51d4 100644 (file)
--- a/test/functional/cases/210_clickhouse/clickhouse.py
+++ b/test/functional/cases/210_clickhouse/clickhouse.py
@@ -66,7 +66,7 @@ def schema_version_should_be(version):
     sql = "select max(Version) as version from rspamd_version"
     r = client().query(sql)
     logger.info("response: %s" % str(r))
-    if r[0]['version'] != 3:
+    if r[0]['version'] != int(version):
         raise Exception("Failed asseting that schema version is '%d'" % r[0]['version'])