48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
# 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())
|
|
|
|
|
|
|
|
|