7f4befdea7
* Wrap .encode and .decode on AnsibleUnsafe objects * runme.sh needs to be executable * ci_complete * Update changelog with CVE
64 lines
2.6 KiB
Python
64 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
# (c) 2016 Toshio Kuratomi <tkuratomi@ansible.com>
|
|
# Copyright (c) 2017 Ansible Project
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
# Make coding more python3-ish
|
|
from __future__ import (absolute_import, division)
|
|
__metaclass__ = type
|
|
|
|
import itertools
|
|
|
|
import pytest
|
|
|
|
from ansible.module_utils.six import PY3
|
|
|
|
# Internal API while this is still being developed. Eventually move to
|
|
# module_utils.common.text
|
|
from ansible.module_utils._text import to_text, to_bytes, to_native
|
|
from ansible.utils.unsafe_proxy import AnsibleUnsafeBytes, AnsibleUnsafeText
|
|
|
|
|
|
# Format: byte representation, text representation, encoding of byte representation
|
|
VALID_STRINGS = (
|
|
(b'abcde', u'abcde', 'ascii'),
|
|
(b'caf\xc3\xa9', u'caf\xe9', 'utf-8'),
|
|
(b'caf\xe9', u'caf\xe9', 'latin-1'),
|
|
# u'くらとみ'
|
|
(b'\xe3\x81\x8f\xe3\x82\x89\xe3\x81\xa8\xe3\x81\xbf', u'\u304f\u3089\u3068\u307f', 'utf-8'),
|
|
(b'\x82\xad\x82\xe7\x82\xc6\x82\xdd', u'\u304f\u3089\u3068\u307f', 'shift-jis'),
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize('in_string, encoding, expected',
|
|
itertools.chain(((d[0], d[2], d[1]) for d in VALID_STRINGS),
|
|
((d[1], d[2], d[1]) for d in VALID_STRINGS)))
|
|
def test_to_text(in_string, encoding, expected):
|
|
"""test happy path of decoding to text"""
|
|
assert to_text(in_string, encoding) == expected
|
|
|
|
|
|
@pytest.mark.parametrize('in_string, encoding, expected',
|
|
itertools.chain(((d[0], d[2], d[0]) for d in VALID_STRINGS),
|
|
((d[1], d[2], d[0]) for d in VALID_STRINGS)))
|
|
def test_to_bytes(in_string, encoding, expected):
|
|
"""test happy path of encoding to bytes"""
|
|
assert to_bytes(in_string, encoding) == expected
|
|
|
|
|
|
@pytest.mark.parametrize('in_string, encoding, expected',
|
|
itertools.chain(((d[0], d[2], d[1] if PY3 else d[0]) for d in VALID_STRINGS),
|
|
((d[1], d[2], d[1] if PY3 else d[0]) for d in VALID_STRINGS)))
|
|
def test_to_native(in_string, encoding, expected):
|
|
"""test happy path of encoding to native strings"""
|
|
assert to_native(in_string, encoding) == expected
|
|
|
|
|
|
def test_to_text_unsafe():
|
|
assert isinstance(to_text(AnsibleUnsafeBytes(b'foo')), AnsibleUnsafeText)
|
|
assert to_text(AnsibleUnsafeBytes(b'foo')) == AnsibleUnsafeText(u'foo')
|
|
|
|
|
|
def test_to_bytes_unsafe():
|
|
assert isinstance(to_bytes(AnsibleUnsafeText(u'foo')), AnsibleUnsafeBytes)
|
|
assert to_bytes(AnsibleUnsafeText(u'foo')) == AnsibleUnsafeBytes(b'foo')
|