Skip to content

Commit

Permalink
refactor get() for multiple resources
Browse files Browse the repository at this point in the history
  • Loading branch information
linguini-dev committed Feb 6, 2025
1 parent 887716e commit 17ecbf9
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions tests/test_gce.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,11 +50,11 @@ def delete(self, *args, **kwargs):
return MockRequest(None)
raise ValueError("Unexpected delete request")

def get(self, project, region):
def something():
pass
something.execute = lambda: self.responses
return something
def get(self, *args, **kwargs):
for resource in ('object', 'image', 'disk', 'instance', 'firewall', 'forwardingRule', 'route', 'network', 'subnetwork'):
if resource in kwargs:
return MockRequest(self.responses)
raise ValueError(f"Unexpected get request")


class MockClient:
Expand Down

0 comments on commit 17ecbf9

Please sign in to comment.