def test_not_revoke_cert(self, _mock_paho, _mock_cert_utils):
"Should not revoke cert"
client = MQTTClient("123", "987", False, False)
client.should_revoke_now = MagicMock()
client.should_revoke_now.return_value = False
client.revoke_cert_and_emit_event = MagicMock()
client.revoke_cert()
self.assertFalse(client.should_revoke)
client.should_revoke_now.assert_called()
self.assertTrue(client.revoke_cert_and_emit_event.call_count == 0)
def test_revoke_cert_and_emit_event(self, mock_utils, _mock_paho,
mock_cert_utils):
"""
Should revoke cert
"""
client = MQTTClient("123", "987", True, False)
mock_cert_utils.has_been_revoked.return_value = False
has_revoked = client.revoke_cert_and_emit_event()
self.assertTrue(has_revoked)
mock_cert_utils.revoke_cert.assert_called_once()
mock_utils.fire_locust_success.assert_called()
def test_not_revoke_cert_and_emit_event(self, mock_utils, _mock_paho,
mock_cert_utils):
"""
Should not revoke cert - Exception thrown by the revoking function
"""
client = MQTTClient("123", "987", True, False)
mock_cert_utils.has_been_revoked.return_value = False
mock_cert_utils.revoke_cert.side_effect = Exception
has_revoked = client.revoke_cert_and_emit_event()
self.assertFalse(has_revoked)
mock_cert_utils.revoke_cert.assert_called_once()
mock_utils.fire_locust_failure.assert_called()
def test_not_renew_cert_and_emit_event(self, mock_utils, _mock_paho,
mock_cert_utils):
"Should not renew cert"
client = MQTTClient("123", "987", False, True)
client.revoke_cert_and_emit_event = MagicMock()
client.revoke_cert_and_emit_event.return_value = False
mock_cert_utils.create_cert_files = MagicMock()
has_renewed = client.renew_cert_and_emit_event()
client.revoke_cert_and_emit_event.assert_called_once()
self.assertTrue(client.new_cert.renew_cert.call_count == 0)
self.assertTrue(mock_cert_utils.create_cert_files.call_count == 0)
mock_utils.fire_locust_failure.assert_called()
self.assertFalse(has_renewed)
def test_renew_cert_and_emit_event(self, mock_utils, _mock_paho,
mock_cert_utils):
"Should renew cert"
client = MQTTClient("123", "987", False, True)
client.revoke_cert_and_emit_event = MagicMock()
client.revoke_cert_and_emit_event.return_value = True
mock_cert_utils.create_cert_files = MagicMock()
has_renewed = client.renew_cert_and_emit_event()
client.revoke_cert_and_emit_event.assert_called_once()
client.new_cert.renew_cert.assert_called_once()
mock_cert_utils.create_cert_files.assert_called_once()
mock_utils.fire_locust_success.assert_called_once()
self.assertTrue(has_renewed)
def test_renew_cert_and_emit_event_exception(self, mock_utils, _mock_paho,
mock_cert_utils):
"Should raise an exception when renewing cert"
client = MQTTClient("123", "987", False, True)
client.revoke_cert_and_emit_event = MagicMock()
client.revoke_cert_and_emit_event.return_value = True
mock_cert_utils.create_cert_files = MagicMock()
client.new_cert.renew_cert.side_effect = Exception
has_renewed = client.renew_cert_and_emit_event()
client.revoke_cert_and_emit_event.assert_called_once()
self.assertRaises(Exception, client.new_cert.renew_cert)
self.assertTrue(mock_cert_utils.create_cert_files.call_count == 0)
mock_utils.fire_locust_failure.assert_called()
self.assertFalse(has_renewed)
