51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
from datetime import datetime
|
|
|
|
import pytest
|
|
from marshmallow.exceptions import ValidationError
|
|
|
|
from lemur.common.utils import parse_private_key
|
|
from lemur.common.validators import verify_private_key_match
|
|
from lemur.tests.vectors import INTERMEDIATE_CERT, SAN_CERT, SAN_CERT_KEY
|
|
|
|
|
|
def test_private_key(session):
|
|
parse_private_key(SAN_CERT_KEY)
|
|
|
|
with pytest.raises(ValueError):
|
|
parse_private_key('invalid_private_key')
|
|
|
|
|
|
def test_validate_private_key(session):
|
|
key = parse_private_key(SAN_CERT_KEY)
|
|
|
|
verify_private_key_match(key, SAN_CERT)
|
|
|
|
with pytest.raises(ValidationError):
|
|
# Wrong key for certificate
|
|
verify_private_key_match(key, INTERMEDIATE_CERT)
|
|
|
|
|
|
def test_sub_alt_type(session):
|
|
from lemur.common.validators import sub_alt_type
|
|
|
|
with pytest.raises(ValidationError):
|
|
sub_alt_type('CNAME')
|
|
|
|
|
|
def test_dates(session):
|
|
from lemur.common.validators import dates
|
|
|
|
dates(dict(validity_start=datetime(2016, 1, 1), validity_end=datetime(2016, 1, 5)))
|
|
|
|
with pytest.raises(ValidationError):
|
|
dates(dict(validity_start=datetime(2016, 1, 1)))
|
|
|
|
with pytest.raises(ValidationError):
|
|
dates(dict(validity_end=datetime(2016, 1, 1)))
|
|
|
|
with pytest.raises(ValidationError):
|
|
dates(dict(validity_start=datetime(2016, 1, 5), validity_end=datetime(2016, 1, 1)))
|
|
|
|
with pytest.raises(ValidationError):
|
|
dates(dict(validity_start=datetime(2016, 1, 1), validity_end=datetime(2016, 1, 10)))
|