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")
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)