Update test_streaming_client_stream_decryptor.py

Author: RitvikKapila

test/unit/test_streaming_client_stream_decryptor.py

@@ -365,143 +365,7 @@ def test_GIVEN_verification_key_AND_has_mpl_AND_has_MPLCMM_WHEN_read_header_THEN
             algorithm=self.mock_header.algorithm, encoded_point=mock_b64encoding()
         )
 
-    @patch("aws_encryption_sdk.streaming_client.derive_data_encryption_key")
-    @patch("aws_encryption_sdk.streaming_client.DecryptionMaterialsRequest")
-    @patch("aws_encryption_sdk.streaming_client.Verifier")
-    # Given: no MPL
-    @pytest.mark.skipif(HAS_MPL, reason="Test should only be executed without MPL in installation")
-    def test_GIVEN_decrypt_config_has_ec_WHEN_read_header_THEN_calls_decrypt_materials_with_reproduced_ec(
-        self,
-        mock_verifier,
-        mock_decrypt_materials_request,
-        *_,
-    ):
-
-        mock_verifier_instance = MagicMock()
-        mock_verifier.from_key_bytes.return_value = mock_verifier_instance
-        ct_stream = io.BytesIO(VALUES["data_128"])
-        mock_commitment_policy = MagicMock(__class__=CommitmentPolicy)
-        test_decryptor = StreamDecryptor(
-            materials_manager=self.mock_materials_manager,
-            source=ct_stream,
-            commitment_policy=mock_commitment_policy,
-        )
-        test_decryptor.source_stream = ct_stream
-        test_decryptor._stream_length = len(VALUES["data_128"])
-        # Given: self.config has "encryption_context"
-        any_reproduced_ec = {"some": "ec"}
-        test_decryptor.config.encryption_context = any_reproduced_ec
-
-        # When: read header
-        test_decryptor._read_header()
-
-        # Then: calls decrypt_materials with reproduced_encryption_context
-        mock_decrypt_materials_request.assert_called_once_with(
-            encrypted_data_keys=self.mock_header.encrypted_data_keys,
-            algorithm=self.mock_header.algorithm,
-            encryption_context=sentinel.encryption_context,
-            commitment_policy=mock_commitment_policy,
-            reproduced_encryption_context=any_reproduced_ec,
-        )
-
-    @patch("aws_encryption_sdk.streaming_client.DecryptionMaterialsRequest")
-    @patch("aws_encryption_sdk.streaming_client.derive_data_encryption_key")
-    @patch("aws_encryption_sdk.streaming_client.Verifier")
-    # Given: no MPL
-    @pytest.mark.skipif(HAS_MPL, reason="Test should only be executed without MPL in installation")
-    def test_GIVEN_verification_key_AND_no_mpl_WHEN_read_header_THEN_calls_from_key_bytes(
-        self,
-        mock_verifier,
-        *_,
-    ):
-        # Given: verification key
-        mock_verifier_instance = MagicMock()
-        mock_verifier.from_key_bytes.return_value = mock_verifier_instance
-        ct_stream = io.BytesIO(VALUES["data_128"])
-        mock_commitment_policy = MagicMock(__class__=CommitmentPolicy)
-        test_decryptor = StreamDecryptor(
-            materials_manager=self.mock_materials_manager,
-            source=ct_stream,
-            commitment_policy=mock_commitment_policy,
-        )
-        test_decryptor.source_stream = ct_stream
-        test_decryptor._stream_length = len(VALUES["data_128"])
-
-        # When: read header
-        test_decryptor._read_header()
-
-        # Then: calls from_key_bytes
-        mock_verifier.from_key_bytes.assert_called_once_with(
-            algorithm=self.mock_header.algorithm, key_bytes=sentinel.verification_key
-        )
-
-    @patch("aws_encryption_sdk.streaming_client.DecryptionMaterialsRequest")
-    @patch("aws_encryption_sdk.streaming_client.derive_data_encryption_key")
-    @patch("aws_encryption_sdk.streaming_client.Verifier")
-    # Given: has MPL
-    @pytest.mark.skipif(not HAS_MPL, reason="Test should only be executed with MPL in installation")
-    def test_GIVEN_verification_key_AND_has_mpl_AND_not_MPLCMM_WHEN_read_header_THEN_calls_from_key_bytes(
-        self,
-        mock_verifier,
-        *_,
-    ):
-        # Given: verification key
-        mock_verifier_instance = MagicMock()
-        mock_verifier.from_key_bytes.return_value = mock_verifier_instance
-        ct_stream = io.BytesIO(VALUES["data_128"])
-        mock_commitment_policy = MagicMock(__class__=CommitmentPolicy)
-        test_decryptor = StreamDecryptor(
-            # Given: native CMM
-            materials_manager=self.mock_materials_manager,
-            source=ct_stream,
-            commitment_policy=mock_commitment_policy,
-        )
-        test_decryptor.source_stream = ct_stream
-        test_decryptor._stream_length = len(VALUES["data_128"])
-
-        # When: read_header
-        test_decryptor._read_header()
-
-        # Then: calls from_key_bytess
-        mock_verifier.from_key_bytes.assert_called_once_with(
-            algorithm=self.mock_header.algorithm, key_bytes=sentinel.verification_key
-        )
-
-    @patch("aws_encryption_sdk.streaming_client.DecryptionMaterialsRequest")
-    @patch("aws_encryption_sdk.streaming_client.derive_data_encryption_key")
-    @patch("aws_encryption_sdk.streaming_client.Verifier")
-    @patch("base64.b64encode")
-    # Given: has MPL
-    @pytest.mark.skipif(not HAS_MPL, reason="Test should only be executed with MPL in installation")
-    def test_GIVEN_verification_key_AND_has_mpl_AND_has_MPLCMM_WHEN_read_header_THEN_calls_from_encoded_point(
-        self,
-        mock_b64encoding,
-        mock_verifier,
-        *_,
-    ):
-        # Given: Verification key
-        mock_verifier_instance = MagicMock()
-        mock_verifier.from_key_bytes.return_value = mock_verifier_instance
-        ct_stream = io.BytesIO(VALUES["data_128"])
-        mock_commitment_policy = MagicMock(__class__=CommitmentPolicy)
-        test_decryptor = StreamDecryptor(
-            # Given: MPL CMM
-            materials_manager=self.mock_mpl_materials_manager,
-            source=ct_stream,
-            commitment_policy=mock_commitment_policy,
-        )
-        test_decryptor.source_stream = ct_stream
-        test_decryptor._stream_length = len(VALUES["data_128"])
-
-        # When: read header
-        test_decryptor._read_header()
-
-        # Then: calls from_encoded_point
-        mock_verifier.from_encoded_point.assert_called_once_with(
-            algorithm=self.mock_header.algorithm, encoded_point=mock_b64encoding()
-        )
-
-    @patch("aws_encryption_sdk.streaming_client.derive_data_encryption_key")
+        @patch("aws_encryption_sdk.streaming_client.derive_data_encryption_key")
     def test_read_header_frame_too_large(self, mock_derive_datakey):
         self.mock_header.content_type = ContentType.FRAMED_DATA
         self.mock_header.frame_length = 1024