def revoke_cert(self) -> None:
"""
Revoke a certificate and emit an event whether it succeeded or not.
"""
if self.should_revoke_now():
try:
CertClient.revoke_cert(self.new_cert)
except Exception as exception:
Utils.fire_locust_failure(request_type=REQUEST_TYPE,
name=MESSAGE_TYPE_REVOKE,
response_time=0,
response_length=0,
exception=exception)
else:
if CertClient.has_been_revoked(self.new_cert):
Utils.fire_locust_success(request_type=REQUEST_TYPE,
name=MESSAGE_TYPE_REVOKE,
response_time=0,
response_length=0)
self.should_revoke = False
else:
Utils.fire_locust_failure(
request_type=REQUEST_TYPE,
name=MESSAGE_TYPE_REVOKE,
response_time=0,
response_length=0,
exception=CertRevogationError("failed to revoke"))
def test_revoke_cert(mock_thing, mock_crypto, mock_request):
"""
Test certifcate revoke cert
"""
mock_crypto.load_certificate.return_value = MagicMock()
mock_crypto.load_certificate(
).get_serial_number.return_value = 0x000111222333
serial_number = hex(
mock_crypto.load_certificate().get_serial_number())[2:]
url = MOCK_CONFIG["security"][
"ejbca_url"] + "/ca/CN={0},O=EJBCA/certificate/{1}".format(
MOCK_CONFIG["security"]["ejbca_ca_name"], serial_number)
CertClient.revoke_cert(mock_thing)
mock_request.delete.assert_called_with(url)
mock_thing.reset_mock()
mock_crypto.reset_mock()
mock_request.reset_mock()