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_unregistr(self):
DiscountRegistr.discounts = {'test': DummyDiscount}
DiscountRegistr.choices = {'test': 'test'}
DiscountRegistr.unregister(DummyDiscount)
self.assertFalse('test' in DiscountRegistr.discounts)
self.assertFalse('test' in DiscountRegistr.choices)
def test_registr(self):
v1 = DummyDiscount
DiscountRegistr.discounts = {}
DiscountRegistr.register(v1)
self.assertTrue('test' in DiscountRegistr.discounts)
self.assertEqual(DiscountRegistr.discounts['test'], v1)
