fix: schemas for upload API (#29604)

This commit is contained in:
Beto Dealmeida
2024-07-16 15:53:25 -04:00
committed by GitHub
parent ec508a70a6
commit b66c0f8d30
3 changed files with 23 additions and 3 deletions

View File

@@ -461,3 +461,23 @@ def test_raw_connection_oauth(mocker: MockerFixture) -> None:
with database.get_raw_connection() as conn:
conn.cursor()
assert str(excinfo.value) == "You don't have permission to access the data."
def test_get_schema_access_for_file_upload() -> None:
"""
Test the `get_schema_access_for_file_upload` method.
"""
database = Database(
database_name="first-database",
sqlalchemy_uri="gsheets://",
extra=json.dumps(
{
"metadata_params": {},
"engine_params": {},
"metadata_cache_timeout": {},
"schemas_allowed_for_file_upload": '["public"]',
}
),
)
assert database.get_schema_access_for_file_upload() == {"public"}