class PeriodPriceModifierFactory(factory.DjangoModelFactory):
class Meta:
model = PeriodPriceModifier
product = factory.SubFactory(ProductFactory)
modifier = FuzzyDecimal(1, 200)
start_date = datetime.date(2015, 11, 1)
end_date = datetime.date(2015, 11, 30)
class CoursePriceFactory(DjangoModelFactory):
"""Factory for CoursePrice"""
course_run = SubFactory(CourseRunFactory)
is_valid = FuzzyAttribute(FAKE.boolean)
price = FuzzyDecimal(low=0, high=12345)
class Meta: # pylint: disable=missing-docstring,no-init,too-few-public-methods,old-style-class
model = CoursePrice
class GuessFactory(DjangoModelFactory):
class Meta:
model = Guess
user = SubFactory(UserFactory)
market = SubFactory(MarketFactory)
value = FuzzyDecimal(1, 10000)
score = LazyAttribute(lambda o: o.get_score())
class ProductFactory(DjangoModelFactory):
name = FuzzyText()
price = FuzzyDecimal(0.99, 99.99)
stock = FuzzyInteger(0, 200)
category = SubFactory(CategoryFactory)
class Meta:
model = Product
class RateFactory(factory.DjangoModelFactory):
class Meta:
model = Rate
workshop = factory.SubFactory(WorkshopFactory)
name = FuzzyText(length=50)
price = FuzzyDecimal(0.5, 1000.0)
capacity = FuzzyInteger(1, 100)
class InstituteQuotaFactory(DjangoModelFactory):
institute = factory.SubFactory(InstituteFactory)
machine_category = factory.SubFactory(MachineCategoryFactory)
quota = FuzzyDecimal(0.0, 999.0)
class Meta:
model = karaage.institutes.models.InstituteQuota
django_get_or_create = ('institute', 'machine_category')
class CourseSkillsFactory(factory.django.DjangoModelFactory):
course_key = FuzzyText()
skill = factory.SubFactory(SkillFactory)
confidence = FuzzyDecimal(0.0, 1.0)
is_blacklisted = False
class Meta:
model = CourseSkills
class TransactionFactory(factory.django.DjangoModelFactory):
class Meta:
model = 'transactions.Transaction'
id = random.randrange(200, 99999)
account = factory.SubFactory(AccountFactory)
date = FuzzyDateTime(datetime.datetime.utcnow().replace(tzinfo=pytz.utc))
amount = FuzzyDecimal(0.5, 9999.9)
class ComponentFactory(factory.alchemy.SQLAlchemyModelFactory):
id = factory.Sequence(lambda n: "%s" % n)
name = FuzzyChoice(["methane", "ethane", "propane"])
value = FuzzyDecimal(100)
class Meta:
model = ComponentModel
sqlalchemy_session = session_factory
class CourseEntitlementFactory(factory.DjangoModelFactory):
mode = FuzzyChoice([name for name, __ in CourseEntitlement.COURSE_MODE_CHOICES])
price = FuzzyDecimal(1.0, 650.0)
currency = factory.Iterator(Currency.objects.all())
course = factory.SubFactory(CourseFactory)
class Meta:
model = CourseEntitlement
class SeatFactory(factory.DjangoModelFactory):
type = FuzzyChoice([name for name, __ in Seat.SEAT_TYPE_CHOICES])
price = FuzzyDecimal(0.0, 650.0)
currency = factory.Iterator(Currency.objects.all())
upgrade_deadline = FuzzyDateTime(datetime(2014, 1, 1, tzinfo=UTC))
course_run = factory.SubFactory(CourseRunFactory)
class Meta:
model = Seat
class PolicyFactory(SQLAFactory):
class Meta:
model = Policy
name = Sequence('policy{}'.format)
description = MimesisField('text', quantity=5)
majority = FuzzyChoice(list(Majority))
proposition_expiration = FuzzyInteger(0, 10000)
qualification_quorum = FuzzyDecimal(0, 1)
qualification_minimum = FuzzyInteger(0, 1000)
range_max = FuzzyInteger(6, 10)
range_small_max = FuzzyInteger(3, 5)
range_small_options = FuzzyInteger(1, 10)
secret_minimum = FuzzyInteger(0, 1000)
secret_quorum = FuzzyDecimal(0, 1)
submitter_minimum = FuzzyInteger(0, 1000)
voting_duration = FuzzyInteger(0, 10000)
voting_system = FuzzyChoice(list(VotingSystem))
class UserGradeFactory(factory.django.DjangoModelFactory):
class Meta:
model = models.UserGrade
username = factory.Sequence(lambda o: 'robot%d' % o)
course_run = factory.SubFactory(CourseRunFactory)
letter_grade = FuzzyChoice(['A', 'B', 'C', 'D', 'F'])
percent_grade = FuzzyDecimal(0.0, 1.0, precision=4)
verified = True
class EntryFactory(factory.django.DjangoModelFactory):
class Meta:
model = Entry
blog = factory.SubFactory(BlogFactory)
headline = 'Some headline'
body_text = 'Some body text'
number_comments = FuzzyInteger(100)
scoring = FuzzyDecimal(0.00, 9.99)
class UserGradeFactory(factory.django.DjangoModelFactory):
class Meta:
model = models.UserGrade
username = factory.Sequence(lambda o: "robot%d" % o)
course_run = factory.SubFactory(CourseRunFactory)
letter_grade = FuzzyChoice(["A", "B", "C", "D", "F"])
percent_grade = FuzzyDecimal(0.0, 1.0, precision=4)
verified = True
class TransactionFactory(factory.django.DjangoModelFactory):
class Meta:
model = 'payments.Transaction'
credit_card = factory.SubFactory(CreditCardFactory)
stripe_id = factory.Sequence(lambda n: 'stripe_%d' % n)
extra_data = {}
amount = FuzzyDecimal(100)
transaction_type = Transaction.ESCROW
def create_docs(self, create, extracted, **kwargs):
if not create:
return
for i in range(0, 3):
measurement_time = FuzzyDateTime(timezone.now() -
datetime.timedelta(
hours=1)).fuzz()
value = FuzzyDecimal(0, 3).fuzz()
self.measurements.add(time=measurement_time, value=value)
class CourseEntitlementFactory(factory.DjangoModelFactory):
mode = factory.SubFactory(SeatTypeFactory)
price = FuzzyDecimal(0.0, 650.0)
currency = factory.Iterator(Currency.objects.all())
sku = FuzzyText(length=8)
course = factory.SubFactory(CourseFactory)
class Meta:
model = CourseEntitlement
class CourseEntitlementFactory(factory.DjangoModelFactory):
mode = factory.SubFactory(SeatTypeFactory)
price = FuzzyDecimal(0.0, 650.0)
currency = factory.Iterator(Currency.objects.all())
sku = FuzzyText(length=8)
expires = FuzzyDateTime(datetime.datetime(2014, 1, 1, tzinfo=UTC))
course = factory.SubFactory(CourseFactory)
class Meta:
model = CourseEntitlement
class PriceFactory(factory.DjangoModelFactory):
class Meta:
model = Price
django_get_or_create = ('pair', )
exchange = factory.SubFactory('apps.order.factories.ExchangeFactory')
pair = 'BTC-ETH'
ask = FuzzyDecimal(0.025, 2.25)
timestamp = factory.LazyFunction(datetime.datetime.now)
data = "{}"
class PaymentFactory(DjangoModelFactory):
class Meta:
model = "finance.Payment"
order = SubFactory(OrderFactory)
amount = FuzzyDecimal(low=1, high=99)
mollie_id = FuzzyText()
qantani_transaction_id = FuzzyInteger(low=1, high=999)
qantani_transaction_code = FuzzyText()
class TickerFactory(factory.alchemy.SQLAlchemyModelFactory):
class Meta:
model = Ticker
sqlalchemy_session = db_session
sqlalchemy_session_persistence = "commit"
currency_code = factory.Iterator(currency_list())
currency_rate = FuzzyDecimal(
low=pow(10, -DECIMAL_EXPONENT),
high=pow(10, DECIMAL_EXPONENT),
precision=DECIMAL_EXPONENT,
)
amount_requested = FuzzyDecimal(
low=pow(10, -DECIMAL_EXPONENT),
high=pow(10, DECIMAL_EXPONENT),
precision=DECIMAL_EXPONENT,
)
final_amount = factory.LazyAttribute(get_final_amount)
created_at = FuzzyDateTime(start_dt=datetime(2020, 10, 1, tzinfo=UTC),
end_dt=datetime.utcnow())
class BookFactory(factory.DjangoModelFactory):
title = factory.Faker('sentence', nb_words=4)
publisher = factory.SubFactory(PublisherFactory)
publication_date = FuzzyDate(datetime.date(2020, 1, 1))
price = FuzzyDecimal(10, 100, 2)
isbn = factory.Faker('isbn13')
state = FuzzyChoice(dict(BOOK_PUBLISHING_STATUS_CHOICES).keys())
class Meta:
model = Book
class ProductFactory(DjangoModelFactory):
class Meta:
model = "ordering.Product"
name = FuzzyText()
description = FuzzyText()
unit_amount = FuzzyInteger(low=1, high=100)
unit = SubFactory(ProductUnitFactory)
base_price = FuzzyDecimal(0.1, 6.0)
supplier = SubFactory(SupplierFactory)
order_round = SubFactory(OrderRoundFactory)
class IndividualPlanFactory(Factory):
class Meta:
model = IndividualPlan
id = LazyAttribute(lambda _: uuid1())
name = Faker('name', locale='pl_PL')
status = FuzzyChoice(Status)
fee_amount = FuzzyDecimal(low=10)
fee_currency = FuzzyChoice(['PLN', 'USD', 'EUR', 'GBP'])
max_no_of_pauses = plan.cmd.MAX_NUMBER_OF_PAUSES
renewal = FuzzyChoice(Renewal)
class OrderFactory(DjangoModelFactory):
"""Factory for Order"""
user = SubFactory(UserFactory)
status = FuzzyChoice(Order.STATUSES)
total_price_paid = FuzzyDecimal(low=0, high=12345)
class Meta:
model = Order
class Params:
fulfilled = Trait(status=Order.FULFILLED)
class TaskFactory(factory.django.DjangoModelFactory):
class Meta:
model = 'projects.Task'
milestone = factory.SubFactory('projects.tests.factories.MilestoneFactory')
status = FuzzyChoice(Task.STATUS_CHOICES)
name = factory.Sequence(lambda n: 'Test name %d' % n)
description = factory.Sequence(lambda n: 'Test description %d' % n)
due_date = FuzzyDate(timezone.now().date())
assigned = factory.SubFactory('accounts.tests.factories.UserFactory')
amount = FuzzyDecimal(100)
class DishFactory(DjangoModelFactory):
class Meta:
model = Dish
name = factory.Sequence(lambda n: f'name{n}')
description = FuzzyText(length=50)
prepare_time = timedelta(minutes=40)
price = FuzzyDecimal(low=1, high=99)
is_vegetarian = FuzzyChoice([False, True])
menu = factory.SubFactory(MenuFactory)
class SeatFactory(factory.DjangoModelFactory):
type = factory.SubFactory(SeatTypeFactory)
price = FuzzyDecimal(0.0, 650.0)
currency = factory.Iterator(Currency.objects.all())
upgrade_deadline = FuzzyDateTime(datetime.datetime(2014, 1, 1, tzinfo=UTC))
sku = FuzzyText(length=8)
bulk_sku = FuzzyText(length=8)
course_run = factory.SubFactory(CourseRunFactory)
class Meta:
model = Seat
class OrderFactory(factory.DjangoModelFactory):
class Meta:
model = Order
django_get_or_create = ('uuid', )
uuid = factory.Faker('uuid4')
exchange = factory.SubFactory('apps.order.factories.ExchangeFactory')
account = factory.SubFactory('apps.profile_app.factories.AccountFactory')
type = factory.Iterator(
[constants.ORDER_TYPE_BUY, constants.ORDER_TYPE_SELL])
pair = 'BTC-ETH'
quantity = FuzzyDecimal(0.01, 1.5)
commission = FuzzyDecimal(0.0001, 0.101)
price = FuzzyDecimal(0.025, 2.25)
opened_at = factory.LazyFunction(datetime.datetime.now)
closed_at = factory.LazyFunction(datetime.datetime.now)
data = "{}"