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

fix test

Browse Source
This commit is contained in:
Bastien Gérard
2018-10-16 22:55:55 +02:00
parent a1a93a4bdd
commit 26b70e9ed3
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/queryset/queryset.py
View File

@@ -884,9 +884,16 @@ class QuerySetTest(unittest.TestCase):
self.assertEqual(Blog.objects.count(), 2)
# test inserting an existing document (shouldn't be allowed)
with self.assertRaises(OperationError):
with self.assertRaises(OperationError) as cm:
blog = Blog.objects.first()
Blog.objects.insert(blog)
self.assertEqual(str(cm.exception), 'Some documents have ObjectIds use doc.update() instead')
# test inserting a query set
with self.assertRaises(OperationError) as cm:
blogs_qs = Blog.objects
Blog.objects.insert(blogs_qs)
self.assertEqual(str(cm.exception), 'Some documents have ObjectIds use doc.update() instead')
# insert 1 new doc
new_post = Blog(title="code123", id=ObjectId())
@@ -939,12 +946,6 @@ class QuerySetTest(unittest.TestCase):
with self.assertRaises(OperationError):
Blog.objects.insert({'name': 'garbage'})
# test inserting a query set
with self.assertRaises(OperationError) as cm:
blogs_qs = Blog.objects
Blog.objects.insert(blogs_qs)
self.assertEqual(cm.exception.message, 'Some documents have ObjectIds use doc.update() instead')
def test_bulk_insert_update_input_document_ids(self):
class Comment(Document):
idx = IntField()