improve test

test_runner/regress/test_extensions_stats.py

@@ -1,13 +1,16 @@
 from logging import info
+
 from fixtures.neon_fixtures import NeonEnv
 
+
 # basic test for the endpoint that returns the list of installed extensions
 def test_extensions_stats(neon_simple_env: NeonEnv):
     env = neon_simple_env
     env.neon_cli.create_branch("test_extensions_stats", "empty")
 
     endpoint = env.endpoints.create_start("test_extensions_stats")
     endpoint.safe_psql("CREATE DATABASE test_extensions_stats")
+    endpoint.safe_psql("CREATE DATABASE test_extensions_stats_2")
 
     client = endpoint.http_client()
     res = client.extensions()
@@ -28,12 +31,40 @@ def test_extensions_stats(neon_simple_env: NeonEnv):
     with pg_conn.cursor() as cur:
         cur.execute("CREATE EXTENSION neon_test_utils")
 
+    with pg_conn.cursor() as cur:
+        cur.execute("CREATE EXTENSION neon version '1.1'")
+
+    pg_conn_2 = endpoint.connect(dbname="test_extensions_stats_2")
+    with pg_conn_2.cursor() as cur:
+        cur.execute("CREATE EXTENSION neon version '1.2'")
+
     res = client.extensions()
 
     info("Extensions list: %s", res)
     info("Extensions: %s", res["extensions"])
 
     # check that the neon_test_utils extension is installed
     assert any(
-        ext["extname"] == "neon_test_utils" and ext["highest_version"] == "1.3" for ext in res["extensions"]
+        ext["extname"] == "neon_test_utils" and ext["highest_version"] == "1.3"
+        for ext in res["extensions"]
     ), "The 'neon_test_utils' extension is missing"
+
+    assert any(
+        ext["extname"] == "neon"
+        and ext["lowest_version"] == "1.1"
+        and ext["highest_version"] == "1.2"
+        for ext in res["extensions"]
+    ), "The 'neon' extension is missing"
+
+    with pg_conn.cursor() as cur:
+        cur.execute("ALTER EXTENSION neon UPDATE TO '1.3'")
+
+    res = client.extensions()
+
+    info("Extensions list: %s", res)
+    info("Extensions: %s", res["extensions"])
+
+    # check that the neon_test_utils extension is installed
+    assert any(
+        ext["extname"] == "neon" and ext["highest_version"] == "1.3" for ext in res["extensions"]
+    ), "The 'neon' extension is missing"