Support special postgres versions

v.shepard · v.shepard · commit b9d4c8f9a4ad · 2023-07-19T14:27:23.000+02:00
diff --git a/setup.py b/setup.py
@@ -27,7 +27,7 @@
     readme = f.read()
 
 setup(
-    version='1.8.8',
+    version='1.8.9',
     name='testgres',
     packages=['testgres'],
     description='Testing utility for PostgreSQL and its extensions',
diff --git a/testgres/utils.py b/testgres/utils.py
@@ -12,6 +12,7 @@
 
 from contextlib import contextmanager
 from packaging.version import Version
+
 try:
     from shutil import which as find_executable
 except ImportError:
@@ -27,8 +28,12 @@
 # ports used by nodes
 bound_ports = set()
 
+
 # re-export version type
-PgVer = Version
+class PgVer(Version):
+    def __init__(self, version: str) -> None:
+        version = version.replace('biha', '')
+        super().__init__(version)
 
 
 def reserve_port():
@@ -208,7 +213,8 @@ def get_pg_version():
                      .partition('devel')[0] \
                      .partition('beta')[0] \
                      .partition('rc')[0]
-
+    if 'biha' in version:
+        version = version.replace('biha', '')
     return version
 
 
diff --git a/tests/test_simple.py b/tests/test_simple.py
@@ -901,13 +901,16 @@ def test_version_management(self):
         d = PgVer('15.0')
         e = PgVer('15rc1')
         f = PgVer('15beta4')
+        g = PgVer('15.2beta4')
+        h = PgVer('15.2biha')
 
         self.assertTrue(a == b)
         self.assertTrue(b > c)
         self.assertTrue(a > c)
         self.assertTrue(d > e)
         self.assertTrue(e > f)
         self.assertTrue(d > f)
+        self.assertTrue(g.base_version == h.base_version)
 
         version = get_pg_version()
         with get_new_node() as node:
