Fix subscriber 2

awslabs · Dec 29, 2023 · dbd8b59 · dbd8b59
1 parent 999ee60
commit dbd8b59
Showing 1 changed file with 11 additions and 6 deletions.
diff --git a/test/test_mqtt5.py b/test/test_mqtt5.py
@@ -1085,17 +1085,22 @@ def test_operation_shared_subscription(self):
 
         # subscriber 2
         connect_subscriber2_options = mqtt5.ConnectPacket(client_id=client_id_subscriber2)
+        subscriber2_generic_callback = Mqtt5TestCallbacks()
 
-        connect_subscriber2_options.tls_ctx = io.ClientTlsContext(tls_ctx_options)
         subscriber2_options = mqtt5.ClientOptions(
             host_name=input_host_name,
             port=8883,
+            tls_ctx=io.ClientTlsContext(tls_ctx_options),
             connect_options=connect_subscriber2_options,
             #session_behavior=mqtt5.ClientSessionBehaviorType.CLEAN,
-        )
-        subscriber2_options.on_publish_callback_fn = self.subscriber2_callback
-        subscriber2_callback = Mqtt5TestCallbacks()
-        subscriber2_options.on_lifecycle_connection_success_fn = subscriber2_callback.on_lifecycle_connection_success
+            on_lifecycle_event_connection_success_fn=subscriber2_generic_callback.on_lifecycle_connection_success,
+            on_publish_callback_fn=self.subscriber2_callback,
+            on_lifecycle_event_stopped_fn=subscriber2_generic_callback.on_lifecycle_stopped,
+            on_lifecycle_event_attempting_connect_fn=subscriber2_generic_callback.on_lifecycle_attempting_connect
+        )
+        #connect_subscriber2_options.tls_ctx = io.ClientTlsContext(tls_ctx_options)
+        #subscriber2_options.on_publish_callback_fn = self.subscriber2_callback
+        #subscriber2_options.on_lifecycle_connection_success_fn = subscriber2_generic_callback.on_lifecycle_connection_success
         #subscriber2_client = self._create_client(client_options=subscriber2_options, callbacks=subscriber2_callback)
         subscriber2_client = mqtt5.Client(client_options=subscriber2_options)
 
@@ -1119,7 +1124,7 @@ def test_operation_shared_subscription(self):
         subscriber1_generic_callback.future_connection_success.result(TIMEOUT)
 
         subscriber2_client.start()
-        subscriber2_callback.future_connection_success.result(TIMEOUT)
+        subscriber2_generic_callback.future_connection_success.result(TIMEOUT)
 
         publisher_client.start()
         publisher_callback.future_connection_success.result(TIMEOUT)