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)