from http import HTTPStatus

from django.contrib.auth.models import Group, User
from django.urls import reverse

from arches.app.models.models import ETLModule, Graph, LoadErrors, LoadEvent
from tests.base_test import ArchesTestCase

# these tests can be run from the command line via
# python manage.py test tests.views.etl_manager_tests --settings="tests.test_settings"


class ETLManagerTests(ArchesTestCase):
    @classmethod
    def setUpTestData(cls):
        super().setUpTestData()
        resource_editor_group = Group.objects.get(name="Resource Editor")
        cls.admin = User.objects.get(username="admin")
        cls.anonymous = User.objects.get(username="anonymous")
        cls.full_name_user = User.objects.create(first_name="Full", last_name="Name")
        cls.full_name_user.groups.add(resource_editor_group)
        cls.csv_module = ETLModule.objects.get(name="Import Single CSV")
        cls.admin_load_event = LoadEvent(user=cls.admin, etl_module=cls.csv_module)
        cls.admin_load_event.save()
        cls.full_name_load_event = LoadEvent(
            user=cls.full_name_user, etl_module=cls.csv_module
        )
        cls.full_name_load_event.save()

    def test_user_model_not_serialized_on_load_event(self):
        self.client.force_login(self.admin)
        response = self.client.get(
            reverse("etl_manager"), QUERY_STRING="action=loadEvent"
        )
        self.assertNotContains(response, "password")
        self.assertContains(response, '"user_displayname": "admin"')

    def test_user_displayname_formatting(self):
        self.client.force_login(self.full_name_user)
        response = self.client.get(
            reverse("etl_manager"), QUERY_STRING="action=loadEvent"
        )
        self.assertContains(response, '"user_displayname": "Full Name"')

    def test_node_error_view(self):
        error = LoadErrors.objects.create(
            load_event=self.full_name_load_event, message="Early failure."
        )

        self.client.force_login(self.full_name_user)
        response = self.client.get(
            reverse("etl_manager"),
            QUERY_STRING=f"action=nodeError&loadid={self.full_name_load_event.pk}&nodeid=null&error=null",
        )
        self.assertContains(response, error.message)

        error.error = "ValueError"
        error.save()

        response = self.client.get(
            reverse("etl_manager"),
            QUERY_STRING=f"action=nodeError&loadid={self.full_name_load_event.pk}&nodeid=null&error={error.error}",
        )
        self.assertContains(response, error.error)

        graph = Graph.objects.create_graph(name="Test graph")
        error.node = graph.node_set.first()
        error.save()

        response = self.client.get(
            reverse("etl_manager"),
            QUERY_STRING=f"action=nodeError&loadid={self.full_name_load_event.pk}&nodeid={error.node_id}&error={error.error}",
        )
        self.assertContains(response, error.node_id)

    def test_no_loadid(self):
        self.client.force_login(self.admin)
        with self.assertLogs("django.request", level="WARNING"):
            response = self.client.get(
                reverse("etl_manager"), QUERY_STRING="action=nodeError"
            )
        self.assertEqual(response.status_code, HTTPStatus.BAD_REQUEST)

    def test_insufficent_permissions(self):
        self.client.force_login(self.anonymous)
        with self.assertLogs("django.request", level="WARNING"):
            response = self.client.get(reverse("etl_manager"))
        self.assertEqual(response.status_code, HTTPStatus.FORBIDDEN)