Skip to content

Migrate "test/unit/test_streaming_client_stream_decryptor.py" from unittest to pytest #126

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 10 commits into from
Dec 17, 2018
Merged

Migrate "test/unit/test_streaming_client_stream_decryptor.py" from unittest to pytest #126

Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
3e3e701
Migrating unit/test_deserialize.py from unittest from pytest
Dec 13, 2018
a070cb6
Migrate unit/test_provides_kms_master_key.py from unittest to pytest
Dec 13, 2018
cba6995
Migrated unit/test/_providers_kms_master_key_provider.py from unittes…
Dec 13, 2018
fdb3b5a
unit/test_providers_raw_master_key.py
Dec 13, 2018
1d2b654
Migrate unit/test_providers_raw_master_key_provider.py from unittest …
Dec 13, 2018
aa3df87
Migrate unit/test_serialize.py from unittest to pytest
Dec 13, 2018
437c391
Migrate unit/test_streaming_client_stream_decryptor.py from unittest …
Dec 13, 2018
5c0acfb
Autoformat
Dec 13, 2018
8dc8804
Removed unit tests not corresponding to this branch
Dec 13, 2018
2e8110b
Removed unused import and added yield and tearDown to apply_fixture f…
Dec 17, 2018
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
45 changes: 22 additions & 23 deletions test/unit/test_streaming_client_stream_decryptor.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,10 +12,8 @@
# language governing permissions and limitations under the License.
"""Unit test suite for aws_encryption_sdk.streaming_client.StreamDecryptor"""
import io
import unittest

import pytest
import six
from mock import MagicMock, call, patch, sentinel

from aws_encryption_sdk.exceptions import CustomMaximumValueExceeded, NotSupportedError, SerializationError
Expand All @@ -29,8 +27,9 @@
pytestmark = [pytest.mark.unit, pytest.mark.local]


class TestStreamDecryptor(unittest.TestCase):
def setUp(self):
class TestStreamDecryptor(object):
@pytest.fixture(autouse=True)
def apply_fixtures(self):
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same as #127 : need to run teardown actions.

self.mock_key_provider = MagicMock(__class__=MasterKeyProvider)
self.mock_materials_manager = MagicMock(__class__=CryptoMaterialsManager)
self.mock_materials_manager.decrypt_materials.return_value = MagicMock(
Expand Down Expand Up @@ -92,8 +91,8 @@ def setUp(self):
# Set up decrypt patch
self.mock_decrypt_patcher = patch("aws_encryption_sdk.streaming_client.decrypt")
self.mock_decrypt = self.mock_decrypt_patcher.start()

def tearDown(self):
yield
# Run tearDown
self.mock_deserialize_header_patcher.stop()
self.mock_deserialize_header_auth_patcher.stop()
self.mock_validate_header_patcher.stop()
Expand Down Expand Up @@ -186,12 +185,11 @@ def test_read_header_frame_too_large(self, mock_derive_datakey):
test_decryptor.key_provider = self.mock_key_provider
test_decryptor.source_stream = ct_stream
test_decryptor._stream_length = len(VALUES["data_128"])
with six.assertRaisesRegex(
self,
CustomMaximumValueExceeded,
"Frame Size in header found larger than custom value: {found} > {custom}".format(found=1024, custom=10),
):
with pytest.raises(CustomMaximumValueExceeded) as excinfo:
test_decryptor._read_header()
excinfo.match(
"Frame Size in header found larger than custom value: {found} > {custom}".format(found=1024, custom=10)
)

@patch("aws_encryption_sdk.streaming_client.Verifier")
@patch("aws_encryption_sdk.streaming_client.DecryptionMaterialsRequest")
Expand Down Expand Up @@ -220,14 +218,13 @@ def test_prep_non_framed_content_length_too_large(self):
mock_data_key = MagicMock()
test_decryptor.data_key = mock_data_key

with six.assertRaisesRegex(
self,
CustomMaximumValueExceeded,
with pytest.raises(CustomMaximumValueExceeded) as excinfo:
test_decryptor._prep_non_framed()
excinfo.match(
"Non-framed message content length found larger than custom value: {found} > {custom}".format(
found=len(VALUES["data_128"]), custom=len(VALUES["data_128"]) // 2
),
):
test_decryptor._prep_non_framed()
)
)

def test_prep_non_framed(self):
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=self.mock_input_stream)
Expand Down Expand Up @@ -288,10 +285,9 @@ def test_read_bytes_from_non_framed_message_body_too_small(self):
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=ct_stream)
test_decryptor.body_length = len(VALUES["data_128"] * 2)
test_decryptor._header = self.mock_header
with six.assertRaisesRegex(
self, SerializationError, "Total message body contents less than specified in body description"
):
with pytest.raises(SerializationError) as excinfo:
test_decryptor._read_bytes_from_non_framed_body(1)
excinfo.match("Total message body contents less than specified in body description")

def test_read_bytes_from_non_framed_no_verifier(self):
ct_stream = io.BytesIO(VALUES["data_128"])
Expand Down Expand Up @@ -497,8 +493,9 @@ def test_read_bytes_from_framed_body_bad_sequence_number(self):
frame_data.final_frame = False
frame_data.ciphertext = b"asdfzxcv"
self.mock_deserialize_frame.return_value = (frame_data, False)
with six.assertRaisesRegex(self, SerializationError, "Malformed message: frames out of order"):
with pytest.raises(SerializationError) as excinfo:
test_decryptor._read_bytes_from_framed_body(4)
excinfo.match("Malformed message: frames out of order")

@patch("aws_encryption_sdk.streaming_client.StreamDecryptor._read_bytes_from_non_framed_body")
@patch("aws_encryption_sdk.streaming_client.StreamDecryptor._read_bytes_from_framed_body")
Expand Down Expand Up @@ -549,8 +546,9 @@ def test_read_bytes_unknown(self, mock_read_frame, mock_read_block):
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=ct_stream)
test_decryptor._header = MagicMock()
test_decryptor._header.content_type = None
with six.assertRaisesRegex(self, NotSupportedError, "Unsupported content type"):
with pytest.raises(NotSupportedError) as excinfo:
test_decryptor._read_bytes(5)
excinfo.match("Unsupported content type")

@patch("aws_encryption_sdk.streaming_client._EncryptionStream.close")
def test_close(self, mock_close):
Expand All @@ -565,5 +563,6 @@ def test_close(self, mock_close):
def test_close_no_footer(self, mock_close):
self.mock_header.content_type = ContentType.FRAMED_DATA
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=self.mock_input_stream)
with six.assertRaisesRegex(self, SerializationError, "Footer not read"):
with pytest.raises(SerializationError) as excinfo:
test_decryptor.close()
excinfo.match("Footer not read")