def setUp(self):
self.mocked_dns_v1 = mock.Client(dns_v1_client.DnsV1)
self.mocked_dns_v1.Mock()
self.addCleanup(self.mocked_dns_v1.Unmock)
def testMockHasUrlProperty(self):
with mock.Client(fusiontables.FusiontablesV1) as mock_client:
self.assertEquals(fusiontables.FusiontablesV1.BASE_URL,
mock_client.url)
self.assertFalse(hasattr(mock_client, 'url'))
def testMockHasHttpProperty(self):
with mock.Client(fusiontables.FusiontablesV1) as mock_client:
self.assertIsInstance(mock_client.http, httplib2.Http)
self.assertFalse(hasattr(mock_client, 'http'))
def SetUp(self):
self.client = mock.Client(
apis.GetClientClass('cloudresourcesearch', 'v1'))
self.client.Mock()
self.addCleanup(self.client.Unmock)
self.messages = self.client.MESSAGES_MODULE
def testFusionUnmock(self):
with mock.Client(fusiontables.FusiontablesV1):
client = fusiontables.FusiontablesV1(get_credentials=False)
mocked_service_type = type(client.column)
client = fusiontables.FusiontablesV1(get_credentials=False)
self.assertNotEqual(type(client.column), mocked_service_type)
def SetUpMockClient(api):
mock_client = mock.Client(core_apis.GetClientClass('compute', api),
real_client=core_apis.GetClientInstance(
'compute', api, no_http=True))
mock_client.Mock()
return mock_client
def SetUp(self):
self.mock_client = api_mock.Client(
base.GetClientClass(), real_client=base.GetClientInstance())
self.mock_client.Mock()
self.addCleanup(self.mock_client.Unmock)
self.messages = base.GetMessagesModule()
def Client(self):
return api_mock.Client(self.client_class,
real_client=core_apis.GetClientInstance(
'compute', 'alpha', no_http=True))
def Client(self):
client_class = core_apis.GetClientClass('compute', 'v1')
return api_mock.Client(client_class,
real_client=core_apis.GetClientInstance(
'compute', 'v1', no_http=True))
def SetUp(self):
self.client = apis.GetClientClass('spanner', 'v1')
self.mocked_client = apitools_mock.Client(self.client)
self.mocked_client.Mock()
self.addCleanup(self.mocked_client.Unmock)
self.messages = self.client.MESSAGES_MODULE
def SetUp(self):
client = apis.GetClientClass('compute', 'alpha')
self.mocked_client = apitools_mock.Client(client)
self.messages = client.MESSAGES_MODULE
self.mocked_client.Mock()
self.addCleanup(self.mocked_client.Unmock)
self.test_instance = self.messages.Instance(
canIpForward=None,
confidentialInstanceConfig=None,
cpuPlatform=None,
creationTimestamp=None,
deletionProtection=None,
description=None,
disks=[],
displayDevice=None,
eraseWindowsVssSignature=None,
fingerprint=None,
guestAccelerators=[],
hostname=None,
id=None,
instanceEncryptionKey=None,
kind='compute#instance',
labelFingerprint=None,
labels=None,
machineType=None,
metadata=None,
minCpuPlatform=None,
name='test',
networkInterfaces=[],
postKeyRevocationActionType=None,
preservedStateSizeGb=None,
privateIpv6GoogleAccess=None,
reservationAffinity=None,
resourcePolicies=[],
scheduling=None,
selfLink=None,
selfLinkWithId=None,
serviceAccounts=[],
shieldedInstanceConfig=None,
shieldedInstanceIntegrityPolicy=None,
shieldedVmConfig=None,
shieldedVmIntegrityPolicy=None,
sourceMachineImage=None,
sourceMachineImageEncryptionKey=None,
startRestricted=None,
status=None,
statusMessage=None,
tags=None,
zone=None,
)
self.test_running_operation = self.messages.Operation(
id=12345,
name='operation-12345',
selfLink=
'https://compute.googleapis.com/compute/v1/projects/p/zones/z/'
'operations/operation-12345',
error=None,
status=self.messages.Operation.StatusValueValuesEnum.RUNNING)
# Set up input file for command to import from.
temp_path = tempfile.mkdtemp(dir=self.root_path)
self.Touch(temp_path, 'test.yaml', 'name: test')
self.input_path = os.path.join(temp_path, 'test.yaml')
def SetUp(self):
self.client = mock.Client(core_apis.GetClientClass('compute', 'v1'))
self.client.Mock()
self.addCleanup(self.client.Unmock)
self.messages = self.client.MESSAGES_MODULE
def SetUp(self):
self.mocked_billing = mock.Client(
core_apis.GetClientClass('cloudbilling', 'v1'))
self.mocked_billing.Mock()
self.addCleanup(self.mocked_billing.Unmock)
self.messages = messages
def SetUp(self):
self.client = mock.Client(
client_class=apis.GetClientClass('storage', 'v1'))
self.client.Mock()
self.addCleanup(self.client.Unmock)
self.messages = apis.GetMessagesModule('storage', 'v1')
def setUp(self):
self.mocked_client = mock.Client(fusiontables.FusiontablesV1)
self.mocked_client.Mock()
self.addCleanup(self.mocked_client.Unmock)
def SetUp(self):
self.mocked_storage_v1 = api_mock.Client(
core_apis.GetClientClass('storage', 'v1'))
self.mocked_storage_v1.Mock()
self.addCleanup(self.mocked_storage_v1.Unmock)
self.storage_v1_messages = core_apis.GetMessagesModule('storage', 'v1')
def testRunWithOverrides(self):
client = apis.GetClientClass('ml', 'v1')
mocked_client = apitools_mock.Client(client)
mocked_client.Mock()
self.addCleanup(mocked_client.Unmock)
messages = client.MESSAGES_MODULE
policy = messages.GoogleIamV1Policy(bindings=[{
'role':
'roles/owner',
'members': [
'user:[email protected]',
'group:[email protected]',
]
}])
updated_policy = messages.GoogleIamV1Policy(
bindings=[{
'role': 'roles/owner',
'members': ['user:[email protected]']
}])
self.StartObjectPatch(iam_util,
'ParsePolicyFileWithUpdateMask',
return_value=(updated_policy,
'bindings,etag,version'))
mocked_client.projects_models.GetIamPolicy.Expect(
messages.MlProjectsModelsGetIamPolicyRequest(
resource='projects/p/models/m'), policy)
set_iam_policy_request = messages.GoogleIamV1SetIamPolicyRequest(
policy=updated_policy)
mocked_client.projects_models.SetIamPolicy.Expect(
messages.MlProjectsModelsSetIamPolicyRequest(
resource='projects/p/models/m',
googleIamV1SetIamPolicyRequest=set_iam_policy_request),
updated_policy)
command_data = yaml_command_schema.CommandData(
'remove_iam_policy_binding',
self.MakeIAMCommandData(
help_text='to remove IAM policy binding from',
another_collection=True))
command_data.iam = yaml_command_schema.IamData({
'set_iam_policy_request_path':
'googleIamV1SetIamPolicyRequest',
'message_type_overrides': {
'policy': 'GoogleIamV1Policy',
'set_iam_policy_request': 'GoogleIamV1SetIamPolicyRequest'
}
})
cli = self.MakeCLI(command_data)
self.AssertArgs(cli, 'MODEL', '--member', '--role')
result = cli.Execute([
'command', '--project', 'p', '--project', 'p', 'm', '--role',
'roles/owner', '--member', 'group:[email protected]'
])
self.assertEqual(result, policy)
self.AssertErrContains('Updated IAM policy for model [m].',
normalize_space=True)
self.AssertOutputEquals("""
bindings:
- members:
- user:[email protected]
role: roles/owner
""".lstrip('\n'),
normalize_space=True)
def SetUp(self):
self.client = mock.Client(
client_class=apis.GetClientClass('gameservices', 'v1alpha'))
self.client.Mock()
self.addCleanup(self.client.Unmock)
self.messages = apis.GetMessagesModule('gameservices', 'v1alpha')
def SetUp(self):
self.mocked_cloudbuild_v1 = client_mocker.Client(
core_apis.GetClientClass('cloudbuild', 'v1'), )
self.mocked_cloudbuild_v1.Mock()
self.addCleanup(self.mocked_cloudbuild_v1.Unmock)
self.cloudbuild_v1_messages = core_apis.GetMessagesModule(
'cloudbuild', 'v1')
self.mocked_storage_v1 = client_mocker.Client(
core_apis.GetClientClass('storage', 'v1'))
self.mocked_storage_v1.Mock()
self.addCleanup(self.mocked_storage_v1.Unmock)
self.storage_v1_messages = core_apis.GetMessagesModule('storage', 'v1')
self.mocked_crm_v1 = client_mocker.Client(
core_apis.GetClientClass('cloudresourcemanager', 'v1'))
self.mocked_crm_v1.Mock()
self.addCleanup(self.mocked_crm_v1.Unmock)
self.crm_v1_messages = core_apis.GetMessagesModule(
'cloudresourcemanager', 'v1')
self.mocked_iam_v1 = client_mocker.Client(
core_apis.GetClientClass('iam', 'v1'))
self.mocked_iam_v1.Mock()
self.addCleanup(self.mocked_iam_v1.Unmock)
self.iam_v1_messages = core_apis.GetMessagesModule('iam', 'v1')
self.mocked_serviceusage_v1 = client_mocker.Client(
core_apis.GetClientClass('serviceusage', 'v1'))
self.mocked_serviceusage_v1.Mock()
self.addCleanup(self.mocked_serviceusage_v1.Unmock)
self.serviceusage_v1_messages = core_apis.GetMessagesModule(
'serviceusage', 'v1')
self.mocked_compute_client_v1 = client_mocker.Client(
core_apis.GetClientClass('compute', 'v1'),
real_client=core_apis.GetClientInstance('compute',
'v1',
no_http=True))
self.mocked_compute_client_v1.Mock()
self.addCleanup(self.mocked_compute_client_v1.Unmock)
self.compute_v1_messages = core_apis.GetMessagesModule('compute', 'v1')
make_requests_patcher = mock.patch(
'googlecloudsdk.api_lib.compute.request_helper.MakeRequests',
autospec=True)
self.addCleanup(make_requests_patcher.stop)
self.make_requests = make_requests_patcher.start()
def MakeRequests(*_, **kwargs):
if False: # pylint: disable=using-constant-test, generator mock
yield
# We check to see if the image we are importing to already exists,
# so in the default case, the image should not exist.
kwargs['errors'].append((404, 'Not Found'))
self.make_requests.side_effect = MakeRequests
self.permissions = self.crm_v1_messages.Policy(bindings=[
self.crm_v1_messages.Binding(members=[
'serviceAccount:[email protected]'
],
role=daisy_utils.ROLE_COMPUTE_ADMIN),
self.crm_v1_messages.Binding(
members=[
'serviceAccount:[email protected]'
],
role=daisy_utils.ROLE_IAM_SERVICE_ACCOUNT_TOKEN_CREATOR),
self.crm_v1_messages.Binding(
members=[
'serviceAccount:[email protected]'
],
role=daisy_utils.ROLE_IAM_SERVICE_ACCOUNT_USER),
self.crm_v1_messages.Binding(members=[
'serviceAccount:[email protected]'
],
role=daisy_utils.
ROLE_COMPUTE_STORAGE_ADMIN),
self.crm_v1_messages.Binding(members=[
'serviceAccount:[email protected]'
],
role=daisy_utils.
ROLE_STORAGE_OBJECT_VIEWER),
])
self.uuid_mock = self.StartObjectPatch(uuid, 'uuid4')
self.uuid_mock.return_value = uuid.UUID(
'12345678123456781234567812345678')
properties.VALUES.core.project.Set('my-project')
self._statuses = self.cloudbuild_v1_messages.Build.StatusValueValuesEnum
self.project = self.crm_v1_messages.Project(projectId='my-project',
projectNumber=123456)
self.tags = ['gce-daisy']
self.network = 'my-network'
self.subnet = 'my-subnet'
def _SetUp(self):
self.track = calliope_base.ReleaseTrack.GA
self.client = mock.Client(core_apis.GetClientClass('compute', 'v1'))
self.client.Mock()
self.addCleanup(self.client.Unmock)
self.messages = self.client.MESSAGES_MODULE
def testMockIfAnotherException(self):
with self.assertRaises(CustomException):
with mock.Client(fusiontables.FusiontablesV1) as client_class:
client_class.column.List.Expect(request=1, response=2)
raise CustomException('Something when wrong')
def MockAPI(self, api, version):
client = apis.GetClientClass(api, version)
self.mocked_client = mock.Client(client)
self.messages = client.MESSAGES_MODULE
self.mocked_client.Mock()
self.addCleanup(self.mocked_client.Unmock)
def setUp(self):
self.mocked_iam_v1 = mock.Client(iam_v1_client.IamV1)
self.mocked_iam_v1.Mock()
self.addCleanup(self.mocked_iam_v1.Unmock)
def SetUp(self):
self.mock_firestore_v1 = mock.Client(
client_class=apis.GetClientClass('firestore', 'v1'))
self.mock_firestore_v1.Mock()
self.addCleanup(self.mock_firestore_v1.Unmock)
properties.VALUES.core.disable_prompts.Set(True)
def testMockHasMessagesModule(self):
with mock.Client(fusiontables.FusiontablesV1) as mock_client:
self.assertEquals(fusiontables_messages,
mock_client.MESSAGES_MODULE)
def SetUp(self):
self.mock_client = mock.Client(
core_apis.GetClientClass('clouderrorreporting', 'v1beta1'))
self.mock_client.Mock()
self.addCleanup(self.mock_client.Unmock)
def testMockHasOverrideUrlProperty(self):
real_client = fusiontables.FusiontablesV1(url='http://localhost:8080',
get_credentials=False)
with mock.Client(fusiontables.FusiontablesV1,
real_client) as mock_client:
self.assertEquals('http://localhost:8080/', mock_client.url)
def SetUp(self):
self.messages = apis.GetMessagesModule(self.DATAFUSION, self.API_VERSION)
self.mock_client = api_mock.Client(
apis.GetClientClass(self.DATAFUSION, self.API_VERSION))
self.mock_client.Mock()
self.addCleanup(self.mock_client.Unmock)
def setUp(self):
self.mocked_client = mock.Client(iam_client.IamV1)
self.mocked_client.Mock()
self.addCleanup(self.mocked_client.Unmock)
def SetUp(self):
"""Runs before any test method to set up the test environment."""
properties.VALUES.run.platform.Set(self.platform)
properties.VALUES.run.region.Set(self.region)
self._registry = resources.REGISTRY.Clone()
self.is_interactive = self.StartObjectPatch(console_io,
'IsInteractive',
return_value=True)
self.namespace = self._NamespaceRef()
self.messages = apis.GetMessagesModule(self.api_name, self.api_version)
self.crd_messages = apis.GetMessagesModule(self.api_name,
_CRD_API_VERSION)
self.core_messages = apis.GetMessagesModule(_CORE_API_NAME,
_CORE_API_VERSION)
# Create mock clients.
client_class = apis.GetClientClass(self.api_name, self.api_version)
real_client = apis.GetClientInstance(self.api_name,
self.api_version,
no_http=True)
self.mock_client = apitools_mock.Client(client_class, real_client)
self.mock_client.Mock()
self.addCleanup(self.mock_client.Unmock)
# Reuse the mock client above, if possible.
self.mock_crd_client = self.mock_client
if self.api_version != _CRD_API_VERSION:
client_class = apis.GetClientClass(self.api_name, _CRD_API_VERSION)
real_client = apis.GetClientInstance(self.api_name,
_CRD_API_VERSION,
no_http=True)
self.mock_crd_client = apitools_mock.Client(
client_class, real_client)
self.mock_crd_client.Mock()
self.addCleanup(self.mock_crd_client.Unmock)
# Reuse one of the mock clients above, if possible.
self.mock_core_client = self.mock_client
if self.api_name != _CORE_API_NAME or self.api_version != _CORE_API_VERSION:
self.mock_core_client = self.mock_crd_client
if self.api_name != _CORE_API_NAME or _CRD_API_VERSION != _CORE_API_VERSION:
client_class = apis.GetClientClass(_CORE_API_NAME,
_CORE_API_VERSION)
real_client = apis.GetClientInstance(_CORE_API_NAME,
_CORE_API_VERSION,
no_http=True)
self.mock_core_client = apitools_mock.Client(
client_class, real_client)
self.mock_core_client.Mock()
self.addCleanup(self.mock_core_client.Unmock)
self.operations = mock.Mock()
self.operations.client = self.mock_client
self.operations.messages = self.messages
operations_context = mock.Mock()
operations_context.__enter__ = mock.Mock(return_value=self.operations)
operations_context.__exit__ = mock.Mock(return_value=False)
self.StartObjectPatch(eventflow_operations,
'Connect',
return_value=operations_context)
