def test_get_or_increase_name(session, certificate): from lemur.certificates.models import get_or_increase_name from lemur.tests.factories import CertificateFactory serial = 'AFF2DB4F8D2D4D8E80FA382AE27C2333' assert get_or_increase_name(certificate.name, certificate.serial) == '{0}-{1}'.format( certificate.name, serial) certificate.name = 'test-cert-11111111' assert get_or_increase_name( certificate.name, certificate.serial) == 'test-cert-11111111-' + serial certificate.name = 'test-cert-11111111-1' assert get_or_increase_name( 'test-cert-11111111-1', certificate.serial) == 'test-cert-11111111-1-' + serial cert2 = CertificateFactory(name='certificate1-' + serial) session.commit() assert get_or_increase_name('certificate1', int(serial, 16)) == 'certificate1-{}-1'.format(serial)
def test_get_or_increase_name(session, certificate): from lemur.certificates.models import get_or_increase_name from lemur.tests.factories import CertificateFactory serial = "AFF2DB4F8D2D4D8E80FA382AE27C2333" assert get_or_increase_name(certificate.name, certificate.serial) == "{0}-{1}".format( certificate.name, serial) certificate.name = "test-cert-11111111" assert (get_or_increase_name(certificate.name, certificate.serial) == "test-cert-11111111-" + serial) certificate.name = "test-cert-11111111-1" assert (get_or_increase_name( "test-cert-11111111-1", certificate.serial) == "test-cert-11111111-1-" + serial) cert2 = CertificateFactory(name="certificate1-" + serial) session.commit() assert get_or_increase_name("certificate1", int(serial, 16)) == "certificate1-{}-1".format(serial)
def test_get_or_increase_name(session, certificate): from lemur.certificates.models import get_or_increase_name assert get_or_increase_name(certificate.name) == '{0}-1'.format(certificate.name) certificate.name = 'test-cert-11111111' assert get_or_increase_name(certificate.name) == 'test-cert-11111111-1' certificate.name = 'test-cert-11111111-1' assert get_or_increase_name('test-cert-11111111-1') == 'test-cert-11111111-2'
def test_get_or_increase_name(session, certificate): from lemur.certificates.models import get_or_increase_name assert get_or_increase_name('test name') == 'test-name' assert get_or_increase_name(certificate.name) == '{0}-1'.format(certificate.name) certificate.name = 'test-cert-11111111' assert get_or_increase_name(certificate.name) == 'test-cert-11111111-1' certificate.name = 'test-cert-11111111-1' assert get_or_increase_name('test-cert-11111111-1') == 'test-cert-11111111-2'
def test_get_or_increase_name(session, certificate): from lemur.certificates.models import get_or_increase_name from lemur.tests.factories import CertificateFactory assert get_or_increase_name(certificate.name, certificate.serial) == '{0}-3E9'.format(certificate.name) certificate.name = 'test-cert-11111111' assert get_or_increase_name(certificate.name, certificate.serial) == 'test-cert-11111111-3E9' certificate.name = 'test-cert-11111111-1' assert get_or_increase_name('test-cert-11111111-1', certificate.serial) == 'test-cert-11111111-1-3E9' cert2 = CertificateFactory(name='certificate1-3E9') session.commit() assert get_or_increase_name('certificate1', 1001) == 'certificate1-3E9-1'
def test_get_or_increase_name(session, certificate): from lemur.certificates.models import get_or_increase_name from lemur.tests.factories import CertificateFactory serial = 'AFF2DB4F8D2D4D8E80FA382AE27C2333' assert get_or_increase_name(certificate.name, certificate.serial) == '{0}-{1}'.format(certificate.name, serial) certificate.name = 'test-cert-11111111' assert get_or_increase_name(certificate.name, certificate.serial) == 'test-cert-11111111-' + serial certificate.name = 'test-cert-11111111-1' assert get_or_increase_name('test-cert-11111111-1', certificate.serial) == 'test-cert-11111111-1-' + serial cert2 = CertificateFactory(name='certificate1-' + serial) session.commit() assert get_or_increase_name('certificate1', int(serial, 16)) == 'certificate1-{}-1'.format(serial)
def __init__(self, **kwargs): self.csr = kwargs.get('csr') self.private_key = kwargs.get('private_key', "") if self.private_key: # If the request does not send private key, the key exists but the value is None self.private_key = self.private_key.strip() self.external_id = kwargs.get('external_id') # when destinations are appended they require a valid name. if kwargs.get('name'): self.name = get_or_increase_name( defaults.text_to_slug(kwargs['name']), 0) self.rename = False else: # TODO: Fix auto-generated name, it should be renamed on creation self.name = get_or_increase_name( defaults.certificate_name(kwargs['common_name'], kwargs['authority'].name, dt.now(), dt.now(), False), self.external_id) self.rename = True self.cn = defaults.common_name(utils.parse_csr(self.csr)) self.owner = kwargs['owner'] self.number_attempts = 0 if kwargs.get('chain'): self.chain = kwargs['chain'].strip() self.notify = kwargs.get('notify', True) self.destinations = kwargs.get('destinations', []) self.notifications = kwargs.get('notifications', []) self.description = kwargs.get('description') self.roles = list(set(kwargs.get('roles', []))) self.replaces = kwargs.get('replaces', []) self.rotation = kwargs.get('rotation') self.rotation_policy = kwargs.get('rotation_policy') try: self.dns_provider_id = kwargs.get('dns_provider').id except (AttributeError, KeyError, TypeError, Exception): pass