def test_gcs_init(self, google_client):
handler = GCSResultHandler(bucket="bob")
assert handler.bucket == "bob"
assert handler.credentials_secret == "GOOGLE_APPLICATION_CREDENTIALS"
assert google_client.called is False
handler.initialize_client()
assert google_client.return_value.bucket.call_args[0][0] == "bob"
def test_gcs_uses_custom_secret_name(self, google_client):
handler = GCSResultHandler(bucket="foo", credentials_secret="TEST_SECRET")
with prefect.context(secrets=dict(TEST_SECRET=94611)):
with set_temporary_config({"cloud.use_local_secrets": True}):
handler.initialize_client()
assert google_client.call_args[1]["credentials"] == 94611
def test_gcs_uses_custom_secret_name(self):
auth = MagicMock()
handler = GCSResultHandler(bucket="foo", credentials_secret="TEST_SECRET")
with prefect.context(secrets=dict(TEST_SECRET=94611)):
with set_temporary_config({"cloud.use_local_secrets": True}):
with patch.dict(
"sys.modules",
{
"google.cloud": MagicMock(),
"google.oauth2.service_account": auth,
},
):
handler.initialize_client()
assert auth.Credentials.from_service_account_info.call_args[0][0] == 94611