def create_vouchers(self, partner=None, count=1): """Helper function that creates vouchers with a mocked coupon relation.""" vouchers = VoucherFactory.create_batch(count) partner = partner or self.partner coupon_vouchers = CouponVouchers.objects.create(coupon=ProductFactory( stockrecords__partner=partner)) for voucher in vouchers: voucher.offers.add(ConditionalOfferFactory()) coupon_vouchers.vouchers.add(voucher) return vouchers
def test_list(self): """ Verify the endpoint lists all vouchers. """ vouchers = VoucherFactory.create_batch(3) for voucher in vouchers: voucher.offers.add(ConditionalOfferFactory()) response = self.client.get(self.path) self.assertEqual(response.data['count'], len(vouchers)) actual_codes = [datum['code'] for datum in response.data['results']] expected_codes = [voucher.code for voucher in vouchers] self.assertEqual(actual_codes, expected_codes)