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

Add support to transform. Add pull tests for and .

Browse Source
This commit is contained in:
Agustin Barto
2019-05-17 13:41:02 -03:00
parent 58b0b18ddd
commit b392e3102e
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/queryset/transform.py
View File

@@ -276,13 +276,18 @@ class TransformTest(unittest.TestCase):
title = StringField()
content = EmbeddedDocumentField(SubDoc)
word = Word(word='abc', index=1)
update = transform.update(MainDoc, pull__content__text=word)
self.assertEqual(update, {'$pull': {'content.text': SON([('word', u'abc'), ('index', 1)])}})
# word = Word(word='abc', index=1)
# update = transform.update(MainDoc, pull__content__text=word)
# self.assertEqual(update, {'$pull': {'content.text': SON([('word', u'abc'), ('index', 1)])}})
update = transform.update(MainDoc, pull__content__heading='xyz')
self.assertEqual(update, {'$pull': {'content.heading': 'xyz'}})
# update = transform.update(MainDoc, pull__content__heading='xyz')
# self.assertEqual(update, {'$pull': {'content.heading': 'xyz'}})
# update = transform.update(MainDoc, pull__content__text__word__in=['foo', 'bar'])
# self.assertEqual(update, {'$pull': {'content.text': {'word': {'$in': ['foo', 'bar']}}}})
update = transform.update(MainDoc, pull__content__text__word__nin=['foo', 'bar'])
self.assertEqual(update, {'$pull': {'content.text': {'word': {'$nin': ['foo', 'bar']}}}})
if __name__ == '__main__':
unittest.main()