diff options
author | Sebastian Bank <sebastian.bank@uni-leipzig.de> | 2016-03-19 12:17:40 +0100 |
---|---|---|
committer | Sebastian Bank <sebastian.bank@uni-leipzig.de> | 2016-03-19 12:19:14 +0100 |
commit | f50d5039770ed0c032975259ebd4df1bb0b036d3 (patch) | |
tree | a0cd9eff808d565efc39a6eb979fb20d2630ba2d | |
parent | c0bf58a9b78d24f89c56416d30fae2c2b4e11ab9 (diff) | |
download | sqlalchemy-f50d5039770ed0c032975259ebd4df1bb0b036d3.tar.gz |
adapt inspector for new include arg of postgres get_view_names, add tests
-rw-r--r-- | lib/sqlalchemy/engine/reflection.py | 4 | ||||
-rw-r--r-- | test/dialect/postgresql/test_reflection.py | 14 |
2 files changed, 15 insertions, 3 deletions
diff --git a/lib/sqlalchemy/engine/reflection.py b/lib/sqlalchemy/engine/reflection.py index eaa5e2e48..0ee7336ff 100644 --- a/lib/sqlalchemy/engine/reflection.py +++ b/lib/sqlalchemy/engine/reflection.py @@ -312,7 +312,7 @@ class Inspector(object): info_cache=self.info_cache, **kw) return {} - def get_view_names(self, schema=None): + def get_view_names(self, schema=None, **kw): """Return all view names in `schema`. :param schema: Optional, retrieve names from a non-default schema. @@ -321,7 +321,7 @@ class Inspector(object): """ return self.dialect.get_view_names(self.bind, schema, - info_cache=self.info_cache) + info_cache=self.info_cache, **kw) def get_view_definition(self, view_name, schema=None): """Return definition for `view_name`. diff --git a/test/dialect/postgresql/test_reflection.py b/test/dialect/postgresql/test_reflection.py index 8da18108f..9770ed6d2 100644 --- a/test/dialect/postgresql/test_reflection.py +++ b/test/dialect/postgresql/test_reflection.py @@ -129,7 +129,19 @@ class MaterializedViewReflectionTest( def test_get_view_names(self): insp = inspect(testing.db) - eq_(set(insp.get_view_names()), set(['test_mview', 'test_regview'])) + eq_(set(insp.get_view_names()), set(['test_regview', 'test_mview'])) + + def test_get_view_names_plain(self): + insp = inspect(testing.db) + eq_(set(insp.get_view_names(include=('plain',))), set(['test_regview'])) + + def test_get_view_names_materialized(self): + insp = inspect(testing.db) + eq_(set(insp.get_view_names(include=('materialized',))), set(['test_mview'])) + + def test_get_view_names_empty(self): + insp = inspect(testing.db) + eq_(set(insp.get_view_names(include=())), set([])) def test_get_view_definition(self): insp = inspect(testing.db) |