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
|
# -*- coding: utf-8 -*-
import unittest
import azure.cosmos.cosmos_client as cosmos_client
import uuid
import pytest
import test_config
pytestmark = pytest.mark.cosmosEmulator
@pytest.mark.usefixtures("teardown")
class EncodingTest(unittest.TestCase):
"""Test to ensure escaping of non-ascii characters from partition key"""
host = test_config._test_config.host
masterKey = test_config._test_config.masterKey
connectionPolicy = test_config._test_config.connectionPolicy
@classmethod
def setUpClass(cls):
if (cls.masterKey == '[YOUR_KEY_HERE]' or
cls.host == '[YOUR_ENDPOINT_HERE]'):
raise Exception(
"You must specify your Azure Cosmos account values for "
"'masterKey' and 'host' at the top of this class to run the "
"tests.")
cls.client = cosmos_client.CosmosClient(cls.host, cls.masterKey, connection_policy=cls.connectionPolicy)
cls.created_collection = test_config._test_config.create_multi_partition_collection_with_custom_pk_if_not_exist(cls.client)
def test_unicode_characters_in_partition_key (self):
test_string = u'€€ کلید پارتیشن विभाजन कुंजी 123' # cspell:disable-line
document_definition = {'pk': test_string, 'id': 'myid' + str(uuid.uuid4())}
created_doc = self.created_collection.create_item(body=document_definition)
read_doc = self.created_collection.read_item(item=created_doc['id'], partition_key=test_string)
self.assertEqual(read_doc['pk'], test_string)
def test_create_document_with_line_separator_para_seperator_next_line_unicodes (self):
test_string = u'Line Separator (
) & Paragraph Separator (
) & Next Line (
) & نیمفاصله' # cspell:disable-line
document_definition = {'pk': 'pk', 'id':'myid' + str(uuid.uuid4()), 'unicode_content':test_string }
created_doc = self.created_collection.create_item(body=document_definition)
read_doc = self.created_collection.read_item(item=created_doc['id'], partition_key='pk')
self.assertEqual(read_doc['unicode_content'], test_string)
def test_create_stored_procedure_with_line_separator_para_seperator_next_line_unicodes (self):
test_string = 'Line Separator (
) & Paragraph Separator (
) & Next Line (
) & نیمفاصله' # cspell:disable-line
test_string_unicode = u'Line Separator (
) & Paragraph Separator (
) & Next Line (
) & نیمفاصله' # cspell:disable-line
stored_proc_definition = {'id':'myid' + str(uuid.uuid4()), 'body': test_string}
created_sp = self.created_collection.scripts.create_stored_procedure(body=stored_proc_definition)
read_sp = self.created_collection.scripts.get_stored_procedure(created_sp['id'])
self.assertEqual(read_sp['body'], test_string_unicode)
if __name__ == '__main__':
unittest.main()
|
