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)