57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
from mock import patch
|
|
|
|
from .test_helper import raises
|
|
|
|
from kiwi.utils.codec import Codec
|
|
from kiwi.exceptions import KiwiDecodingError
|
|
|
|
|
|
class TestCodec:
|
|
|
|
def setup(self):
|
|
self.literal = bytes(b'\xc3\xbc')
|
|
|
|
@patch('kiwi.utils.codec.Codec._wrapped_decode')
|
|
@patch('kiwi.logger.log.warning')
|
|
def test_decode_ascii_failure(self, mock_warn, mock_decode):
|
|
msg = 'utf-8 compatible string'
|
|
|
|
def mocked_decode(literal, charset):
|
|
if charset:
|
|
return msg
|
|
else:
|
|
raise KiwiDecodingError('ascii decoding failure')
|
|
|
|
mock_decode.side_effect = mocked_decode
|
|
assert msg == Codec.decode(self.literal)
|
|
assert mock_warn.called
|
|
|
|
def test_decode_None_literal(self):
|
|
assert '' == Codec.decode(None)
|
|
|
|
@patch('kiwi.utils.codec.Codec._wrapped_decode')
|
|
def test_decode(self, mock_decode):
|
|
msg = 'utf-8 compatible string'
|
|
mock_decode.return_value = msg
|
|
assert msg == Codec.decode(self.literal)
|
|
|
|
@raises(KiwiDecodingError)
|
|
@patch('kiwi.utils.codec.Codec._wrapped_decode')
|
|
@patch('kiwi.logger.log.warning')
|
|
def test_decode_utf8_failure(self, mock_warn, mock_decode):
|
|
|
|
def mocked_decode(literal, charset):
|
|
if charset:
|
|
raise KiwiDecodingError('utf-8 decoding failure')
|
|
else:
|
|
raise KiwiDecodingError('ascii decoding failure')
|
|
|
|
mock_decode.side_effect = mocked_decode
|
|
Codec.decode(self.literal)
|
|
assert mock_warn.called
|
|
|
|
def test_wrapped_decode(self):
|
|
assert self.literal.decode() == Codec._wrapped_decode(
|
|
self.literal, 'utf-8'
|
|
)
|