From 71a6f3d1a46702bf5d66e704519671e828626cd2 Mon Sep 17 00:00:00 2001 From: otrofimov Date: Wed, 21 Aug 2019 18:26:10 +0300 Subject: [PATCH] test_collation: Added test with `pymongo.collation.Collation` object Readable list of BlogPost names for test --- tests/document/indexes.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/tests/document/indexes.py b/tests/document/indexes.py index fa3d1706..dcd3fc6a 100644 --- a/tests/document/indexes.py +++ b/tests/document/indexes.py @@ -3,6 +3,7 @@ import unittest from datetime import datetime from nose.plugins.skip import SkipTest +from pymongo.collation import Collation from pymongo.errors import OperationFailure import pymongo from six import iteritems @@ -552,7 +553,7 @@ class IndexesTest(unittest.TestCase): 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: BlogPost(name=name).save() @@ -560,7 +561,13 @@ class IndexesTest(unittest.TestCase): self.assertEqual( [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"} with self.assertRaises(OperationFailure):