예제 #1
0
    def test_food_and_atmosphere_rating(self):
        vendor = Vendor(name="Test Vendor")
        vendor.save()

        self.assertEqual(vendor.food_rating(), None)
        self.assertEqual(vendor.atmosphere_rating(), None)

        Review(vendor=vendor, approval_status=SF.APPROVED, food_rating=1, atmosphere_rating=1, author=self.user).save()

        self.assertEqual(vendor.food_rating(), 1)
        self.assertEqual(vendor.atmosphere_rating(), 1)

        review2 = Review(
            vendor=vendor, approval_status=SF.PENDING, food_rating=4, atmosphere_rating=4, author=self.user
        )
        review2.save()

        self.assertEqual(vendor.food_rating(), 1)
        self.assertEqual(vendor.atmosphere_rating(), 1)

        review2.approval_status = SF.APPROVED
        review2.save()

        # Floored Average
        self.assertEqual(vendor.food_rating(), 2)
        self.assertEqual(vendor.atmosphere_rating(), 2)

        review3 = Review(
            vendor=vendor, approval_status=SF.APPROVED, food_rating=4, atmosphere_rating=4, author=self.user
        )
        review3.save()

        # Floored Average
        self.assertEqual(vendor.food_rating(), 3)
        self.assertEqual(vendor.atmosphere_rating(), 3)
예제 #2
0
    def test_food_and_atmosphere_rating(self):
        vendor = Vendor(name="Test Vendor")
        vendor.save()

        self.assertEqual(vendor.food_rating(), None)
        self.assertEqual(vendor.atmosphere_rating(), None)

        Review(vendor=vendor,
               approved=True,
               food_rating=1,
               atmosphere_rating=1,
               author=self.user).save()

        self.assertEqual(vendor.food_rating(), 1)
        self.assertEqual(vendor.atmosphere_rating(), 1)

        review2 = Review(vendor=vendor,
                         approved=False,
                         food_rating=4,
                         atmosphere_rating=4,
                         author=self.user)
        review2.save()

        self.assertEqual(vendor.food_rating(), 1)
        self.assertEqual(vendor.atmosphere_rating(), 1)

        review2.approved = True
        review2.save()

        # Floored Average
        self.assertEqual(vendor.food_rating(), 2)
        self.assertEqual(vendor.atmosphere_rating(), 2)

        review3 = Review(vendor=vendor,
                         approved=True,
                         food_rating=4,
                         atmosphere_rating=4,
                         author=self.user)
        review3.save()

        # Floored Average
        self.assertEqual(vendor.food_rating(), 3)
        self.assertEqual(vendor.atmosphere_rating(), 3)