def test_read_bytes_no_read_required(self, mock_read_frame, mock_read_block):
ct_stream = io.BytesIO(VALUES["data_128"])
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=ct_stream)
test_decryptor.output_buffer = b"1234567"
test_decryptor._read_bytes(5)
assert not mock_read_frame.called
assert not mock_read_block.called
def test_read_bytes_completed(self, mock_read_frame, mock_read_block):
ct_stream = io.BytesIO(VALUES["data_128"])
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=ct_stream)
test_decryptor.footer = None
test_decryptor._read_bytes(5)
assert not mock_read_frame.called
assert not mock_read_block.called
def test_read_bytes_unknown(self, mock_read_frame, mock_read_block):
ct_stream = io.BytesIO(VALUES["data_128"])
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=ct_stream)
test_decryptor._header = MagicMock()
test_decryptor._header.content_type = None
with pytest.raises(NotSupportedError) as excinfo:
test_decryptor._read_bytes(5)
excinfo.match("Unsupported content type")
def test_read_bytes_non_framed(self, mock_read_frame, mock_read_block):
ct_stream = io.BytesIO(VALUES["data_128"])
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=ct_stream)
test_decryptor._header = MagicMock()
test_decryptor._header.content_type = ContentType.NO_FRAMING
test_decryptor._read_bytes(5)
mock_read_block.assert_called_once_with(5)
assert not mock_read_frame.called
def test_read_bytes_closed(self, mock_read_frame, mock_read_block):
ct_stream = io.BytesIO(VALUES['data_128'])
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider,
source=ct_stream)
test_decryptor.source_stream.close()
test_decryptor._read_bytes(5)
assert not mock_read_frame.called
assert not mock_read_block.called
def test_read_bytes_unknown(self, mock_read_frame, mock_read_block):
ct_stream = io.BytesIO(VALUES['data_128'])
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'):
test_decryptor._read_bytes(5)
def test_read_bytes_completed(self, mock_read_frame, mock_read_block):
ct_stream = io.BytesIO(VALUES["data_128"])
test_decryptor = StreamDecryptor(
materials_manager=self.mock_materials_manager,
source=ct_stream,
commitment_policy=self.mock_commitment_policy,
)
test_decryptor.footer = None
test_decryptor._read_bytes(5)
assert not mock_read_frame.called
assert not mock_read_block.called
def test_read_bytes_non_framed(self, mock_read_frame, mock_read_block):
ct_stream = io.BytesIO(VALUES["data_128"])
test_decryptor = StreamDecryptor(
materials_manager=self.mock_materials_manager,
source=ct_stream,
commitment_policy=self.mock_commitment_policy,
)
test_decryptor._header = MagicMock()
test_decryptor._header.content_type = ContentType.NO_FRAMING
test_decryptor._read_bytes(5)
mock_read_block.assert_called_once_with(5)
assert not mock_read_frame.called
def test_read_bytes_no_read_required(self, mock_read_frame,
mock_read_block):
ct_stream = io.BytesIO(VALUES["data_128"])
test_decryptor = StreamDecryptor(
materials_manager=self.mock_materials_manager,
source=ct_stream,
commitment_policy=self.mock_commitment_policy,
)
test_decryptor.output_buffer = b"1234567"
test_decryptor._read_bytes(5)
assert not mock_read_frame.called
assert not mock_read_block.called
