updated regex

This commit is contained in:
Brian Duffy
2018-02-16 16:06:33 +00:00
parent 6b5a6e9226
commit 235213bd3b
2 changed files with 13 additions and 4 deletions

View File

@@ -26,7 +26,7 @@ def test_invalid_keys():
"-----BEGIN FOO-----foobar---END FOO----",
"----- BEGIN FOO ----- foobar ----- FAIL FOO ----",
"----- FAIL FOO ----- foobar ----- END FOO ----",
"----BEGIN FOO----foobar----END FOO--------BEGIN FOO----foobar----END FOO----"
"----BEGIN FOO----foobar----END BAR----"
]
for invalid_key in invalid_keys:
with pytest.raises(ValidationError):
@@ -37,6 +37,16 @@ def test_invalid_keys():
validate_ssh_private_key(invalid_key)
def test_invalid_rsa_key():
invalid_key = TEST_SSH_KEY_DATA.replace('-----END', '----END')
with pytest.raises(ValidationError):
validate_private_key(invalid_key)
with pytest.raises(ValidationError):
validate_certificate(invalid_key)
with pytest.raises(ValidationError):
validate_ssh_private_key(invalid_key)
def test_valid_catted_rsa_key():
valid_key = TEST_CATTED_SSH_KEY_DATA
pem_objects = validate_private_key(valid_key)
@@ -48,7 +58,6 @@ def test_valid_catted_rsa_key():
assert pem_objects[0]['key_type'] == 'rsa'
assert not pem_objects[0]['key_enc']
def test_valid_rsa_key():
valid_key = TEST_SSH_KEY_DATA
pem_objects = validate_private_key(valid_key)