def FormatOutput(self, out, args):
if out:
return domainmapping.DomainMapping(json.loads(out)).records
else:
raise exceptions.Error(
'Could not map domain [{}] to service [{}]'.format(
args.domain, args.service))
def testList_Succeed(self):
command = """kuberun core domain-mappings list --cluster foo
--cluster-location us-central1"""
expected_out = '[' + testdata.DOMAIN_MAPPING_STRING + ']'
self.mock_bin_exec.return_value = bin_ops.BinaryBackedOperation.OperationResult(
command, output=expected_out)
result = self.Run(command)
self.AssertExecuteCalledOnce(command_args=[
'core', 'domain-mappings', 'list', '--cluster', 'foo',
'--cluster-location', 'us-central1'
])
expected_result = domainmapping.DomainMapping(
json.loads(testdata.DOMAIN_MAPPING_STRING))
self.assertEqual(result[0], expected_result)
def testDescribe_Succeed(self):
command = """kuberun core domain-mappings describe hello.example.com
--cluster foo --cluster-location us-central1"""
expected_out = testdata.DOMAIN_MAPPING_STRING
self.mock_bin_exec.return_value = (
bin_ops.BinaryBackedOperation.OperationResult(command,
output=expected_out))
result = self.Run(command)
self.AssertExecuteCalledOnce(command_args=[
'core', 'domain-mappings', 'describe', 'hello.example.com',
'--cluster', 'foo', '--cluster-location', 'us-central1'
])
expected_result = domainmapping.DomainMapping(json.loads(expected_out))
self.assertEqual(result, expected_result)
def SuccessResult(self, out, args):
if out:
return domainmapping.DomainMapping(json.loads(out))
else:
raise exceptions.Error('Cannot find domain mapping [{}]'.format(
args.domain))
def FormatOutput(self, out, args):
if out:
json_object = json.loads(out)
return [domainmapping.DomainMapping(x) for x in json_object]
else:
raise exceptions.Error('Cannot list domain mappings')