30 lines
863 B
Python
30 lines
863 B
Python
"""
|
|
Experimentation factories
|
|
"""
|
|
|
|
|
|
import factory
|
|
import factory.fuzzy
|
|
|
|
from lms.djangoapps.experiments.models import ExperimentData, ExperimentKeyValue
|
|
from common.djangoapps.student.tests.factories import UserFactory
|
|
|
|
|
|
class ExperimentDataFactory(factory.DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
|
class Meta(object):
|
|
model = ExperimentData
|
|
|
|
user = factory.SubFactory(UserFactory)
|
|
experiment_id = factory.fuzzy.FuzzyInteger(0)
|
|
key = factory.Sequence(lambda n: n)
|
|
value = factory.Faker('word')
|
|
|
|
|
|
class ExperimentKeyValueFactory(factory.DjangoModelFactory): # lint-amnesty, pylint: disable=missing-class-docstring
|
|
class Meta(object):
|
|
model = ExperimentKeyValue
|
|
|
|
experiment_id = factory.fuzzy.FuzzyInteger(0)
|
|
key = factory.Sequence(lambda n: n)
|
|
value = factory.Faker('word')
|