def credentials(mocker, now): gmd_mock = mocker.Mock(return_value='test') gmd_mock.range_keyname = 'test' mocker.patch( 'confidant.scripts.archive.Credential._get_meta_data', return_value=gmd_mock, ) mocker.patch( 'confidant.scripts.archive.CredentialArchive._get_meta_data', return_value=gmd_mock, ) credential = Credential( id='1234', name='test', data_type='credential', revision=2, enabled=True, modified_date=now, modified_by='*****@*****.**', ) credential.delete = mocker.Mock() archive_credential = CredentialArchive.from_credential(credential) revision1 = Credential( id='1234-1', name='test revision1', data_type='archive-credential', revision=1, enabled=True, modified_date=now, modified_by='*****@*****.**', ) revision1.delete = mocker.Mock() archive_revision1 = CredentialArchive.from_credential(revision1) revision2 = Credential( id='1234-2', name='test revision2', data_type='archive-credential', revision=2, enabled=True, modified_date=now, modified_by='*****@*****.**', ) revision2.delete = mocker.Mock() archive_revision2 = CredentialArchive.from_credential(revision2) def from_credential(credential): if credential.id == '1234': return archive_credential elif credential.id == '1234-1': return archive_revision1 elif credential.id == '1234-2': return archive_revision2 mocker.patch.object(CredentialArchive, 'from_credential', from_credential) return { 'credentials': [credential], 'archive_credentials': [archive_credential], 'revisions': [revision1, revision2], 'archive_revisions': [archive_revision1, archive_revision2], }