def test_try_connect_as_user(self): """Test user credentials.""" # Create a new user and save it User.objects.create( email="*****@*****.**", password="******", fullname="Bob") # Test self.assertTrue(User.connect("*****@*****.**", "secret")) self.assertFalse(User.connect("*****@*****.**", "badpassword")) self.assertFalse(User.connect("*****@*****.**", "secret"))
def test_full(self): """Test data from fixtures.""" import_data_cmd = "loaddata" management.call_command(import_data_cmd, "data.yaml") # Count things self.assertEqual(2, User.objects.count()) self.assertEqual(3, Post.objects.count()) self.assertEqual(3, Comment.objects.count()) # Try to connect as users self.assertTrue(User.connect("*****@*****.**", "secret")) self.assertTrue(User.connect("*****@*****.**", "secret")) self.assertFalse(User.connect("*****@*****.**", "badpassword")) self.assertFalse(User.connect("*****@*****.**", "secret")) # Find all of Bob's posts bobPosts = Post.objects.filter(author__email="*****@*****.**") self.assertEqual(2, bobPosts.count()) # Find all comments related to Bob's posts bobComments = \ Comment.objects.filter(post__author__email="*****@*****.**") self.assertEqual(3, bobComments.count()) # Find the most recent post frontPost = Post.objects.latest("postedAt") self.assertTrue(frontPost) self.assertEqual("About the model layer", frontPost.title) # Check that this post has two comments self.assertEqual(2, frontPost.comment_set.count()) # Post a new comment frontPost.addComment("Jim", "Hello guys") self.assertEqual(3, frontPost.comment_set.count()) self.assertEqual(4, Comment.objects.count())