path: root/_pytest/test_utf8_helpers.py

# -*- coding: utf-8 -*-

from __future__ import print_function, unicode_literals

import sys
from collections import OrderedDict
from wee_slack import decode_from_utf8, encode_to_utf8, utf8_decode


b_ae = "æ".encode("utf-8")
b_oe = "ø".encode("utf-8")
b_aa = "å".encode("utf-8")
b_word = b_ae + b_oe + b_aa


if sys.version_info.major > 2:  # noqa: C901

    def test_decode_should_not_transform_str():
        assert "æøå" == decode_from_utf8("æøå")

    def test_decode_should_not_transform_bytes():
        assert b_word == decode_from_utf8(b_word)

    def test_encode_should_not_transform_str():
        assert "æøå" == encode_to_utf8("æøå")

    def test_encode_should_not_transform_bytes():
        assert b_word == encode_to_utf8(b_word)

else:

    def test_decode_preserves_string_without_utf8():
        assert "test" == decode_from_utf8(b"test")

    def test_decode_preserves_unicode_strings():
        assert "æøå" == decode_from_utf8("æøå")

    def test_decode_preserves_mapping_type():
        value_dict = {"a": "x", "b": "y", "c": "z"}
        value_ord_dict = OrderedDict(value_dict)
        assert type(value_dict) == type(decode_from_utf8(value_dict))
        assert type(value_ord_dict) == type(decode_from_utf8(value_ord_dict))

    def test_decode_preserves_iterable_type():
        value_set = {"a", "b", "c"}
        value_tuple = ("a", "b", "c")
        assert type(value_set) == type(decode_from_utf8(value_set))
        assert type(value_tuple) == type(decode_from_utf8(value_tuple))

    def test_decodes_utf8_string_to_unicode():
        assert "æøå" == decode_from_utf8(b_word)

    def test_decodes_utf8_dict_to_unicode():
        assert {"æ": "å", "ø": "å"} == decode_from_utf8({b_ae: b_aa, b_oe: b_aa})

    def test_decodes_utf8_list_to_unicode():
        assert ["æ", "ø", "å"] == decode_from_utf8([b_ae, b_oe, b_aa])

    def test_encode_preserves_string_without_utf8():
        assert b"test" == encode_to_utf8("test")

    def test_encode_preserves_byte_strings():
        assert b_word == encode_to_utf8(b_word)

    def test_encode_preserves_mapping_type():
        value_dict = {"a": "x", "b": "y", "c": "z"}
        value_ord_dict = OrderedDict(value_dict)
        assert type(value_dict) == type(encode_to_utf8(value_dict))
        assert type(value_ord_dict) == type(encode_to_utf8(value_ord_dict))

    def test_encode_preserves_iterable_type():
        value_set = {"a", "b", "c"}
        value_tuple = ("a", "b", "c")
        assert type(value_set) == type(encode_to_utf8(value_set))
        assert type(value_tuple) == type(encode_to_utf8(value_tuple))

    def test_encodes_utf8_string_to_unicode():
        assert b_word == encode_to_utf8("æøå")

    def test_encodes_utf8_dict_to_unicode():
        assert {b_ae: b_aa, b_oe: b_aa} == encode_to_utf8({"æ": "å", "ø": "å"})

    def test_encodes_utf8_list_to_unicode():
        assert [b_ae, b_oe, b_aa] == encode_to_utf8(["æ", "ø", "å"])

    @utf8_decode
    def method_with_utf8_decode(*args, **kwargs):
        return (args, kwargs)

    def test_utf8_decode():
        args = (b_ae, b_oe, b_aa)
        kwargs = {b_ae: b_aa, b_oe: b_aa}

        result_args, result_kwargs = method_with_utf8_decode(*args, **kwargs)

        assert result_args == decode_from_utf8(args)
        assert result_kwargs == decode_from_utf8(kwargs)