Skip to content

Commit 2298641

Browse files
committed
test(iam, set_iam_credentials): ensure force_lowercase is used
1 parent 3348262 commit 2298641

File tree

2 files changed

+29
-0
lines changed

2 files changed

+29
-0
lines changed

test/integration/conftest.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,10 @@ def db_kwargs():
5757
return _get_default_connection_args()
5858

5959

60+
def db_groups():
61+
return conf.get("cluster-setup", "groups").split(sep=",")
62+
63+
6064
@pytest.fixture(scope="class")
6165
def perf_db_kwargs():
6266
db_connect = {

test/integration/plugin/test_credentials_providers.py

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -182,3 +182,28 @@ def test_stl_connection_log_contains_plugin_name(idp_arg, db_kwargs):
182182
res = cursor.fetchone()
183183
assert res is not None
184184
assert res[0] == 1
185+
186+
187+
@pytest.mark.parametrize("idp_arg", NON_BROWSER_IDP, indirect=True)
188+
def raise_exception_when_uppercase_db_groups(idp_arg, db_groups):
189+
idp_arg["db_groups"] = [group.upper() for group in db_groups]
190+
191+
with pytest.raises():
192+
redshift_connector.connect(**idp_arg)
193+
194+
195+
@pytest.mark.parametrize("idp_arg", NON_BROWSER_IDP, indirect=True)
196+
def uses_force_lowercase_when_db_groups_uppercase(idp_arg, db_groups):
197+
idp_arg["db_groups"] = [group.upper() for group in db_groups]
198+
idp_arg["force_lowercase"] = True
199+
200+
with redshift_connector.connect(**idp_arg):
201+
pass
202+
203+
204+
@pytest.mark.parametrize("idp_arg", NON_BROWSER_IDP, indirect=True)
205+
def uses_db_groups_nominal(idp_arg, db_groups):
206+
idp_arg["db_groups"] = [group for group in db_groups]
207+
208+
with redshift_connector.connect(**idp_arg):
209+
pass

0 commit comments

Comments
 (0)