class PricingServiceTestCase(TestCase):
def setUp(self):
self.repo = Mock()
self.repo.find = Mock()
self.repo.find = Mock(side_effect=lambda category, tier: discounts.get((category, tier)))
discounts = {
("MANF-A", "GRADE-A"): Discount(0.3, "MANF-A", "GRADE-A")
}
self.service = PricingService(self.repo)
def test_customer_discount(self):
product = ProductFactory.build(price_category="MANF-A")
customer = CustomerFactory.build(discount_tier="GRADE-A")
discount = self.service.get_customer_discount(product, customer)
self.assertEquals(0.3, discount, "Wrong discount calculated")
self.repo.find.assert_called_with("MANF-A", "GRADE-A")
def test_discount_nonexistent(self):
with self.assertRaises(PricingError):
self.service.get_customer_discount(ProductFactory.build(price_category="MANF-A"),
CustomerFactory.build(discount_tier="FAKE"))
with self.assertRaises(PricingError):
self.service.get_customer_discount(ProductFactory.build(price_category="FAKE"),
CustomerFactory.build(discount_tier="GRADE-A"))
