66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
import arrow
|
|
|
|
|
|
def test_build_certificate_authority():
|
|
from lemur.plugins.lemur_cryptography.plugin import build_certificate_authority
|
|
|
|
options = {
|
|
"key_type": "RSA2048",
|
|
"country": "US",
|
|
"state": "CA",
|
|
"location": "Example place",
|
|
"organization": "Example, Inc.",
|
|
"organizational_unit": "Example Unit",
|
|
"common_name": "Example ROOT",
|
|
"validity_start": arrow.get("2016-12-01").datetime,
|
|
"validity_end": arrow.get("2016-12-02").datetime,
|
|
"first_serial": 1,
|
|
"serial_number": 1,
|
|
"owner": "owner@example.com",
|
|
}
|
|
cert_pem, private_key_pem, chain_cert_pem = build_certificate_authority(options)
|
|
|
|
assert cert_pem
|
|
assert private_key_pem
|
|
assert chain_cert_pem == ""
|
|
|
|
|
|
def test_build_intermediate_certificate_authority(authority):
|
|
from lemur.plugins.lemur_cryptography.plugin import build_certificate_authority
|
|
|
|
options = {
|
|
"key_type": "RSA2048",
|
|
"country": "US",
|
|
"state": "CA",
|
|
"location": "Example place",
|
|
"organization": "Example, Inc.",
|
|
"organizational_unit": "Example Unit",
|
|
"common_name": "Example INTERMEDIATE",
|
|
"validity_start": arrow.get("2016-12-01").datetime,
|
|
"validity_end": arrow.get("2016-12-02").datetime,
|
|
"first_serial": 1,
|
|
"serial_number": 1,
|
|
"owner": "owner@example.com",
|
|
"parent": authority
|
|
}
|
|
cert_pem, private_key_pem, chain_cert_pem = build_certificate_authority(options)
|
|
|
|
assert cert_pem
|
|
assert private_key_pem
|
|
assert chain_cert_pem == authority.authority_certificate.body
|
|
|
|
|
|
def test_issue_certificate(authority):
|
|
from lemur.tests.vectors import CSR_STR
|
|
from lemur.plugins.lemur_cryptography.plugin import issue_certificate
|
|
|
|
options = {
|
|
"common_name": "Example.com",
|
|
"authority": authority,
|
|
"validity_start": arrow.get("2016-12-01").datetime,
|
|
"validity_end": arrow.get("2016-12-02").datetime,
|
|
}
|
|
cert_pem, chain_cert_pem = issue_certificate(CSR_STR, options)
|
|
assert cert_pem
|
|
assert chain_cert_pem
|