pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test_collation:

Browse Source 
Added test with `pymongo.collation.Collation` object
Readable list of BlogPost names for test
This commit is contained in:
otrofimov 2019-08-21 18:26:10 +03:00
parent fbb3bf869c
commit 71a6f3d1a4
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/document/indexes.py
View File

@ -3,6 +3,7 @@ import unittest
from datetime import datetime from datetime import datetime
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
from pymongo.collation import Collation
from pymongo.errors import OperationFailure from pymongo.errors import OperationFailure
import pymongo import pymongo
from six import iteritems from six import iteritems
@ -552,7 +553,7 @@ class IndexesTest(unittest.TestCase):
BlogPost.drop_collection() BlogPost.drop_collection()
names = tuple("%sag %i" % ("t" if n % 2 == 0 else "T", n) for n in range(10)) names = ["tag1", "Tag2", "tag3", "Tag4", "tag5"]
for name in names: for name in names:
BlogPost(name=name).save() BlogPost(name=name).save()
@ -560,7 +561,13 @@ class IndexesTest(unittest.TestCase):
self.assertEqual( self.assertEqual(
[x.name for x in query_result], sorted(names, key=lambda x: x.lower()) [x.name for x in query_result], sorted(names, key=lambda x: x.lower())
) )
self.assertEqual(10, query_result.count()) self.assertEqual(5, query_result.count())
query_result = BlogPost.objects.collation(Collation(**base)).order_by("name")
self.assertEqual(
[x.name for x in query_result], sorted(names, key=lambda x: x.lower())
)
self.assertEqual(5, query_result.count())
incorrect_collation = {"arndom": "wrdo"} incorrect_collation = {"arndom": "wrdo"}
with self.assertRaises(OperationFailure): with self.assertRaises(OperationFailure):