mirror of
https://github.com/narrrl/holy-flavor-rater.git
synced 2026-11-03 16:41:27 +01:00
20 lines
844 B
Python
20 lines
844 B
Python
from django.test import TestCase
|
|
|
|
from api.models import Category, Flavor, Rating, User
|
|
|
|
|
|
class FlavorAverageRatingTests(TestCase):
|
|
def setUp(self) -> None:
|
|
self.category = Category.objects.create(name="Energy", slug="energy")
|
|
self.flavor = Flavor.objects.create(name="Test", category=self.category)
|
|
self.u1 = User.objects.create_user(username="u1", password="x")
|
|
self.u2 = User.objects.create_user(username="u2", password="x")
|
|
|
|
def test_average_zero_when_no_ratings(self) -> None:
|
|
self.assertEqual(self.flavor.get_average_rating, 0.0)
|
|
|
|
def test_average_uses_aggregate(self) -> None:
|
|
Rating.objects.create(user=self.u1, flavor=self.flavor, score=4)
|
|
Rating.objects.create(user=self.u2, flavor=self.flavor, score=8)
|
|
self.assertEqual(self.flavor.get_average_rating, 6.0)
|