def test_uservalidator_case3(self):
users = create_users()
ds = create_discounts()
ds[0].valid_users.add(users[0])
validator = UserValidator()
qs = Discount._base_manager.filter(validator.get_q_object(users[0]))
self.assertEqual(set(ds), set(qs))
def test_clean(self):
ds = create_discounts()[0]
users = users = create_users()
validator = UserValidator()
self.assertEqual(ds, validator.clean(ds, users[0]))
ds.valid_users.add(users[0])
self.assertRaises(ValidationError, validator.clean, ds, users[1])
self.assertEqual(ds, validator.clean(ds, users[0]))
