Skip to content

Commit

Permalink
test(iam, set_iam_credentials): ensure force_lowercase is used
Browse files Browse the repository at this point in the history
  • Loading branch information
Brooke-white committed Aug 16, 2021
1 parent 3348262 commit 2298641
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 0 deletions.
4 changes: 4 additions & 0 deletions test/integration/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,10 @@ def db_kwargs():
return _get_default_connection_args()


def db_groups():
return conf.get("cluster-setup", "groups").split(sep=",")


@pytest.fixture(scope="class")
def perf_db_kwargs():
db_connect = {
Expand Down
25 changes: 25 additions & 0 deletions test/integration/plugin/test_credentials_providers.py
Original file line number Diff line number Diff line change
Expand Up @@ -182,3 +182,28 @@ def test_stl_connection_log_contains_plugin_name(idp_arg, db_kwargs):
res = cursor.fetchone()
assert res is not None
assert res[0] == 1


@pytest.mark.parametrize("idp_arg", NON_BROWSER_IDP, indirect=True)
def raise_exception_when_uppercase_db_groups(idp_arg, db_groups):
idp_arg["db_groups"] = [group.upper() for group in db_groups]

with pytest.raises():
redshift_connector.connect(**idp_arg)


@pytest.mark.parametrize("idp_arg", NON_BROWSER_IDP, indirect=True)
def uses_force_lowercase_when_db_groups_uppercase(idp_arg, db_groups):
idp_arg["db_groups"] = [group.upper() for group in db_groups]
idp_arg["force_lowercase"] = True

with redshift_connector.connect(**idp_arg):
pass


@pytest.mark.parametrize("idp_arg", NON_BROWSER_IDP, indirect=True)
def uses_db_groups_nominal(idp_arg, db_groups):
idp_arg["db_groups"] = [group for group in db_groups]

with redshift_connector.connect(**idp_arg):
pass

0 comments on commit 2298641

Please sign in to comment.