def test_remove_document(self): BlogPost(valid_doc()).save() BlogPost(valid_doc()).save() blog_post = BlogPost(valid_doc()) blog_post.save() with self.assert_difference(blog_post_collection.count, -1): blog_post.remove()
def setUp(self): self.document_1 = valid_doc() self.document_2 = valid_doc(overrides={"author": {"first_name": "John", "last_name": "Humphreys", "birth_year": 1978, "birth_month": 8, "birth_day": 15}})
def test_update(self): other = BlogPost(valid_doc()) other.save() blog_post = BlogPost(valid_doc()) blog_post.save() blog_post.update({"likes": 5}) reloaded_blog_post = BlogPost.find_by_id(blog_post['_id']) reloaded_other_blog_post = BlogPost.find_by_id(other['_id']) self.assertEqual(5, reloaded_blog_post['likes']) self.assertEqual(0, reloaded_other_blog_post['likes'])
def setUp(self): self.document_1 = valid_doc() self.document_2 = valid_doc( overrides={ "author": { "first_name": "John", "last_name": "Humphreys", "birth_year": 1978, "birth_month": 8, "birth_day": 15 } })
def test_update(self): other = BlogPost(valid_doc()) other.save() blog_post = BlogPost(valid_doc()) blog_post.save() blog_post.update_instance({'$set': {'likes': 5}}) BlogPost.update({'_id': blog_post['_id']}, {'$set': {'misc': 'important'}}) reloaded_blog_post = BlogPost.find_by_id(blog_post['_id']) reloaded_other_blog_post = BlogPost.find_by_id(other['_id']) self.assertEqual(5, reloaded_blog_post['likes']) self.assertEqual('important', reloaded_blog_post['misc']) self.assertEqual(0, reloaded_other_blog_post['likes']) self.assertNotIn('misc', reloaded_other_blog_post)
def test_update(self): other = BlogPost(valid_doc()) other.save() blog_post = BlogPost(valid_doc()) blog_post.save() blog_post.update_instance({'$set': {'likes': 5}}) BlogPost.update({'_id': blog_post['_id']}, {'$set': { 'misc': 'important' }}) reloaded_blog_post = BlogPost.find_by_id(blog_post['_id']) reloaded_other_blog_post = BlogPost.find_by_id(other['_id']) self.assertEqual(5, reloaded_blog_post['likes']) self.assertEqual('important', reloaded_blog_post['misc']) self.assertEqual(0, reloaded_other_blog_post['likes']) self.assertNotIn('misc', reloaded_other_blog_post)
def test_does_not_persist_invalid_document(self): blog_post = BlogPost(valid_doc()) del blog_post['author'] with self.assert_no_difference(blog_post_collection.count): try: blog_post.save() except: pass
def test_save_then_load_new_document(self): blog_post = BlogPost(valid_doc()) with self.assert_difference(blog_post_collection.count, 1): blog_post.save() self.assertIsNotNone(blog_post['_id']) reloaded_blog_post = BlogPost.find_by_id(blog_post['_id']) self.assertDictEqual(expected_db_doc(blog_post['_id']), reloaded_blog_post)
def test_save_then_load_new_document(self): blog_post = BlogPost(valid_doc()) with self.assert_difference(blog_post_collection.count, 1): blog_post.save() self.assertIsNotNone(blog_post['_id']) reloaded_blog_post = BlogPost.find_by_id(blog_post['_id']) self.assertDictEqual( expected_db_doc(blog_post['_id']), reloaded_blog_post)
def test_find_and_modify_selection(self): for i in range(5): BlogPost(valid_doc()).save() posts = BlogPost.find() self.assertEquals(posts.count(), 5) post = posts[0] del post['author'] try: post.validate() except: pass # expected
def test_update_existing_document(self): blog_post = BlogPost(valid_doc()) blog_post.save() blog_post['author']['first'] = "Troy" with self.assert_no_difference(blog_post_collection.count): blog_post.save() # Reload to check the change is there reloaded_blog_post = BlogPost.find_by_id(blog_post['_id']) expected = expected_db_doc(blog_post['_id']) expected['author']['first'] = u"Troy" self.assertDictEqual(expected, reloaded_blog_post)
def test_update_existing_document(self): blog_post = BlogPost(valid_doc()) blog_post.save() blog_post['author']['first'] = "Troy" with self.assert_no_difference(blog_post_collection.count): blog_post.save() # Reload to check the change is there reloaded_blog_post = BlogPost.find_by_id(blog_post['_id']) expected = expected_db_doc(blog_post['_id']) expected['author']['first'] = u"Troy" self.assertDictEqual( expected, reloaded_blog_post)
def test_count(self): for i in range(5): BlogPost(valid_doc()).save() self.assertEqual(5, BlogPost.count())
def test_find_one(self): for i in range(10): BlogPost(valid_doc({"likes": i})).save() post = BlogPost.find_one({"likes": {"$gte": 5}}) self.assertIsInstance(post, BlogPost)
def setUp(self): self.document = valid_doc()
def test_find_query(self): for i in range(10): BlogPost(valid_doc({"likes": i})).save() posts = BlogPost.find({"likes": {"$gte": 5}}) self.assertEquals(5, posts.count())