def test_configure_mqtt(self, mock_cert_utils, mock_paho):
"""
Should configure the MQTT connection.
"""
client = MQTTClient("123", "987", False, True)
# configure_mqtt() function is called at the constructor, so
# we need to reset the mocks to be able to test it
mock_cert_utils.reset_mock()
mock_paho.reset_mock()
client.configure_mqtt()
mock_cert_utils.get_certificate_file.assert_called_once_with(
client.device_id)
mock_cert_utils.get_private_key_file.assert_called_once_with(
client.device_id)
self.assertIsNotNone(client.mqttc)
mock_paho.Client.assert_called_once_with(client_id=client.device_id)
client.mqttc.reconnect_delay_set.assert_called_once_with(
min_delay=MOCK_CONFIG["security"]["min_time_reconn"],
max_delay=MOCK_CONFIG["security"]["max_time_reconn"])
client.mqttc.tls_set.assert_called_once()
client.mqttc.tls_insecure_set.assert_called_once_with(True)
self.assertIsNotNone(client.mqttc.on_connect)
self.assertIsNotNone(client.mqttc.on_disconnect)
self.assertIsNotNone(client.mqttc.on_publish)
self.assertIsNotNone(client.mqttc.on_subscribe)
self.assertIsNotNone(client.mqttc.on_message)
def test_configure_mqtt(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"""
Should configure the MQTT connection.
"""
client = MQTTClient("123", "987", False, True)
client.configure_mqtt()
mock_cert_client.get_certificate_file.assert_called_once_with(
client.device_id)
mock_cert_client.get_private_key_file.assert_called_once_with(
client.device_id)
self.assertIsNotNone(client.mqttc)
paho_mqtt_mock.Client.assert_called_once_with(
client_id=client.device_id)
paho_mqtt_mock.Client().reconnect_delay_set.assert_called_once_with(
min_delay=MOCK_CONFIG["security"]["min_time_reconn"],
max_delay=MOCK_CONFIG["security"]["max_time_reconn"])
paho_mqtt_mock.Client().tls_set.assert_called_once()
paho_mqtt_mock.Client().tls_insecure_set.assert_called_once_with(True)
self.assertIsNotNone(client.mqttc.on_connect)
self.assertIsNotNone(client.mqttc.on_disconnect)
self.assertIsNotNone(client.mqttc.on_publish)
self.assertIsNotNone(client.mqttc.on_subscribe)
self.assertIsNotNone(client.mqttc.on_message)
mock_utils.reset_mock()
mock_logging.reset_mock()
mock_cert_client.reset_mock()
paho_mqtt_mock.reset_mock()
json_mock.reset_mock()
def test_setup_success(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"""
Should setup the parameters successfully.
"""
client = MQTTClient("123", "987", False, False)
client.configure_mqtt = MagicMock()
client.setup()
mock_logging.basicConfig.assert_called_once()
client.configure_mqtt.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_setup_success_renew(self, mock_utils,
mock_logging,
mock_cert_client,
paho_mqtt_mock,
json_mock):
"""
Should setup the parameters successfully with certificate renovation.
"""
client = MQTTClient("123", "987", False, True)
client.configure_mqtt = MagicMock()
client.setup()
mock_cert_client.new_cert.assert_called_once_with(
client.tenant, client.device_id)
mock_cert_client.create_cert_files.assert_called_once_with(
client.new_cert, client.device_cert_dir)
mock_utils.reset_mock()
mock_logging.reset_mock()
mock_cert_client.reset_mock()
paho_mqtt_mock.reset_mock()
json_mock.reset_mock()
