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

Fix for py3 in binary field test

Browse Source
This commit is contained in:
Bastien Gérard 2019-02-17 22:56:56 +01:00
parent e02261be6d
commit 201f3008b1
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/fields/test_binary_field.py
View File

@ -119,7 +119,10 @@ class TestBinaryField(MongoDBTestCase):
set__bin_field=BIN_VALUE set__bin_field=BIN_VALUE
) )
self.assertEqual(doc.some_field, 'test') self.assertEqual(doc.some_field, 'test')
self.assertEqual(doc.bin_field, Binary(BIN_VALUE)) if six.PY3:
self.assertEqual(doc.bin_field, BIN_VALUE)
else:
self.assertEqual(doc.bin_field, Binary(BIN_VALUE))
def test_update_one(self): def test_update_one(self):
"""Ensures no regression of bug #1127""" """Ensures no regression of bug #1127"""
@ -134,4 +137,7 @@ class TestBinaryField(MongoDBTestCase):
n_updated = MyDocument.objects(bin_field=bin_data).update_one(bin_field=BIN_VALUE) n_updated = MyDocument.objects(bin_field=bin_data).update_one(bin_field=BIN_VALUE)
self.assertEqual(n_updated, 1) self.assertEqual(n_updated, 1)
fetched = MyDocument.objects.with_id(doc.id) fetched = MyDocument.objects.with_id(doc.id)
self.assertEqual(fetched.bin_field, Binary(BIN_VALUE)) if six.PY3:
self.assertEqual(fetched.bin_field, BIN_VALUE)
else:
self.assertEqual(fetched.bin_field, Binary(BIN_VALUE))