diff --git a/testgres/node.py b/testgres/node.py index 0b5e55cf..b171d965 100644 --- 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 index d395f25b..2e557309 100755 --- 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.assertEqual(node.version, str(version)) + def test_child_pids(self): master_processes = [ ProcessType.AutovacuumLauncher,