From a739a3449ff20ba90f92305ed4bbdd36e05bc862 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Fri, 19 Jul 2019 10:40:27 -0400 Subject: Include 'p' for get_indexes() query Added support for reflection of indexes on PostgreSQL partitioned tables, which was added to PostgreSQL as of version 11. Fixes: #4771 Change-Id: I9e8e75c4d8a667b4d52d12afbd384e0a8db00466 --- test/dialect/postgresql/test_reflection.py | 22 ++++++++++++++++++++-- 1 file changed, 20 insertions(+), 2 deletions(-) (limited to 'test/dialect/postgresql/test_reflection.py') diff --git a/test/dialect/postgresql/test_reflection.py b/test/dialect/postgresql/test_reflection.py index ea72d5710..6e7d7202a 100644 --- a/test/dialect/postgresql/test_reflection.py +++ b/test/dialect/postgresql/test_reflection.py @@ -111,6 +111,7 @@ class PartitionedReflectionTest(fixtures.TablesTest, AssertsExecutionResults): metadata, Column("modulus", Integer, nullable=False), Column("data", String(30)), + Column("q", Integer), postgresql_partition_by="range(modulus)", ) @@ -124,6 +125,9 @@ class PartitionedReflectionTest(fixtures.TablesTest, AssertsExecutionResults): ), ) + if testing.against("postgresql >= 11"): + Index("my_index", dv.c.q) + def test_get_tablenames(self): assert {"data_values", "data_values_4_10"}.issubset( inspect(testing.db).get_table_names() @@ -131,11 +135,25 @@ class PartitionedReflectionTest(fixtures.TablesTest, AssertsExecutionResults): def test_reflect_cols(self): cols = inspect(testing.db).get_columns("data_values") - eq_([c["name"] for c in cols], ["modulus", "data"]) + eq_([c["name"] for c in cols], ["modulus", "data", "q"]) def test_reflect_cols_from_partition(self): cols = inspect(testing.db).get_columns("data_values_4_10") - eq_([c["name"] for c in cols], ["modulus", "data"]) + eq_([c["name"] for c in cols], ["modulus", "data", "q"]) + + @testing.only_on("postgresql >= 11") + def test_reflect_index(self): + idx = inspect(testing.db).get_indexes("data_values") + eq_( + idx, [{"column_names": ["q"], "name": "my_index", "unique": False}] + ) + + @testing.only_on("postgresql >= 11") + def test_reflect_index_from_partition(self): + idx = inspect(testing.db).get_indexes("data_values_4_10") + # note the name appears to be generated by PG, currently + # 'data_values_4_10_q_idx' + eq_(idx, [{"column_names": ["q"], "name": mock.ANY, "unique": False}]) class MaterializedViewReflectionTest( -- cgit v1.2.1