kopia lustrzana https://github.com/longclawshop/longclaw
131 wiersze
4.4 KiB
Python
131 wiersze
4.4 KiB
Python
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.longclawproducts.models import Product
|
|
from longclaw.longclawbasket.models import BasketItem
|
|
from longclaw.longclaworders.models import Order, OrderItem
|
|
from longclaw.longclawshipping.models import Address, Country, ShippingRate
|
|
from longclaw.utils import ProductVariant
|
|
|
|
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 = Product
|
|
|
|
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))
|
|
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
|