def testUpdateQuotaOverrideCall(self):
want = self.services_v1beta1_messages.Operation(
name=self.OPERATION_NAME, done=False)
self.ExpectUpdateQuotaOverrideCall(self.mutate_limit_name,
self.mutate_metric, self.unit, 666,
self.OPERATION_NAME)
dimensions = None
got = serviceusage.UpdateQuotaOverrideCall(self.DEFAULT_CONSUMER,
self.DEFAULT_SERVICE,
self.mutate_metric, self.unit,
dimensions, 666)
self.assertEqual(got, want)
def testUpdateQuotaOverrideCall_failed(self):
server_error = http_error.MakeDetailedHttpError(code=403, message='Error.')
self.ExpectUpdateQuotaOverrideCall(
self.mutate_limit_name,
self.mutate_metric,
self.unit,
666,
None,
error=server_error)
with self.assertRaisesRegex(
exceptions.UpdateQuotaOverridePermissionDeniedException, r'Error.'):
dimensions = None
serviceusage.UpdateQuotaOverrideCall(self.DEFAULT_CONSUMER,
self.DEFAULT_SERVICE,
self.mutate_metric, self.unit,
dimensions, 666)
def Run(self, args):
"""Run 'endpoints quota create'.
Args:
args: argparse.Namespace, The arguments that this command was invoked
with.
Returns:
Nothing.
"""
op = serviceusage.UpdateQuotaOverrideCall(args.consumer, args.service,
args.metric, args.unit,
args.dimensions, args.value,
args.force)
if op.done:
return
op = services_util.WaitOperation(op.name, serviceusage.GetOperation)
services_util.PrintOperation(op)
def testUpdateQuotaOverrideCall_WithDimensions(self):
want = self.services_v1beta1_messages.Operation(
name=self.OPERATION_NAME, done=False)
self.ExpectUpdateQuotaOverrideCall(
self.mutate_limit_name,
self.mutate_metric,
self.unit,
666,
self.OPERATION_NAME,
dimensions=[('regions', 'us-central1')],
force=True)
got = serviceusage.UpdateQuotaOverrideCall(
self.DEFAULT_CONSUMER,
self.DEFAULT_SERVICE,
self.mutate_metric,
self.unit, {'regions': 'us-central1'},
666,
force=True)
self.assertEqual(got, want)