def test_revoke_cert(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"Sould not revoke cert"
client = MQTTClient("123", "987", True, False)
client.should_revoke_now = MagicMock()
client.should_revoke_now.return_value = True
mock_cert_client.revoke_cert.return_value = MagicMock()
mock_cert_client.has_been_revoked.return_value = True
client.setup()
client.revoke_cert()
mock_cert_client.revoke_cert.assert_called_once()
mock_cert_client.has_been_revoked.assert_called_once()
mock_utils.fire_locust_success.assert_called_once()
mock_cert_client.has_been_revoked.return_value = False
client.revoke_cert()
mock_utils.fire_locust_success.assert_called_once()
mock_utils.reset_mock()
mock_logging.reset_mock()
mock_cert_client.reset_mock()
paho_mqtt_mock.reset_mock()
json_mock.reset_mock()
def test_not_revoke_cert(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"Sould not revoke cert"
client = MQTTClient("123", "987", True, False)
client.should_revoke_now = MagicMock()
client.should_revoke_now.return_value = True
mock_cert_client.revoke_cert = None
client.setup()
with self.assertRaises(Exception):
client.revoke_cert()
mock_utils.fire_locust_failure.assert_called_once()
client.should_revoke_now.return_value = False
client.revoke_cert()
mock_utils.reset_mock()
mock_logging.reset_mock()
mock_cert_client.reset_mock()
paho_mqtt_mock.reset_mock()
json_mock.reset_mock()
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)
