# type: ignore

from django.contrib.auth.models import User
from rest_framework.test import APITestCase
from rest_framework_api_key.models import APIKey
from taskbar.models import Workplace

from links.models import Process, Source


class TestCaseBase(APITestCase):

    def setUp(self):
        self.href = "https://example.com"
        self.jwt_user = User.objects.create_superuser("admin",
                                                      "admin@test.com",
                                                      "test123")
        self.process_twisting = Process.objects.create(name="Twisting")
        self.process_crimping = Process.objects.create(name="Crimping")
        self.process_cutting = Process.objects.create(name="Cutting")
        self.source_twisting_1 = Source.objects.create(
            title="Twisting source1",
            process=self.process_twisting.id,
            href=self.href)
        self.source_twisting_2 = Source.objects.create(
            title="Twisting source2",
            process=self.process_twisting.id,
            href=self.href)
        self.source_twisting_3 = Source.objects.create(
            title="Twisting source3",
            process=self.process_twisting.id,
            href=self.href)
        self.source_twisting_4 = Source.objects.create(
            title="Twisting source4",
            process=self.process_twisting.id,
            href=self.href)
        self.source_crimping_1 = Source.objects.create(
            title="Crimping source5",
            process=self.process_crimping.id,
            href=self.href)
        self.source_crimping_2 = Source.objects.create(
            title="Crimping source6",
            process=self.process_crimping.id,
            href=self.href)
        self.source_crimping_3 = Source.objects.create(
            title="Crimping source7",
            process=self.process_crimping.id,
            href=self.href)
        self.source_crimping_4 = Source.objects.create(
            title="Crimping source8",
            process=self.process_crimping.id,
            href=self.href)
        self.Workplace_twisting = Workplace.objects.create(
            hostname="TWIST", process=self.process_twisting.id)
        self.Workplace_crimping = Workplace.objects.create(
            hostname="CRIMP", process=self.process_twisting.id)
        self.Workplace_cuttings = Workplace.objects.create(
            hostname="CUT", process=self.process_twisting.id)

    @property
    def api_key(self):
        api_key, key = APIKey.objects.create_key(name="my-remote-service")
        return {"HTTPs_AUTHORIZATION": f"Api-Key {key}"}