Пример #1
0
    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"))
Пример #2
0
    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())