import factory

import random
from django.conf import settings
from TWLight.resources.models import Partner, Stream, Video, Suggestion


class PartnerFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Partner
        strategy = factory.CREATE_STRATEGY

    company_name = factory.Faker(
        "company", locale=random.choice(settings.FAKER_LOCALES)
    )
    terms_of_use = factory.Faker("uri", locale=random.choice(settings.FAKER_LOCALES))
    status = Partner.AVAILABLE  # not the default, but usually wanted in tests


class StreamFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Stream
        strategy = factory.CREATE_STRATEGY

    partner = factory.SubFactory(PartnerFactory)
    name = factory.Faker("bs", locale=random.choice(settings.FAKER_LOCALES))


class SuggestionFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Suggestion
        strategy = factory.CREATE_STRATEGY

    suggested_company_name = factory.Faker(
        "company", locale=random.choice(settings.FAKER_LOCALES)
    )
    company_url = factory.Faker("url", locale=random.choice(settings.FAKER_LOCALES))


class VideoFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Video
        strategy = factory.CREATE_STRATEGY

    partner = factory.SubFactory(PartnerFactory)

import factory

import random
from django.conf import settings
from TWLight.resources.models import Partner, Stream, Video, Suggestion


class PartnerFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Partner
        strategy = factory.CREATE_STRATEGY

    company_name = factory.Faker(
        "company", locale=random.choice(settings.FAKER_LOCALES)
    )
    terms_of_use = factory.Faker("uri", locale=random.choice(settings.FAKER_LOCALES))
    status = Partner.AVAILABLE  # not the default, but usually wanted in tests


class StreamFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Stream
        strategy = factory.CREATE_STRATEGY

    partner = factory.SubFactory(PartnerFactory)
    name = factory.Faker("bs", locale=random.choice(settings.FAKER_LOCALES))


class SuggestionFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Suggestion
        strategy = factory.CREATE_STRATEGY

    suggested_company_name = factory.Faker("pystr", max_chars=40)
    company_url = factory.Faker("url", locale=random.choice(settings.FAKER_LOCALES))


class VideoFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Video
        strategy = factory.CREATE_STRATEGY

    partner = factory.SubFactory(PartnerFactory)

Change suggested_company_name factory var to pystr
Change suggested_company_name factory var to pystr

  import factory
  
  import random
  from django.conf import settings
  from TWLight.resources.models import Partner, Stream, Video, Suggestion
  
  
  class PartnerFactory(factory.django.DjangoModelFactory):
      class Meta:
          model = Partner
          strategy = factory.CREATE_STRATEGY
  
      company_name = factory.Faker(
          "company", locale=random.choice(settings.FAKER_LOCALES)
      )
      terms_of_use = factory.Faker("uri", locale=random.choice(settings.FAKER_LOCALES))
      status = Partner.AVAILABLE  # not the default, but usually wanted in tests
  
  
  class StreamFactory(factory.django.DjangoModelFactory):
      class Meta:
          model = Stream
          strategy = factory.CREATE_STRATEGY
  
      partner = factory.SubFactory(PartnerFactory)
      name = factory.Faker("bs", locale=random.choice(settings.FAKER_LOCALES))
  
  
  class SuggestionFactory(factory.django.DjangoModelFactory):
      class Meta:
          model = Suggestion
          strategy = factory.CREATE_STRATEGY
  
-     suggested_company_name = factory.Faker(
+     suggested_company_name = factory.Faker("pystr", max_chars=40)
-         "company", locale=random.choice(settings.FAKER_LOCALES)
-     )
      company_url = factory.Faker("url", locale=random.choice(settings.FAKER_LOCALES))
  
  
  class VideoFactory(factory.django.DjangoModelFactory):
      class Meta:
          model = Video
          strategy = factory.CREATE_STRATEGY
  
      partner = factory.SubFactory(PartnerFactory)
  
Change suggested_company_name factory var to pystr
## Code Before:
import factory

import random
from django.conf import settings
from TWLight.resources.models import Partner, Stream, Video, Suggestion


class PartnerFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Partner
        strategy = factory.CREATE_STRATEGY

    company_name = factory.Faker(
        "company", locale=random.choice(settings.FAKER_LOCALES)
    )
    terms_of_use = factory.Faker("uri", locale=random.choice(settings.FAKER_LOCALES))
    status = Partner.AVAILABLE  # not the default, but usually wanted in tests


class StreamFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Stream
        strategy = factory.CREATE_STRATEGY

    partner = factory.SubFactory(PartnerFactory)
    name = factory.Faker("bs", locale=random.choice(settings.FAKER_LOCALES))


class SuggestionFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Suggestion
        strategy = factory.CREATE_STRATEGY

    suggested_company_name = factory.Faker(
        "company", locale=random.choice(settings.FAKER_LOCALES)
    )
    company_url = factory.Faker("url", locale=random.choice(settings.FAKER_LOCALES))


class VideoFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Video
        strategy = factory.CREATE_STRATEGY

    partner = factory.SubFactory(PartnerFactory)

## Instruction:
Change suggested_company_name factory var to pystr
## Code After:
import factory

import random
from django.conf import settings
from TWLight.resources.models import Partner, Stream, Video, Suggestion


class PartnerFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Partner
        strategy = factory.CREATE_STRATEGY

    company_name = factory.Faker(
        "company", locale=random.choice(settings.FAKER_LOCALES)
    )
    terms_of_use = factory.Faker("uri", locale=random.choice(settings.FAKER_LOCALES))
    status = Partner.AVAILABLE  # not the default, but usually wanted in tests


class StreamFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Stream
        strategy = factory.CREATE_STRATEGY

    partner = factory.SubFactory(PartnerFactory)
    name = factory.Faker("bs", locale=random.choice(settings.FAKER_LOCALES))


class SuggestionFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Suggestion
        strategy = factory.CREATE_STRATEGY

    suggested_company_name = factory.Faker("pystr", max_chars=40)
    company_url = factory.Faker("url", locale=random.choice(settings.FAKER_LOCALES))


class VideoFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Video
        strategy = factory.CREATE_STRATEGY

    partner = factory.SubFactory(PartnerFactory)
