Esempio n. 1
0
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],
    }