def test_get_choices(self): DiscountRegistr.discounts = {} DiscountRegistr.choices = {} DiscountRegistr.register(DummyDiscount) DiscountRegistr.register(DummyDiscount2) self.assertEqual(sorted(list(DiscountRegistr.get_choices())), sorted([('test', 'test'), ('test2', 'test2')]))
from salest.discounts.validators import DiscountRegistr from salest.discounts.discounts import CartCodeMinOrderDiscount DiscountRegistr.register(CartCodeMinOrderDiscount)
def test_force_registr(self): DiscountRegistr.discounts = {} DiscountRegistr.choices = {} DiscountRegistr.register(DummyDiscount) self.assertTrue('test' in DiscountRegistr.discounts) self.assertEqual(DiscountRegistr.discounts['test'], DummyDiscount)
def test_registr(self): v1 = DummyDiscount DiscountRegistr.discounts = {} DiscountRegistr.register(v1) self.assertTrue('test' in DiscountRegistr.discounts) self.assertEqual(DiscountRegistr.discounts['test'], v1)