def configure_mqtt(self) -> None:
"""
Configures the MQTT connection.
"""
# Certification files
cert_dir = CONFIG["security"]["cert_dir"]
ca_cert_file = cert_dir + CONFIG["security"]["ca_cert_file"]
cert_file = self.device_cert_dir + CertClient.get_certificate_file(
self.device_id)
key_file = self.device_cert_dir + CertClient.get_private_key_file(
self.device_id)
# Configuring MQTT client
self.mqttc = mqtt.Client(client_id=self.device_id)
# Sets exponential reconnect delay
self.mqttc.reconnect_delay_set(
min_delay=CONFIG["security"]["min_time_reconn"],
max_delay=CONFIG["security"]["max_time_reconn"])
# Setting up TLS
self.mqttc.tls_set(ca_cert_file, cert_file, key_file)
# TODO: investigate the problem when the insecure TLS mode is False
# This problem seems to happen because the TLS implementation does not
# expects an IP, but a hostname
self.mqttc.tls_insecure_set(True)
# Registering MQTT client callbacks
self.mqttc.on_connect = self.locust_on_connect
self.mqttc.on_disconnect = self.locust_on_disconnect
self.mqttc.on_publish = self.locust_on_publish
self.mqttc.on_subscribe = self.locust_on_subscribe
self.mqttc.on_message = self.locust_on_message
def test_get_certificate_file(self, mock_utils):
"""
Should build a correct filename for the certificate.
"""
mock_utils.validate_device_id.return_value = MagicMock()
device_id = "dev-id"
filename = device_id + ".crt"
self.assertEqual(CertClient.get_certificate_file(device_id), filename)
mock_utils.reset_mock()
def initialize_mqtt(self) -> None:
"""
Initializes the MQTT connection.
"""
# Certification files
cert_dir = CONFIG["security"]["cert_dir"]
ca_cert_file = cert_dir + CONFIG["security"]["ca_cert_file"]
cert_file = self.device_cert_dir + CertClient.get_certificate_file(
self.device_id)
key_file = self.device_cert_dir + CertClient.get_private_key_file(
self.device_id)
self.username = '******'.format(self.tenant, self.device_id)
self.topic = "{0}/attrs".format(self.username)
self.sub_topic = "{0}/config".format(self.username)
self.log = LogController(self.run_id)
# Configuring MQTT client
self.mqttc = mqtt.Client(client_id=self.device_id)
# Sets exponential reconnect delay
self.mqttc.reconnect_delay_set(
min_delay=CONFIG["security"]["min_time_reconn"],
max_delay=CONFIG["security"]["max_time_reconn"])
# Setting up TLS
self.mqttc.tls_set(ca_cert_file, cert_file, key_file)
# TODO: investigate the problem when the insecure TLS mode is False
self.mqttc.tls_insecure_set(True)
# Registering MQTT client callbacks
self.mqttc.on_connect = self.locust_on_connect
self.mqttc.on_disconnect = self.locust_on_disconnect
self.mqttc.on_publish = self.locust_on_publish
self.mqttc.on_subscribe = self.locust_on_subscribe
self.mqttc.on_message = self.locust_on_message