Skip to content

Commit

Permalink
[BUGFIX] Patch bad mock in GCS test (#8128)
Browse files Browse the repository at this point in the history
  • Loading branch information
cdkini authored Jun 15, 2023
1 parent 71b15d0 commit 80529a5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/data_context/store/test_store_backends.py
Original file line number Diff line number Diff line change
Expand Up @@ -1153,7 +1153,7 @@ def test_TupleGCSStoreBackend(): # noqa: PLR0915
mock_client = mock_gcs_client.return_value
mock_bucket = mock_client.bucket.return_value
mock_blob = mock_bucket.get_blob.return_value
mock_str = mock_blob.download_as_string.return_value
mock_str = mock_blob.download_as_bytes.return_value

my_store.get(("BBB",))

Expand All @@ -1162,7 +1162,7 @@ def test_TupleGCSStoreBackend(): # noqa: PLR0915
mock_bucket.get_blob.assert_called_once_with(
"this_is_a_test_prefix/my_file_BBB"
)
mock_blob.download_as_string.assert_called_once()
mock_blob.download_as_bytes.assert_called_once()
mock_str.decode.assert_called_once_with("utf-8")

with mock.patch("google.cloud.storage.Client", autospec=True) as mock_gcs_client:
Expand Down

0 comments on commit 80529a5

Please sign in to comment.