Add an integration test

Author: vchan

sqlmesh/core/engine_adapter/postgres.py

@@ -127,8 +127,9 @@ def merge(
     @cached_property
     def server_version(self) -> t.Tuple[int, int]:
         """Lazily fetch and cache major and minor server version"""
-        server_version, *_ = self.fetchone("SHOW server_version")
-        match = re.search(r"(\d+)\.(\d+)", server_version)
-        if match:
-            return int(match.group(1)), int(match.group(2))
+        if result := self.fetchone("SHOW server_version"):
+            server_version, *_ = result
+            match = re.search(r"(\d+)\.(\d+)", server_version)
+            if match:
+                return int(match.group(1)), int(match.group(2))
         return 0, 0

tests/core/engine_adapter/integration/test_integration_postgres.py

@@ -2,7 +2,6 @@
 import pytest
 from pytest import FixtureRequest
 from sqlmesh.core.engine_adapter import PostgresEngineAdapter
-from tests.core.engine_adapter.integration import TestContext
 
 from tests.core.engine_adapter.integration import (
     TestContext,
@@ -29,3 +28,8 @@ def engine_adapter(ctx: TestContext) -> PostgresEngineAdapter:
 def test_engine_adapter(ctx: TestContext):
     assert isinstance(ctx.engine_adapter, PostgresEngineAdapter)
     assert ctx.engine_adapter.fetchone("select 1") == (1,)
+
+
+def test_server_version_psycopg(ctx: TestContext):
+    assert isinstance(ctx.engine_adapter, PostgresEngineAdapter)
+    assert ctx.engine_adapter.server_version != (0, 0)