opensearch-pyd/test_elasticsearch/test_client/test_utils.py

# -*- coding: utf-8 -*-
from __future__ import unicode_literals

from elasticsearch.client.utils import _make_path, _escape
from elasticsearch.compat import PY2

from ..test_cases import TestCase, SkipTest

class TestMakePath(TestCase):
    def test_handles_unicode(self):
        id = "中文"
        self.assertEquals('/some-index/type/%E4%B8%AD%E6%96%87', _make_path('some-index', 'type', id))

    def test_handles_utf_encoded_string(self):
        if not PY2:
            raise SkipTest('Only relevant for py2')
        id = "中文".encode('utf-8')
        self.assertEquals('/some-index/type/%E4%B8%AD%E6%96%87', _make_path('some-index', 'type', id))


class TestEscape(TestCase):
    def test_handles_ascii(self):
        string = "abc123"
        self.assertEquals(
            b'abc123',
            _escape(string)
        )
    def test_handles_unicode(self):
        string = "中文"
        self.assertEquals(
            b'\xe4\xb8\xad\xe6\x96\x87',
            _escape(string)
        )

    def test_handles_bytestring(self):
        string = b'celery-task-meta-c4f1201f-eb7b-41d5-9318-a75a8cfbdaa0'
        self.assertEquals(
            string,
            _escape(string)
        )