diff --git a/azure-iot-device/azure/iot/device/iothub/abstract_clients.py b/azure-iot-device/azure/iot/device/iothub/abstract_clients.py index 0cd5a8137..a867dd614 100644 --- a/azure-iot-device/azure/iot/device/iothub/abstract_clients.py +++ b/azure-iot-device/azure/iot/device/iothub/abstract_clients.py @@ -231,7 +231,10 @@ def _replace_user_supplied_sastoken(self, sastoken_str: str) -> None: raise ValueError("Provided SasToken is for a device") if self._mqtt_pipeline.pipeline_configuration.device_id != vals["device_id"]: raise ValueError("Provided SasToken does not match existing device id") - if vals["module_id"] != "" and self._mqtt_pipeline.pipeline_configuration.module_id != vals["module_id"]: + if ( + vals["module_id"] != "" + and self._mqtt_pipeline.pipeline_configuration.module_id != vals["module_id"] + ): raise ValueError("Provided SasToken does not match existing module id") if self._mqtt_pipeline.pipeline_configuration.hostname != vals["hostname"]: raise ValueError("Provided SasToken does not match existing hostname") @@ -433,9 +436,7 @@ def receive_method_request(self, method_name: Optional[str] = None) -> None: pass @abc.abstractmethod - def send_method_response( - self, method_response: MethodResponse - ) -> None: + def send_method_response(self, method_response: MethodResponse) -> None: pass @abc.abstractmethod @@ -607,7 +608,6 @@ def create_from_x509_certificate( device_id=device_id, hostname=hostname, x509=x509, **config_kwargs ) pipeline_configuration.blob_upload = True # Blob Upload is a feature on Device Clients - pipeline_configuration.ensure_desired_properties = True # Pipeline setup http_pipeline = pipeline.HTTPPipeline(pipeline_configuration) @@ -680,7 +680,6 @@ def create_from_symmetric_key( device_id=device_id, hostname=hostname, sastoken=sastoken, **config_kwargs ) pipeline_configuration.blob_upload = True # Blob Upload is a feature on Device Clients - pipeline_configuration.ensure_desired_properties = True # Pipeline setup http_pipeline = pipeline.HTTPPipeline(pipeline_configuration) @@ -844,8 +843,6 @@ def create_from_edge_environment(cls, **kwargs) -> Self: server_verification_cert=server_verification_cert, **config_kwargs, ) - pipeline_configuration.ensure_desired_properties = True - pipeline_configuration.method_invoke = ( True # Method Invoke is allowed on modules created from edge environment ) @@ -912,7 +909,6 @@ def create_from_x509_certificate( pipeline_configuration = pipeline.IoTHubPipelineConfig( device_id=device_id, module_id=module_id, hostname=hostname, x509=x509, **config_kwargs ) - pipeline_configuration.ensure_desired_properties = True # Pipeline setup http_pipeline = pipeline.HTTPPipeline(pipeline_configuration) diff --git a/requirements_test.txt b/requirements_test.txt index d9b6c8163..ea6d086a5 100644 --- a/requirements_test.txt +++ b/requirements_test.txt @@ -1,4 +1,4 @@ -pytest +pytest < 8.0.0 pytest-asyncio <= 0.16 pytest-mock pytest-testdox>=1.1.1 diff --git a/tests/unit/iothub/shared_client_tests.py b/tests/unit/iothub/shared_client_tests.py index 1e8f893ba..1c0780b6c 100644 --- a/tests/unit/iothub/shared_client_tests.py +++ b/tests/unit/iothub/shared_client_tests.py @@ -165,6 +165,7 @@ def test_product_info_option( @pytest.mark.it( "Sets the 'ensure_desired_properties' user option parameter on the PipelineConfig, if provided" ) + @pytest.mark.parametrize("edp_value", [True, False]) def test_ensure_desired_properties_option( self, option_test_required_patching, @@ -172,9 +173,10 @@ def test_ensure_desired_properties_option( create_method_args, mock_mqtt_pipeline_init, mock_http_pipeline_init, + edp_value, ): - client_create_method(*create_method_args, ensure_desired_properties=True) + client_create_method(*create_method_args, ensure_desired_properties=edp_value) # Get configuration object, and ensure it was used for both protocol pipelines assert mock_mqtt_pipeline_init.call_count == 1 @@ -182,11 +184,12 @@ def test_ensure_desired_properties_option( assert isinstance(config, IoTHubPipelineConfig) assert config == mock_http_pipeline_init.call_args[0][0] - assert config.ensure_desired_properties is True + assert config.ensure_desired_properties is edp_value @pytest.mark.it( "Sets the 'websockets' user option parameter on the PipelineConfig, if provided" ) + @pytest.mark.parametrize("ws_value", [True, False]) def test_websockets_option( self, option_test_required_patching, @@ -194,9 +197,10 @@ def test_websockets_option( create_method_args, mock_mqtt_pipeline_init, mock_http_pipeline_init, + ws_value, ): - client_create_method(*create_method_args, websockets=True) + client_create_method(*create_method_args, websockets=ws_value) # Get configuration object, and ensure it was used for both protocol pipelines assert mock_mqtt_pipeline_init.call_count == 1 @@ -204,7 +208,7 @@ def test_websockets_option( assert isinstance(config, IoTHubPipelineConfig) assert config == mock_http_pipeline_init.call_args[0][0] - assert config.websockets + assert config.websockets is ws_value # TODO: Show that input in the wrong format is formatted to the correct one. This test exists # in the IoTHubPipelineConfig object already, but we do not currently show that this is felt @@ -320,6 +324,7 @@ def test_keep_alive_options( @pytest.mark.it( "Sets the 'auto_connect' user option parameter on the PipelineConfig, if provided" ) + @pytest.mark.parametrize("auto_connect_value", [True, False]) def test_auto_connect_option( self, option_test_required_patching, @@ -327,8 +332,8 @@ def test_auto_connect_option( create_method_args, mock_mqtt_pipeline_init, mock_http_pipeline_init, + auto_connect_value, ): - auto_connect_value = False client_create_method(*create_method_args, auto_connect=auto_connect_value) # Get configuration object, and ensure it was used for both protocol pipelines @@ -342,6 +347,7 @@ def test_auto_connect_option( @pytest.mark.it( "Sets the 'connection_retry' user option parameter on the PipelineConfig, if provided" ) + @pytest.mark.parametrize("connection_retry_value", [True, False]) def test_connection_retry_option( self, option_test_required_patching, @@ -349,6 +355,7 @@ def test_connection_retry_option( create_method_args, mock_mqtt_pipeline_init, mock_http_pipeline_init, + connection_retry_value, ): connection_retry_value = False client_create_method(*create_method_args, connection_retry=connection_retry_value)