summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSebastian Bank <sebastian.bank@uni-leipzig.de>2016-03-19 12:17:40 +0100
committerSebastian Bank <sebastian.bank@uni-leipzig.de>2016-03-19 12:19:14 +0100
commitf50d5039770ed0c032975259ebd4df1bb0b036d3 (patch)
treea0cd9eff808d565efc39a6eb979fb20d2630ba2d
parentc0bf58a9b78d24f89c56416d30fae2c2b4e11ab9 (diff)
downloadsqlalchemy-f50d5039770ed0c032975259ebd4df1bb0b036d3.tar.gz
adapt inspector for new include arg of postgres get_view_names, add tests
-rw-r--r--lib/sqlalchemy/engine/reflection.py4
-rw-r--r--test/dialect/postgresql/test_reflection.py14
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)