Stabilize tests.ptrack.PtrackTest.test_horizon_lsn_ptrack

Author: kulaginm

tests/helpers/ptrack_helpers.py

@@ -1602,6 +1602,13 @@ def get_version(self, node):
         return self.version_to_num(
             testgres.get_pg_config()['VERSION'].split(" ")[1])
 
+    def get_ptrack_version(self, node):
+        version = node.safe_psql(
+            "postgres",
+            "SELECT extversion "
+                        "FROM pg_catalog.pg_extension WHERE extname = 'ptrack'").decode('utf-8').rstrip()
+        return self.version_to_num(version)
+
     def get_bin_path(self, binary):
         return testgres.get_bin_path(binary)
 

tests/ptrack.py

@@ -4465,16 +4465,10 @@ def test_horizon_lsn_ptrack(self):
             "postgres",
             "CREATE EXTENSION ptrack")
 
-        # TODO: ptrack version must be 2.1
-        ptrack_version = node.safe_psql(
-            "postgres",
-            "SELECT extversion "
-			"FROM pg_catalog.pg_extension WHERE extname = 'ptrack'").decode('utf-8').rstrip()
-
-        self.assertEqual(
-            ptrack_version,
-            "2.1",
-            "You need ptrack 2.1 for this test")
+        self.assertGreaterEqual(
+            self.get_ptrack_version(node),
+            self.version_to_num("2.1"),
+            "You need ptrack >=2.1 for this test")
 
         # set map_size to a minimal value
         self.set_auto_conf(node, {'ptrack.map_size': '1'})