@@ -1,47 +0,0 @@
|
||||
# import pickle
|
||||
# import pymongo
|
||||
# import bson
|
||||
# import warnings
|
||||
|
||||
# from datetime import datetime
|
||||
|
||||
# import tempfile
|
||||
# import pymongo, gridfs
|
||||
|
||||
import unittest
|
||||
from mongoengine import *
|
||||
from bson.objectid import ObjectId
|
||||
|
||||
class BugFixTest(unittest.TestCase):
|
||||
|
||||
|
||||
def setUp(self):
|
||||
|
||||
conn = connect(db='mongoenginetest')
|
||||
|
||||
def test_items_list(self):
|
||||
|
||||
class ActivityType1(EmbeddedDocument):
|
||||
activity_id = IntField()
|
||||
activity_name = StringField()
|
||||
|
||||
class ActivityType2(EmbeddedDocument):
|
||||
activity_id = IntField()
|
||||
activity_status = StringField()
|
||||
|
||||
class UserActivities(Document):
|
||||
user_id = IntField()
|
||||
activity = GenericEmbeddedDocumentField(choices=(ActivityType1, ActivityType2))
|
||||
|
||||
|
||||
UserActivities.drop_collection()
|
||||
|
||||
user_id = 123
|
||||
activity_id = 321
|
||||
UserActivities(user_id=user_id, activity=ActivityType2(activity_id=activity_id, activity_status="A")).save()
|
||||
|
||||
self.assertEquals(1, UserActivities.objects(user_id=user_id, __raw__={'activity.activity_status': 'A'}).count())
|
||||
|
||||
|
||||
|
||||
|
||||
@@ -664,6 +664,26 @@ class DocumentTest(unittest.TestCase):
|
||||
|
||||
BlogPost.drop_collection()
|
||||
|
||||
def test_db_field_load(self):
|
||||
"""Ensure we load data correctly
|
||||
"""
|
||||
class Person(Document):
|
||||
name = StringField(required=True)
|
||||
_rank = StringField(required=False, db_field="rank")
|
||||
|
||||
@property
|
||||
def rank(self):
|
||||
return self._rank or "Private"
|
||||
|
||||
Person.drop_collection()
|
||||
|
||||
Person(name="Jack", _rank="Corporal").save()
|
||||
|
||||
Person(name="Fred").save()
|
||||
|
||||
self.assertEquals(Person.objects.get(name="Jack").rank, "Corporal")
|
||||
self.assertEquals(Person.objects.get(name="Fred").rank, "Private")
|
||||
|
||||
def test_explicit_geo2d_index(self):
|
||||
"""Ensure that geo2d indexes work when created via meta[indexes]
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user