from django.shortcuts import reverse
from rest_framework.test import APITestCase
from model_bakery import baker
from glitchtip import test_utils  # pylint: disable=unused-import
from organizations_ext.models import OrganizationUserRole
from ..models import ProjectKey


class ProjectKeyAPITestCase(APITestCase):
    def setUp(self):
        self.user = baker.make("users.user")
        self.client.force_login(self.user)
        self.organization = baker.make("organizations_ext.Organization")
        self.organization.add_user(self.user, role=OrganizationUserRole.OWNER)
        self.project = baker.make("projects.Project", organization=self.organization)
        self.url = reverse(
            "project-keys-list",
            kwargs={"project_pk": self.organization.slug + "/" + self.project.slug},
        )

    def test_key_dsn(self):
        project_key = ProjectKey.objects.get()
        res = self.client.get(self.url)
        self.assertContains(res, project_key.public_key_hex)