from factory.django import DjangoModelFactory, SubFactory

from apps.user.models import *
from apps.course.models import *


class UserFactory(DjangoModelFactory):
    class Meta:
        model = User


class CategoryFactory(DjangoModelFactory):
    class Meta:
        model = Category


class CourseFactory(DjangoModelFactory):
    class Meta:
        model = Course

    author = SubFactory(UserFactory)
    category = SubFactory(CategoryFactory)
    cover = None
    gallery = None