Exemplo n.º 1
0
    def test_review_comments(self):
        """Test commenting on a review"""

        r1 = Review(
            drink_id=self.d1id,
            rating=3.5,
            review="Test Review",
            user_id=self.uid,
            image=
            "https://images.unsplash.com/photo-1561150169-371f366b828a?ixid=MXwxMjA3fDB8MHxzZWFyY2h8MXx8YWxjb2hvbHxlbnwwfHwwfA%3D%3D&ixlib=rb-1.2.1&w=1000&q=80"
        )

        r2 = Review(
            drink_id=self.d1id,
            rating=3.3,
            review="Another review",
            user_id=self.uid,
            image=
            "https://images.unsplash.com/photo-1561150169-371f366b828a?ixid=MXwxMjA3fDB8MHxzZWFyY2h8MXx8YWxjb2hvbHxlbnwwfHwwfA%3D%3D&ixlib=rb-1.2.1&w=1000&q=80"
        )

        r1.id = 8888
        r2.id = 5599
        db.session.add(r1)
        db.session.add(r2)

        db.session.commit()

        c1 = Comment(user_id=self.uid, reviews_id=r1.id, text="Comment on r1")

        c2 = Comment(user_id=self.uid, reviews_id=r2.id, text="Comment on r2")

        db.session.add(c1)
        db.session.add(c2)

        db.session.commit()

        comments = Comment.query.filter(Comment.user_id == self.uid).all()
        self.assertEqual(len(comments), 2)

        comment1 = Comment.query.filter(Comment.reviews_id == r1.id).all()
        self.assertEqual(len(comment1), 1)
        self.assertEqual(comment1[0].reviews_id, r1.id)
        self.assertEqual(comment1[0].text, "Comment on r1")
        self.assertNotEqual(comment1[0].text, "Comment on r2")
        # self.assertNotEqual(l[0].reviews_id, r2.id)

        comment2 = Comment.query.filter(Comment.reviews_id == r2.id).all()
        self.assertEqual(len(comment2), 1)
        self.assertEqual(comment2[0].reviews_id, r2.id)
        self.assertEqual(comment2[0].text, "Comment on r2")
        self.assertNotEqual(comment2[0].text, "Comment on r1")
Exemplo n.º 2
0
    def test_delete_review(self):
        """Can a user delete their reviews?"""
        with self.client as c:
            with c.session_transaction() as sess:
                sess[CURR_USER_KEY] = self.testuser.id
            review = Review(location='9191 S Polk Street Dallas', description='Tester', user_id=self.testuser.id)
            review.id = 1
            db.session.add(review)
            db.session.commit()

            resp = c.post(f'/delete/review/{review.id}', follow_redirects=True)

            html = resp.get_data(as_text=True)

            self.assertEqual(resp.status_code, 200)
            self.assertIn('Welcome', html)