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]))