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

Added some tests #384

Browse Source
This commit is contained in:
Ross Lawley 2011-12-07 01:14:43 -08:00
parent 8d2bc444bb
commit fbe1901e65
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/document.py
View File

@ -2559,15 +2559,29 @@ class DocumentTest(unittest.TestCase):
author = ReferenceField(User) author = ReferenceField(User)
book = ReferenceField(Book) book = ReferenceField(Book)
meta = {"db_alias" : "testdb-2" } meta = {"db_alias" : "testdb-2" }
# Drops # Drops
AuthorBooks.drop_collection() AuthorBooks.drop_collection()
ab = AuthorBooks.objects.create( author = bob, book = hp) ab = AuthorBooks.objects.create( author = bob, book = hp)
# select # select
self.assertEqual( AuthorBooks.objects.first(), ab) self.assertEqual( AuthorBooks.objects.first(), ab)
self.assertEqual( AuthorBooks.objects.first().book, hp) self.assertEqual( AuthorBooks.objects.first().book, hp)
self.assertEqual( AuthorBooks.objects.first().author, bob) self.assertEqual( AuthorBooks.objects.first().author, bob)
self.assertEqual( AuthorBooks.objects.filter(author = bob).first() , ab)
self.assertEqual( AuthorBooks.objects.filter(book = hp).first() , ab)
# DB Alias
self.assertEqual( User._get_db() , get_db())
self.assertEqual( Book._get_db() , get_db("testdb-1"))
self.assertEqual( AuthorBooks._get_db() , get_db("testdb-2"))
# Collections
self.assertEqual( User._get_collection() , get_db()[User._get_collection_name()] )
self.assertEqual( Book._get_collection() , get_db("testdb-2")[Book._get_collection_name()] )
self.assertEqual( AuthorBooks._get_collection() , get_db("testdb-2")[AuthorBooks._get_collection_name()] )