aboutsummaryrefslogtreecommitdiffstats
path: root/_pytest/test_utf8_helpers.py
blob: a2dea428139eaa333753f30275695e945faa2156 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# -*- 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:
    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)