Merge branch 'version_property'

ildus · ildus · commit 7a7dfdd919e3 · 2018-06-25T18:57:27.000+03:00
diff --git a/testgres/node.py b/testgres/node.py
@@ -278,6 +278,16 @@ def utils_log_file(self):
     def pg_log_file(self):
         return os.path.join(self.logs_dir, PG_LOG_FILE)
 
+    @property
+    def version(self):
+        """
+        Return PostgreSQL version for this node.
+
+        Returns:
+            Instance of :class:`distutils.version.LooseVersion`.
+        """
+        return self._pg_version
+
     def _try_shutdown(self, max_attempts):
         attempts = 0
 
diff --git a/tests/test_simple.py b/tests/test_simple.py
@@ -7,6 +7,7 @@
 import tempfile
 import testgres
 import time
+import six
 import unittest
 
 import logging.config
@@ -857,6 +858,12 @@ def test_version_management(self):
         self.assertTrue(b > c)
         self.assertTrue(a > c)
 
+        version = get_pg_version()
+        with get_new_node() as node:
+            self.assertTrue(isinstance(version, six.string_types))
+            self.assertTrue(isinstance(node.version, PgVer))
+            self.assertTrue(node.version == version)
+
     def test_child_pids(self):
         master_processes = [
             ProcessType.AutovacuumLauncher,
