def test_not_renew_cert(self, _mock_paho, _mock_cert_utils): "Should not renew cert" client = MQTTClient("123", "987", False, False) client.should_renew_now = MagicMock() client.should_renew_now.return_value = False client.renew_cert_and_emit_event = MagicMock() client.renew_cert() self.assertFalse(client.should_renew) client.should_renew_now.assert_called() self.assertTrue(client.renew_cert_and_emit_event.call_count == 0)
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)