From b0b438250e6ed6b67665261499cfcf545188ec93 Mon Sep 17 00:00:00 2001 From: Tom Price Date: Tue, 31 Mar 2015 22:13:31 +0100 Subject: [PATCH] Add item total cost test --- RIGS/test_models.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/RIGS/test_models.py b/RIGS/test_models.py index 658b0985..0d926f6e 100644 --- a/RIGS/test_models.py +++ b/RIGS/test_models.py @@ -167,4 +167,19 @@ class EventTestCase(TestCase): self.assertEqual(event.status, models.Event.CONFIRMED) self.assertTrue(event.confirmed) event.status = models.Event.PROVISIONAL - event.save() \ No newline at end of file + event.save() + +class EventItemTestCase(TestCase): + def setUp(self): + self.e1 = models.Event.objects.create(name="TI E1", start_date=date.today()) + self.e2 = models.Event.objects.create(name="TI E2", start_date=date.today()) + + def test_item_cost(self): + item = models.EventItem.objects.create(event=self.e1, name="TI I1", quantity=1, cost=1.00, order=1) + self.assertEqual(item.total_cost, 1.00) + + item.cost = 2.50 + self.assertEqual(item.total_cost, 2.50) + + item.quantity = 4 + self.assertEqual(item.total_cost, 10.00) \ No newline at end of file