from datetime import timedelta
import re

from django.utils import timezone
from django.test import TestCase

from coupons.models import Coupon
from coupons.settings import (
    CODE_LENGTH,
    CODE_CHARS,
    SEGMENT_LENGTH,
    SEGMENT_SEPARATOR,
)


class CouponTestCase(TestCase):
    def test_generate_code(self):
        self.assertIsNotNone(re.match("^[%s]{%d}" % (CODE_CHARS, CODE_LENGTH,), Coupon.generate_code()))

    def test_generate_code_segmented(self):
        num_segments = CODE_LENGTH // SEGMENT_LENGTH  # full ones
        num_rest = CODE_LENGTH - num_segments * SEGMENT_LENGTH
        self.assertIsNotNone(
            re.match(
                "^([{chars}]{{{sl}}}{sep}){{{ns}}}[{chars}]{{{nr}}}$".format(
                    chars=CODE_CHARS,
                    sep=SEGMENT_SEPARATOR,
                    sl=SEGMENT_LENGTH,
                    ns=num_segments,
                    nr=num_rest),
                Coupon.generate_code("", True)
            )
        )

    def test_save(self):
        coupon = Coupon(type='monetary', value=100)
        coupon.save()
        self.assertTrue(coupon.pk)

    def test_create_coupon(self):
        coupon = Coupon.objects.create_coupon('monetary', 100)
        self.assertTrue(coupon.pk)

    def test_create_coupons(self):
        coupons = Coupon.objects.create_coupons(50, 'monetary', 100)
        for coupon in coupons:
            self.assertTrue(coupon.pk)

    def test_redeem(self):
        coupon = Coupon.objects.create_coupon('monetary', 100)
        coupon.redeem()
        self.assertIsNotNone(coupon.redeemed_at)

    def test_expired(self):
        coupon = Coupon.objects.create_coupon('monetary', 100)
        self.assertFalse(coupon.expired())
        self.assertEqual(Coupon.objects.expired().count(), 0)
        coupon.valid_until = timezone.now()-timedelta(1)
        coupon.save()
        self.assertTrue(coupon.expired())
        self.assertEqual(Coupon.objects.expired().count(), 1)

    def test_str(self):
        coupon = Coupon.objects.create_coupon('monetary', 100)
        self.assertEqual(coupon.code, str(coupon))

    def test_prefix(self):
        coupon = Coupon.objects.create_coupon('monetary', 100, None, None, "prefix-")
        self.assertTrue(coupon.code.startswith("prefix-"))

    def test_used_unused(self):
        coupon = Coupon.objects.create_coupon('monetary', 100)
        self.assertEqual(Coupon.objects.used().count(), 0)
        self.assertEqual(Coupon.objects.unused().count(), 1)
        coupon.redeemed_at = timezone.now()
        coupon.save()
        self.assertEqual(Coupon.objects.used().count(), 1)
        self.assertEqual(Coupon.objects.unused().count(), 0)

@@ -136,16 +136,26 @@
 t Coupon
+, Campaign
 %0Afrom co
@@ -2680,24 +2680,197 @@
 ts.unused().count(), 0)%0A
+%0A%0Aclass CampaignTestCase(TestCase):%0A    def test_str(self):%0A        campaign = Campaign(name=%22test%22)%0A        campaign.save()%0A        self.assertEqual(%22test%22, str(campaign))%0A

add simple CampaignTestCase