Skip to content

Commit 5758994

Browse files
committed
test: update tests to reflect import change
Signed-off-by: Mike Fiedler <[email protected]>
1 parent 8cc3d29 commit 5758994

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

tests/unit/test_gcloud.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,14 +45,12 @@ def test_gcloud_bigquery_factory(monkeypatch):
4545
def test_gcloud_gcs_factory(monkeypatch):
4646
client = pretend.stub()
4747

48-
storage = pretend.stub(
49-
Client=pretend.stub(
50-
from_service_account_json=pretend.call_recorder(
51-
lambda path, project: client
52-
)
48+
storage_Client = pretend.stub(
49+
from_service_account_json=pretend.call_recorder(
50+
lambda path, project: client
5351
)
5452
)
55-
monkeypatch.setattr(gcloud, "storage", storage)
53+
monkeypatch.setattr(gcloud, "storage_Client", storage_Client)
5654

5755
request = pretend.stub(
5856
registry=pretend.stub(
@@ -64,7 +62,7 @@ def test_gcloud_gcs_factory(monkeypatch):
6462
)
6563

6664
assert gcloud.gcloud_gcs_factory(None, request) is client
67-
assert storage.Client.from_service_account_json.calls == [
65+
assert storage_Client.from_service_account_json.calls == [
6866
pretend.call("/the/path/to/gcloud.json", project="my-cool-project")
6967
]
7068

0 commit comments

Comments
 (0)