jyapayne/mongoengine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Mongomallard

Browse source
This commit is contained in:
Thomas Steinacher 2013-06-13 16:41:04 -07:00
commit dcc5d3c858
24 changed files with 584 additions and 1102 deletions
Download patch file Download diff file Expand all files Collapse all files

77
tests/document/delta.py
View file

@ -48,41 +48,42 @@ class DeltaTest(unittest.TestCase):
doc.save()
doc = Doc.objects.first()
self.assertEqual(doc._get_changed_fields(), [])
self.assertEqual(doc._get_changed_fields(), set())
self.assertEqual(doc._delta(), ({}, {}))
doc.string_field = 'hello'
self.assertEqual(doc._get_changed_fields(), ['string_field'])
self.assertEqual(doc._get_changed_fields(), set(['string_field']))
self.assertEqual(doc._delta(), ({'string_field': 'hello'}, {}))
doc._changed_fields = []
doc._changed_fields = set()
doc.int_field = 1
self.assertEqual(doc._get_changed_fields(), ['int_field'])
self.assertEqual(doc._get_changed_fields(), set(['int_field']))
self.assertEqual(doc._delta(), ({'int_field': 1}, {}))
doc._changed_fields = []
doc._changed_fields = set()
dict_value = {'hello': 'world', 'ping': 'pong'}
doc.dict_field = dict_value
self.assertEqual(doc._get_changed_fields(), ['dict_field'])
self.assertEqual(doc._get_changed_fields(), set(['dict_field']))
self.assertEqual(doc._delta(), ({'dict_field': dict_value}, {}))
doc._changed_fields = []
doc._changed_fields = set()
list_value = ['1', 2, {'hello': 'world'}]
doc.list_field = list_value
self.assertEqual(doc._get_changed_fields(), ['list_field'])
self.assertEqual(doc._get_changed_fields(), set(['list_field']))
self.assertEqual(doc._delta(), ({'list_field': list_value}, {}))
# Test unsetting
doc._changed_fields = []
doc._changed_fields = set()
doc.dict_field = {}
self.assertEqual(doc._get_changed_fields(), ['dict_field'])
self.assertEqual(doc._get_changed_fields(), set(['dict_field']))
self.assertEqual(doc._delta(), ({}, {'dict_field': 1}))
doc._changed_fields = []
doc._changed_fields = set()
doc.list_field = []
self.assertEqual(doc._get_changed_fields(), ['list_field'])
self.assertEqual(doc._get_changed_fields(), set(['list_field']))
self.assertEqual(doc._delta(), ({}, {'list_field': 1}))
@unittest.skip("not fully implemented")
def test_delta_recursive(self):
self.delta_recursive(Document, EmbeddedDocument)
self.delta_recursive(DynamicDocument, EmbeddedDocument)
@ -109,7 +110,7 @@ class DeltaTest(unittest.TestCase):
doc.save()
doc = Doc.objects.first()
self.assertEqual(doc._get_changed_fields(), [])
self.assertEqual(doc._get_changed_fields(), set())
self.assertEqual(doc._delta(), ({}, {}))
embedded_1 = Embedded()
@ -119,7 +120,7 @@ class DeltaTest(unittest.TestCase):
embedded_1.list_field = ['1', 2, {'hello': 'world'}]
doc.embedded_field = embedded_1
self.assertEqual(doc._get_changed_fields(), ['embedded_field'])
self.assertEqual(doc._get_changed_fields(), set(['embedded_field']))
embedded_delta = {
'string_field': 'hello',
@ -136,7 +137,7 @@ class DeltaTest(unittest.TestCase):
doc.embedded_field.dict_field = {}
self.assertEqual(doc._get_changed_fields(),
['embedded_field.dict_field'])
set(['embedded_field.dict_field']))
self.assertEqual(doc.embedded_field._delta(), ({}, {'dict_field': 1}))
self.assertEqual(doc._delta(), ({}, {'embedded_field.dict_field': 1}))
doc.save()
@ -145,7 +146,7 @@ class DeltaTest(unittest.TestCase):
doc.embedded_field.list_field = []
self.assertEqual(doc._get_changed_fields(),
['embedded_field.list_field'])
set(['embedded_field.list_field']))
self.assertEqual(doc.embedded_field._delta(), ({}, {'list_field': 1}))
self.assertEqual(doc._delta(), ({}, {'embedded_field.list_field': 1}))
doc.save()
@ -160,7 +161,7 @@ class DeltaTest(unittest.TestCase):
doc.embedded_field.list_field = ['1', 2, embedded_2]
self.assertEqual(doc._get_changed_fields(),
['embedded_field.list_field'])
set(['embedded_field.list_field']))
self.assertEqual(doc.embedded_field._delta(), ({
'list_field': ['1', 2, {
@ -192,7 +193,7 @@ class DeltaTest(unittest.TestCase):
doc.embedded_field.list_field[2].string_field = 'world'
self.assertEqual(doc._get_changed_fields(),
['embedded_field.list_field.2.string_field'])
set(['embedded_field.list_field.2.string_field']))
self.assertEqual(doc.embedded_field._delta(),
({'list_field.2.string_field': 'world'}, {}))
self.assertEqual(doc._delta(),
@ -206,7 +207,7 @@ class DeltaTest(unittest.TestCase):
doc.embedded_field.list_field[2].string_field = 'hello world'
doc.embedded_field.list_field[2] = doc.embedded_field.list_field[2]
self.assertEqual(doc._get_changed_fields(),
['embedded_field.list_field'])
set(['embedded_field.list_field']))
self.assertEqual(doc.embedded_field._delta(), ({
'list_field': ['1', 2, {
'_cls': 'Embedded',
@ -269,7 +270,7 @@ class DeltaTest(unittest.TestCase):
doc.dict_field['Embedded'].string_field = 'Hello World'
self.assertEqual(doc._get_changed_fields(),
['dict_field.Embedded.string_field'])
set(['dict_field.Embedded.string_field']))
self.assertEqual(doc._delta(),
({'dict_field.Embedded.string_field': 'Hello World'}, {}))
@ -371,39 +372,39 @@ class DeltaTest(unittest.TestCase):
doc.save()
doc = Doc.objects.first()
self.assertEqual(doc._get_changed_fields(), [])
self.assertEqual(doc._get_changed_fields(), set())
self.assertEqual(doc._delta(), ({}, {}))
doc.string_field = 'hello'
self.assertEqual(doc._get_changed_fields(), ['db_string_field'])
self.assertEqual(doc._get_changed_fields(), set(['string_field']))
self.assertEqual(doc._delta(), ({'db_string_field': 'hello'}, {}))
doc._changed_fields = []
doc._changed_fields = set()
doc.int_field = 1
self.assertEqual(doc._get_changed_fields(), ['db_int_field'])
self.assertEqual(doc._get_changed_fields(), set(['int_field']))
self.assertEqual(doc._delta(), ({'db_int_field': 1}, {}))
doc._changed_fields = []
doc._changed_fields = set()
dict_value = {'hello': 'world', 'ping': 'pong'}
doc.dict_field = dict_value
self.assertEqual(doc._get_changed_fields(), ['db_dict_field'])
self.assertEqual(doc._get_changed_fields(), set(['dict_field']))
self.assertEqual(doc._delta(), ({'db_dict_field': dict_value}, {}))
doc._changed_fields = []
doc._changed_fields = set()
list_value = ['1', 2, {'hello': 'world'}]
doc.list_field = list_value
self.assertEqual(doc._get_changed_fields(), ['db_list_field'])
self.assertEqual(doc._get_changed_fields(), set(['list_field']))
self.assertEqual(doc._delta(), ({'db_list_field': list_value}, {}))
# Test unsetting
doc._changed_fields = []
doc._changed_fields = set()
doc.dict_field = {}
self.assertEqual(doc._get_changed_fields(), ['db_dict_field'])
self.assertEqual(doc._get_changed_fields(), set(['dict_field']))
self.assertEqual(doc._delta(), ({}, {'db_dict_field': 1}))
doc._changed_fields = []
doc._changed_fields = set()
doc.list_field = []
self.assertEqual(doc._get_changed_fields(), ['db_list_field'])
self.assertEqual(doc._get_changed_fields(), set(['list_field']))
self.assertEqual(doc._delta(), ({}, {'db_list_field': 1}))
# Test it saves that data
@ -415,13 +416,15 @@ class DeltaTest(unittest.TestCase):
doc.dict_field = {'hello': 'world'}
doc.list_field = ['1', 2, {'hello': 'world'}]
doc.save()
doc = doc.reload(10)
#doc = doc.reload(10)
doc = doc.reload()
self.assertEqual(doc.string_field, 'hello')
self.assertEqual(doc.int_field, 1)
self.assertEqual(doc.dict_field, {'hello': 'world'})
self.assertEqual(doc.list_field, ['1', 2, {'hello': 'world'}])
@unittest.skip("not fully implemented")
def test_delta_recursive_db_field(self):
self.delta_recursive_db_field(Document, EmbeddedDocument)
self.delta_recursive_db_field(Document, DynamicEmbeddedDocument)
@ -449,7 +452,7 @@ class DeltaTest(unittest.TestCase):
doc.save()
doc = Doc.objects.first()
self.assertEqual(doc._get_changed_fields(), [])
self.assertEqual(doc._get_changed_fields(), set())
self.assertEqual(doc._delta(), ({}, {}))
embedded_1 = Embedded()
@ -459,7 +462,7 @@ class DeltaTest(unittest.TestCase):
embedded_1.list_field = ['1', 2, {'hello': 'world'}]
doc.embedded_field = embedded_1
self.assertEqual(doc._get_changed_fields(), ['db_embedded_field'])
self.assertEqual(doc._get_changed_fields(), set(['embedded_field']))
embedded_delta = {
'db_string_field': 'hello',
@ -487,7 +490,7 @@ class DeltaTest(unittest.TestCase):
doc.embedded_field.list_field = []
self.assertEqual(doc._get_changed_fields(),
['db_embedded_field.db_list_field'])
set(['db_embedded_field.db_list_field']))
self.assertEqual(doc.embedded_field._delta(),
({}, {'db_list_field': 1}))
self.assertEqual(doc._delta(),
@ -605,6 +608,7 @@ class DeltaTest(unittest.TestCase):
self.assertEqual(doc._delta(), ({},
{'db_embedded_field.db_list_field.2.db_list_field': 1}))
@unittest.skip("DynamicDocument not implemented")
def test_delta_for_dynamic_documents(self):
class Person(DynamicDocument):
name = StringField()
@ -640,6 +644,7 @@ class DeltaTest(unittest.TestCase):
p.save()
self.assertEqual(1, self.Person.objects(age=24).count())
@unittest.skip("DynamicDocument not implemented")
def test_dynamic_delta(self):
class Doc(DynamicDocument):

1
tests/document/dynamic.py
View file

@ -8,6 +8,7 @@ from mongoengine.connection import get_db
__all__ = ("DynamicTest", )
@unittest.skip("DynamicDocument not implemented")
class DynamicTest(unittest.TestCase):
def setUp(self):

1
tests/document/indexes.py
View file

@ -632,6 +632,7 @@ class IndexesTest(unittest.TestCase):
pass
Customer.drop_collection()
@unittest.skip("behavior differs")
def test_unique_and_primary(self):
"""If you set a field as primary, then unexpected behaviour can occur.
You won't create a duplicate but you will update an existing document.

8
tests/document/inheritance.py
View file

@ -182,10 +182,10 @@ class InheritanceTest(unittest.TestCase):
self.assertEqual(['age', 'id', 'name', 'salary'],
sorted(Employee._fields.keys()))
self.assertEqual(Person(name="Bob", age=35).to_mongo().keys(),
['_cls', 'name', 'age'])
self.assertEqual(Employee(name="Bob", age=35, salary=0).to_mongo().keys(),
['_cls', 'name', 'age', 'salary'])
self.assertEqual(set(Person(name="Bob", age=35).to_mongo().keys()),
set(['_cls', 'name', 'age']))
self.assertEqual(set(Employee(name="Bob", age=35, salary=0).to_mongo().keys()),
set(['_cls', 'name', 'age', 'salary']))
self.assertEqual(Employee._get_collection_name(),
Person._get_collection_name())

63
tests/document/instance.py
View file

@ -390,24 +390,27 @@ class InstanceTest(unittest.TestCase):
doc.embedded_field = embedded_1
doc.save()
doc = doc.reload(10)
doc = doc.reload()
doc.list_field.append(1)
doc.dict_field['woot'] = "woot"
doc.embedded_field.list_field.append(1)
doc.embedded_field.dict_field['woot'] = "woot"
self.assertEqual(doc._get_changed_fields(), [
'list_field', 'dict_field', 'embedded_field.list_field',
'embedded_field.dict_field'])
self.assertEqual(doc._get_changed_fields(), set([
'list_field', 'dict_field', 'embedded_field']))
#self.assertEqual(doc._get_changed_fields(), [
# 'list_field', 'dict_field', 'embedded_field.list_field',
# 'embedded_field.dict_field'])
doc.save()
doc = doc.reload(10)
self.assertEqual(doc._get_changed_fields(), [])
doc = doc.reload()
self.assertEqual(doc._get_changed_fields(), set())
self.assertEqual(len(doc.list_field), 4)
self.assertEqual(len(doc.dict_field), 2)
self.assertEqual(len(doc.embedded_field.list_field), 4)
self.assertEqual(len(doc.embedded_field.dict_field), 2)
@unittest.skip("not implemented")
def test_dictionary_access(self):
"""Ensure that dictionary-style field access works properly.
"""
@ -438,10 +441,10 @@ class InstanceTest(unittest.TestCase):
class Employee(Person):
salary = IntField()
self.assertEqual(Person(name="Bob", age=35).to_mongo().keys(),
['_cls', 'name', 'age'])
self.assertEqual(Employee(name="Bob", age=35, salary=0).to_mongo().keys(),
['_cls', 'name', 'age', 'salary'])
self.assertEqual(set(Person(name="Bob", age=35).to_mongo().keys()),
set(['_cls', 'name', 'age']))
self.assertEqual(set(Employee(name="Bob", age=35, salary=0).to_mongo().keys()),
set(['_cls', 'name', 'age', 'salary']))
def test_embedded_document(self):
"""Ensure that embedded documents are set up correctly.
@ -452,6 +455,7 @@ class InstanceTest(unittest.TestCase):
self.assertTrue('content' in Comment._fields)
self.assertFalse('id' in Comment._fields)
@unittest.skip("not implemented")
def test_embedded_document_instance(self):
"""Ensure that embedded documents can reference parent instance
"""
@ -460,6 +464,7 @@ class InstanceTest(unittest.TestCase):
class Doc(Document):
embedded_field = EmbeddedDocumentField(Embedded)
meta = { 'cascade': True }
Doc.drop_collection()
Doc(embedded_field=Embedded(string="Hi")).save()
@ -467,6 +472,7 @@ class InstanceTest(unittest.TestCase):
doc = Doc.objects.get()
self.assertEqual(doc, doc.embedded_field._instance)
@unittest.skip("not implemented")
def test_embedded_document_complex_instance(self):
"""Ensure that embedded documents in complex fields can reference
parent instance"""
@ -623,6 +629,7 @@ class InstanceTest(unittest.TestCase):
p0.name = 'wpjunior'
p0.save()
@unittest.skip("FileField not implemented")
def test_save_max_recursion_not_hit_with_file_field(self):
class Foo(Document):
@ -771,6 +778,7 @@ class InstanceTest(unittest.TestCase):
p1.reload()
self.assertEqual(p1.name, p.parent.name)
@unittest.skip("not implemented")
def test_update(self):
"""Ensure that an existing document is updated instead of be
overwritten."""
@ -885,7 +893,6 @@ class InstanceTest(unittest.TestCase):
reference_field = ReferenceField(Simple, default=lambda:
Simple().save())
map_field = MapField(IntField(), default=lambda: {"simple": 1})
decimal_field = DecimalField(default=1.0)
complex_datetime_field = ComplexDateTimeField(default=datetime.now)
url_field = URLField(default="http://mongoengine.org")
dynamic_field = DynamicField(default=1)
@ -1054,9 +1061,9 @@ class InstanceTest(unittest.TestCase):
user = User.objects.first()
# Even if stored as ObjectId's internally mongoengine uses DBRefs
# As ObjectId's aren't automatically derefenced
self.assertTrue(isinstance(user._data['orgs'][0], DBRef))
#self.assertTrue(isinstance(user._data['orgs'][0], DBRef))
self.assertTrue(isinstance(user.orgs[0], Organization))
self.assertTrue(isinstance(user._data['orgs'][0], Organization))
#self.assertTrue(isinstance(user._data['orgs'][0], Organization))
# Changing a value
with query_counter() as q:
@ -1136,6 +1143,7 @@ class InstanceTest(unittest.TestCase):
foo.save()
self.assertEqual(1, q)
@unittest.skip("not implemented")
def test_save_only_changed_fields_recursive(self):
"""Ensure save only sets / unsets changed fields
"""
@ -1433,8 +1441,8 @@ class InstanceTest(unittest.TestCase):
post_obj = BlogPost.objects.first()
# Test laziness
self.assertTrue(isinstance(post_obj._data['author'],
bson.DBRef))
#self.assertTrue(isinstance(post_obj._data['author'],
# bson.DBRef))
self.assertTrue(isinstance(post_obj.author, self.Person))
self.assertEqual(post_obj.author.name, 'Test User')
@ -1458,6 +1466,7 @@ class InstanceTest(unittest.TestCase):
self.assertRaises(InvalidDocumentError, throw_invalid_document_error)
@unittest.skip("not implemented")
def test_invalid_son(self):
"""Raise an error if loading invalid data"""
class Occurrence(EmbeddedDocument):
@ -1801,6 +1810,7 @@ class InstanceTest(unittest.TestCase):
self.assertTrue(u1 in all_user_set)
@unittest.skip("not implemented")
def test_picklable(self):
pickle_doc = PickleTest(number=1, string="One", lists=['1', '2'])
@ -1827,6 +1837,7 @@ class InstanceTest(unittest.TestCase):
self.assertEqual(pickle_doc.string, "Two")
self.assertEqual(pickle_doc.lists, ["1", "2", "3"])
@unittest.skip("not implemented")
def test_picklable_on_signals(self):
pickle_doc = PickleSignalsTest(number=1, string="One", lists=['1', '2'])
pickle_doc.embedded = PickleEmbedded()
@ -1887,6 +1898,7 @@ class InstanceTest(unittest.TestCase):
self.assertEqual(Doc.objects(archived=False).count(), 1)
@unittest.skip("DynamicDocument not implemented")
def test_can_save_false_values_dynamic(self):
"""Ensures you can save False values on dynamic docs"""
class Doc(DynamicDocument):
@ -2026,6 +2038,7 @@ class InstanceTest(unittest.TestCase):
self.assertEqual('testdb-1', B._meta.get('db_alias'))
@unittest.skip("not implemented")
def test_db_ref_usage(self):
""" DB Ref usage in dict_fields"""
@ -2104,6 +2117,7 @@ class InstanceTest(unittest.TestCase):
})]),
"1,2")
@unittest.skip("not implemented")
def test_switch_db_instance(self):
register_connection('testdb-1', 'mongoenginetest2')
@ -2175,9 +2189,10 @@ class InstanceTest(unittest.TestCase):
user = User.objects.first()
self.assertEqual("Ross", user.username)
self.assertEqual(True, user.foo)
self.assertEqual("Bar", user._data["foo"])
self.assertEqual([1, 2, 3], user._data["data"])
self.assertEqual("Bar", user._db_data["foo"])
self.assertEqual([1, 2, 3], user._db_data["data"])
@unittest.skip("DynamicDocument not implemented")
def test_spaces_in_keys(self):
class Embedded(DynamicEmbeddedDocument):
@ -2194,6 +2209,7 @@ class InstanceTest(unittest.TestCase):
one = Doc.objects.filter(**{'hello world': 1}).count()
self.assertEqual(1, one)
@unittest.skip("not implemented")
def test_shard_key(self):
class LogEntry(Document):
machine = StringField()
@ -2217,6 +2233,7 @@ class InstanceTest(unittest.TestCase):
self.assertRaises(OperationError, change_shard_key)
@unittest.skip("not implemented")
def test_shard_key_primary(self):
class LogEntry(Document):
machine = StringField(primary_key=True)
@ -2240,6 +2257,7 @@ class InstanceTest(unittest.TestCase):
self.assertRaises(OperationError, change_shard_key)
@unittest.skip("not implemented")
def test_kwargs_simple(self):
class Embedded(EmbeddedDocument):
@ -2254,8 +2272,9 @@ class InstanceTest(unittest.TestCase):
"doc": {"name": "embedded doc"}})
self.assertEqual(classic_doc, dict_doc)
self.assertEqual(classic_doc._data, dict_doc._data)
self.assertEqual(classic_doc.to_dict(), dict_doc.to_dict())
@unittest.skip("not implemented")
def test_kwargs_complex(self):
class Embedded(EmbeddedDocument):
@ -2273,8 +2292,9 @@ class InstanceTest(unittest.TestCase):
{"name": "embedded doc2"}]})
self.assertEqual(classic_doc, dict_doc)
self.assertEqual(classic_doc._data, dict_doc._data)
self.assertEqual(classic_doc.to_dict(), dict_doc.to_dict())
@unittest.skip("not implemented")
def test_positional_creation(self):
"""Ensure that document may be created using positional arguments.
"""
@ -2282,6 +2302,7 @@ class InstanceTest(unittest.TestCase):
self.assertEqual(person.name, "Test User")
self.assertEqual(person.age, 42)
@unittest.skip("not implemented")
def test_mixed_creation(self):
"""Ensure that document may be created using mixed arguments.
"""
@ -2307,8 +2328,8 @@ class InstanceTest(unittest.TestCase):
Person(name="Harry Potter").save()
person = Person.objects.first()
self.assertTrue('id' in person._data.keys())
self.assertEqual(person._data.get('id'), person.id)
self.assertTrue('id' in person.to_dict().keys())
self.assertEqual(person.to_dict().get('id'), person.id)
def test_complex_nesting_document_and_embedded_document(self):

1
tests/document/json_serialisation.py
View file

@ -58,7 +58,6 @@ class TestJson(unittest.TestCase):
reference_field = ReferenceField(Simple, default=lambda:
Simple().save())
map_field = MapField(IntField(), default=lambda: {"simple": 1})
decimal_field = DecimalField(default=1.0)
complex_datetime_field = ComplexDateTimeField(default=datetime.now)
url_field = URLField(default="http://mongoengine.org")
dynamic_field = DynamicField(default=1)

5
tests/document/validation.py
View file

@ -53,11 +53,12 @@ class ValidatorErrorTest(unittest.TestCase):
self.assertEqual(error.message, "root(2nd.3rd.4th.Inception: ['1st'])")
def test_model_validation(self):
class User(Document):
username = StringField(primary_key=True)
name = StringField(required=True)
User.drop_collection()
try:
User().validate()
except ValidationError, e:
@ -128,7 +129,7 @@ class ValidatorErrorTest(unittest.TestCase):
Doc(id="test", e=SubDoc(val=15)).save()
doc = Doc.objects.first()
keys = doc._data.keys()
keys = doc.to_dict().keys()
self.assertEqual(2, len(keys))
self.assertTrue('e' in keys)
self.assertTrue('id' in keys)