#!/usr/bin/env python

"""
Copyright (c) 2006-2026 sqlmap developers (https://sqlmap.org)
See the file 'LICENSE' for copying permission

Core text<->bytes conversions (lib/core/convert.py): getBytes, getUnicode,
getText. (getOrds is covered in test_convert.py.)

These are called on essentially every request and response, on both Python 2
and 3, and are the main thing standing between sqlmap and a UnicodeDecodeError
mid-scan. Pinned with known vectors, non-string coercion, and an encoding
round-trip property over multiple charsets.
"""

import os
import random
import sys
import unittest

sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from _testutils import bootstrap
bootstrap()

from lib.core.convert import getBytes, getUnicode, getText

RND = random.Random(2024)


class TestTypes(unittest.TestCase):
    # value+type (not type alone): a stub returning b"" would pass an isinstance-only check, and
    # on py3 a getBytes that wrongly returned str would slip past a round-trip on the unicode path
    def test_getBytes_returns_bytes(self):
        out = getBytes(u"abc")
        self.assertIsInstance(out, bytes)
        self.assertEqual(out, b"abc")

    def test_getUnicode_returns_unicode(self):
        out = getUnicode(b"abc")
        self.assertIsInstance(out, type(u""))
        self.assertEqual(out, u"abc")

    def test_getText_returns_native_str(self):
        self.assertIsInstance(getText(b"abc"), str)
        self.assertEqual(getText(b"abc"), "abc")


class TestCoercion(unittest.TestCase):
    def test_getUnicode_of_number(self):
        self.assertEqual(getUnicode(123), u"123")


class TestRoundTrip(unittest.TestCase):
    def test_known_utf8(self):
        self.assertEqual(getUnicode(getBytes(u"caf\xe9", "utf-8"), "utf-8"), u"caf\xe9")

    def test_property_multi_charset(self):
        # printable BMP-ish range, round-trip through utf-8 and latin1-safe subset
        for encoding, hi in (("utf-8", 0x2000), ("latin-1", 0x100)):
            for _ in range(1000):
                s = u"".join(unichr(RND.randint(0, hi - 1)) if sys.version_info[0] < 3
                             else chr(RND.randint(0, hi - 1)) for _ in range(RND.randint(0, 16)))
                self.assertEqual(getUnicode(getBytes(s, encoding), encoding), s,
                                 msg="round-trip failed (%s): %r" % (encoding, s))


# py2 has unichr, py3 does not; normalize so the file imports cleanly on both
try:
    unichr
except NameError:
    unichr = chr


if __name__ == "__main__":
    unittest.main(verbosity=2)