import factory
from django.core.urlresolvers import reverse

from rest_framework.test import APITestCase
from rest_framework import status

from wagtail_factories import PageFactory


from longclaw.longclawbasket.models import BasketItem
from longclaw.longclaworders.models import Order
from longclaw.longclawshipping.models import Address, Country, ShippingRate
from longclaw.utils import ProductVariant, maybe_get_product_model

class OrderFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Order

class CountryFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Country

    iso = factory.Faker('pystr', max_chars=2, min_chars=2)
    name_official = factory.Faker('text', max_nb_chars=128)
    name = factory.Faker('text', max_nb_chars=128)


class AddressFactory(factory.django.DjangoModelFactory):

    class Meta:
        model = Address

    name = factory.Faker('text', max_nb_chars=64)
    line_1 = factory.Faker('text', max_nb_chars=128)
    line_2 = factory.Faker('text', max_nb_chars=128)
    city = factory.Faker('text', max_nb_chars=64)
    postcode = factory.Faker('text', max_nb_chars=10)
    country = factory.SubFactory(CountryFactory)

class ShippingRateFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = ShippingRate

    name = factory.Faker('text', max_nb_chars=32)
    rate = 1.0
    carrier = 'Royal Mail'
    description = 'Test'

    @factory.post_generation
    def countries(self, create, extracted, **kwargs):
        if not create:
            # Simple build, do nothing.
            return

        if extracted:
            # A list of countries were passed in, use them
            for country in extracted:
                self.countries.add(country)

class ProductFactory(PageFactory):
    ''' Create a random Product
    '''

    class Meta:
        model = maybe_get_product_model()

    title = factory.Faker('sentence', nb_words=1)
    description = factory.Faker('text')

    @classmethod
    def _create(cls, model_class, *args, **kwargs):
        kwargs['parent'] = None
        return super(ProductFactory, cls)._create(model_class, *args, **kwargs)

class ProductVariantFactory(factory.django.DjangoModelFactory):

    class Meta:
        model = ProductVariant

    product = factory.SubFactory(ProductFactory)
    description = factory.Faker('text')
    price = factory.Faker('pyfloat', positive=True, left_digits=2, right_digits=2)
    ref = factory.Faker('pystr', min_chars=3, max_chars=10)
    stock = factory.Faker('pyint')


class BasketItemFactory(factory.django.DjangoModelFactory):

    class Meta:
        model = BasketItem

    quantity = 1
    variant = factory.SubFactory(ProductVariantFactory)

class LongclawTestCase(APITestCase):

    def get_test(self, urlname, urlkwargs=None, **kwargs):
        ''' Submit a GET request and assert the response status code is 200

        Arguments:
            urlname (str): The url name to pass to the 'reverse' function
            urlkwargs (dict): The `kwargs` parameter to pass to the `reverse` function
        '''
        response = self.client.get(reverse(urlname, kwargs=urlkwargs), **kwargs)
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        return response

    def post_test(self, data, urlname, urlkwargs=None, **kwargs):
        ''' Submit a POST request and assert the response status code is 201

        Arguments:
            data (dict): The data to pass to the post request
            urlname (str): The url name to pass to the 'reverse' function
            urlkwargs (dict): The `kwargs` parameter to pass to the `reverse` function
        '''
        response = self.client.post(reverse(urlname, kwargs=urlkwargs), data, **kwargs)
        self.assertIn(response.status_code,
                      (status.HTTP_201_CREATED, status.HTTP_200_OK, status.HTTP_204_NO_CONTENT))
        return response

    def patch_test(self, data, urlname, urlkwargs=None, **kwargs):
        ''' Submit a PATCH request and assert the response status code is 200
        '''
        response = self.client.patch(reverse(urlname, kwargs=urlkwargs), data, **kwargs)
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        return response

    def put_test(self, data, urlname, urlkwargs=None, **kwargs):
        response = self.client.put(reverse(urlname, kwargs=urlkwargs), data, **kwargs)
        self.assertEqual(response.status_code, status.HTTP_202_ACCEPTED)
        return response

    def del_test(self, urlname, urlkwargs=None, **kwargs):
        response = self.client.delete(reverse(urlname, kwargs=urlkwargs), **kwargs)
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        return response