class OperationsUnitTestBase(CrmUnitTestBase):
"""Base class for all Operations unit tests with fake auth and mocks."""
TEST_OPERATION = operations.OperationsMessages().Operation(
name='operations/fc.58219052', done=False)
def SetUp(self):
self.track = calliope_base.ReleaseTrack.ALPHA
def RunOperations(self, *command):
return self.Run(['resource-manager', 'operations'] + list(command))
def testCreateFolder(self):
response_value = operations.ToOperationResponse(self.TEST_FOLDER)
test_create_operation = self.messages.Operation(
done=False, name='operations/fc.123')
test_create_operation_done = operations.OperationsMessages().Operation(
done=True, name='operations/fc.123', response=response_value)
self.mock_folders.Create.Expect(self.ExpectedCreateRequest(),
test_create_operation)
self.mock_operations.Get.Expect(
self.ExpectedGetOperationRequest(test_create_operation),
test_create_operation_done)
self.assertEqual(self.DoCommand(), self.TEST_FOLDER)
def SetUpFolderMove(self, original, moved, is_async=False):
response_value = operations.ToOperationResponse(moved)
test_move_operation = self.messages.Operation(done=False,
name=TEST_OPERATION_NAME)
expected_move_request = folders.FoldersMessages(
).CloudresourcemanagerFoldersMoveRequest(
foldersId=folders.FolderNameToId(original.name),
moveFolderRequest=folders.FoldersMessages().MoveFolderRequest(
destinationParent=moved.parent))
self.mock_folders.Move.Expect(expected_move_request,
test_move_operation)
if not is_async:
test_move_operation_done = operations.OperationsMessages(
).Operation(done=True,
name=TEST_OPERATION_NAME,
response=response_value)
self.mock_operations.Get.Expect(
self.ExpectedGetOperationRequest(test_move_operation),
test_move_operation_done)
def ExpectedGetOperationRequest(self, create_operation):
expected_id = operations.OperationNameToId(create_operation.name)
return operations.OperationsMessages(
).CloudresourcemanagerOperationsGetRequest(operationsId=expected_id)
def ExpectedRequest(self):
messages = operations.OperationsMessages()
return messages.CloudresourcemanagerOperationsGetRequest(
operationsId=operations.OperationNameToId(
self.TEST_OPERATION.name))