"""
Base tests.
"""
import logging
from django.test import TestCase
from rest_framework.test import APITestCase
from faker import Faker
import pytest
import factories
__author__ = "Artur Barseghyan <artur.barseghyan@gmail.com>"
__copyright__ = "2017-2022 Artur Barseghyan"
__license__ = "GPL-2.0-only OR LGPL-2.1-or-later"
__all__ = (
"BaseRestFrameworkTestCase",
"BaseTestCase",
)
LOGGER = logging.getLogger(__name__)
[docs]@pytest.mark.django_db
class BaseRestFrameworkTestCase(APITestCase):
"""Base REST framework test case."""
pytestmark = pytest.mark.django_db
[docs] @classmethod
def setUpTestData(cls):
"""Set up class."""
# Create user
cls.user = factories.TestUsernameSuperAdminUserFactory()
# Fake data
cls.faker = Faker()
[docs] def authenticate(self):
"""Helper for logging the user in.
:return:
"""
self.client.login(
username=factories.auth_user.TEST_USERNAME,
password=factories.auth_user.TEST_PASSWORD,
)
[docs]@pytest.mark.django_db
class BaseTestCase(TestCase):
"""Base test case."""
pytestmark = pytest.mark.django_db
[docs] @classmethod
def setUpTestData(cls):
"""Set up class."""
# Create user
cls.user = factories.TestUsernameSuperAdminUserFactory()
# Fake data
cls.faker = Faker()