added tests for company viewset

This commit is contained in:
Sam
2021-01-22 11:42:24 +00:00
parent 40e6c09d10
commit 1df9aac998
4 changed files with 306 additions and 1 deletions

34
companies/factories.py Normal file
View File

@@ -0,0 +1,34 @@
import datetime
from django.utils import timezone
from factory import LazyAttribute, SubFactory
from factory.fuzzy import FuzzyText, FuzzyInteger, FuzzyDateTime, FuzzyDate, FuzzyDecimal, FuzzyChoice
from factory.django import DjangoModelFactory
from companies.models import Company
class CompanyFactory(DjangoModelFactory):
cif = FuzzyText(prefix='CIF_', length=10)
company_name = FuzzyText(prefix='COMPANY_NAME_', length=10)
short_name = FuzzyText(prefix='SHORT_NAME_', length=10)
web_link = FuzzyText(prefix='http://WEB_LINK_', suffix='.test', length=10)
shop = FuzzyChoice(choices=(True, False))
shop_link = FuzzyText(prefix='http://SHOP_LINK_', suffix='.test', length=10)
platform = FuzzyChoice(choices=[x[1] for x in Company.PLATFORMS])
email = FuzzyText(prefix='EMAIL_', suffix='@test.com', length=10)
logo = None
city = None
address = FuzzyText(prefix='ADDRESS_', length=10)
geo = None
phone = '+34666555444'
mobile = '+34666555333'
other_phone = '+34666555222'
description = FuzzyText(prefix='DESCRIPTION_', length=250)
shop_rss_feed = FuzzyText(prefix='http://SHOP_RSS_FEED_', suffix='.test', length=10)
sale_terms = FuzzyText(prefix='SALES_TERMS', length=250)
shipping_cost = FuzzyDecimal(low=1.00)
# tags = models.ManyToMany(Tag, null=True)
sync = FuzzyChoice(choices=(True, False))
class Meta:
model = 'companies.Company'