# Copyright (C) 2017-2022 The Sipwise Team - http://sipwise.com
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along
# with this program.  If not, see <http://www.gnu.org/licenses/>.
from django.test import override_settings
from django.urls import reverse
from django.urls.exceptions import NoReverseMatch
from rest_framework import status
from rest_framework.test import APISimpleTestCase

from .test_models import set_build_done
from build import models
from repoapi.test.base import APIAuthenticatedTestCase


@override_settings(JBI_ALLOWED_HOSTS=["fake.local"])
class TestRest(APIAuthenticatedTestCase):
    def setUp(self):
        super(TestRest, self).setUp()
        self.url = reverse("build:list")

    def test_wrong_release(self):
        data = {
            "uuid": "fake_uuid",
            "release": "wrong-release",
        }
        response = self.client.post(self.url, data, format="json")
        self.assertNotEqual(response.status_code, status.HTTP_201_CREATED)

    def test_trunk_wrong_debian_release(self):
        data = {
            "uuid": "fake_uuid",
            "release": "release-trunk-stretch",
        }
        response = self.client.post(self.url, data, format="json")
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertEqual(response.data["release"], "trunk")
        self.assertIsNone(response.data["tag"])
        self.assertEqual(response.data["branch"], "master")
        self.assertEqual(response.data["distribution"], "buster")

    def test_trunk(self):
        data = {
            "uuid": "fake_uuid",
            "release": "trunk",
        }
        response = self.client.post(self.url, data, format="json")
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertEqual(response.data["uuid"], data["uuid"])
        self.assertEqual(response.data["release"], "trunk")
        self.assertIsNone(response.data["tag"])
        self.assertEqual(response.data["branch"], "master")
        self.assertEqual(response.data["distribution"], "buster")
        projects = response.data["projects"].split(",")
        self.assertEqual(len(projects), 73)

    def test_trunk_weekly(self):
        data = {
            "uuid": "fake_uuid",
            "release": "trunk-weekly",
        }
        response = self.client.post(self.url, data, format="json")
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertEqual(response.data["uuid"], data["uuid"])
        self.assertEqual(response.data["release"], "release-trunk-weekly")
        self.assertIsNone(response.data["tag"])
        self.assertEqual(response.data["branch"], "master")
        self.assertEqual(response.data["distribution"], "bullseye")
        projects = response.data["projects"].split(",")
        self.assertEqual(len(projects), 72)

    def test_mrXX(self):
        data = {
            "uuid": "fake_uuid",
            "release": "mr8.1",
        }
        response = self.client.post(self.url, data, format="json")
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertEqual(response.data["uuid"], data["uuid"])
        self.assertEqual(response.data["release"], "release-mr8.1")
        self.assertIsNone(response.data["tag"])
        self.assertEqual(response.data["branch"], "mr8.1")
        projects = response.data["projects"].split(",")
        self.assertEqual(len(projects), 73)

    def test_mrXXX(self):
        data = {
            "uuid": "fake_uuid",
            "release": "mr7.5.2",
        }
        response = self.client.post(self.url, data, format="json")
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        self.assertEqual(response.data["uuid"], data["uuid"])
        self.assertEqual(response.data["release"], "release-mr7.5.2")
        self.assertEqual(response.data["tag"], "mr7.5.2.1")
        self.assertEqual(response.data["branch"], "mr7.5.2")
        projects = response.data["projects"].split(",")
        self.assertEqual(len(projects), 75)


@override_settings(JBI_ALLOWED_HOSTS=["fake.local"])
class TestBuildRest(APIAuthenticatedTestCase):
    fixtures = [
        "test_models",
    ]
    release = "release-mr8.1"
    release_uuid = "dbe569f7-eab6-4532-a6d1-d31fb559649b"

    def test_build_project(self):
        data = {}
        url = reverse(
            "build:build_project", args=[self.release_uuid, "kamailio"]
        )
        response = self.client.post(url, data, format="json")
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)


@override_settings(JBI_ALLOWED_HOSTS=["fake.local"])
class TestBuildDeleteRest(APIAuthenticatedTestCase):
    fixtures = [
        "test_models",
        "test_models_jbi",
    ]
    release = "release-mr8.1"
    release_uuid = "dbe569f7-eab6-4532-a6d1-d31fb559649b"

    def test_get_br(self):
        br = models.BuildRelease.objects.get(uuid=self.release_uuid)
        data = {}
        url = reverse("build:detail", args=[br.id])
        response = self.client.get(url, data, format="json")
        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_delete_br(self):
        br = models.BuildRelease.objects.get(uuid=self.release_uuid)
        self.assertEqual(
            models.JenkinsBuildInfo.objects.filter(
                param_release_uuid=self.release_uuid
            ).count(),
            4,
        )
        url = reverse("build:detail", args=[br.id])
        response = self.client.delete(url, format="json")
        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
        with self.assertRaises(models.BuildRelease.DoesNotExist):
            models.BuildRelease.objects.get(uuid=self.release_uuid)
        self.assertEqual(
            models.JenkinsBuildInfo.objects.filter(
                param_release_uuid=self.release_uuid
            ).count(),
            0,
        )

    def test_delete_br_empty(self):
        br = models.BuildRelease.objects.get(uuid=self.release_uuid)
        self.assertEqual(
            models.JenkinsBuildInfo.objects.filter(
                param_release_uuid=self.release_uuid
            ).count(),
            4,
        )
        models.BuildRelease.objects.jbi(self.release_uuid).delete()
        self.assertEqual(
            models.JenkinsBuildInfo.objects.filter(
                param_release_uuid=self.release_uuid
            ).count(),
            0,
        )
        url = reverse("build:detail", args=[br.id])
        response = self.client.delete(url, format="json")
        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
        with self.assertRaises(models.BuildRelease.DoesNotExist):
            models.BuildRelease.objects.get(uuid=self.release_uuid)
        self.assertEqual(
            models.JenkinsBuildInfo.objects.filter(
                param_release_uuid=self.release_uuid
            ).count(),
            0,
        )


@override_settings(JBI_ALLOWED_HOSTS=["fake.local"])
class TestBuildPatchRest(APIAuthenticatedTestCase):
    fixtures = [
        "test_models",
    ]
    release = "release-mr8.1"
    release_uuid = "dbe569f7-eab6-4532-a6d1-d31fb559649b"

    def test_refresh(self):
        br = models.BuildRelease.objects.get(uuid=self.release_uuid)
        self.assertEqual(br.projects, "kamailio,lua-ngcp-kamailio,ngcp-panel")
        data = {"action": "refresh"}
        url = reverse("build:detail", args=[br.id])
        response = self.client.patch(url, data, format="json")
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        br = models.BuildRelease.objects.get(uuid=self.release_uuid)
        self.assertNotEqual(
            br.projects, "kamailio,lua-ngcp-kamailio,ngcp-panel"
        )
        self.assertEqual(len(br.projects_list), 73)

    def test_no_action(self):
        br = models.BuildRelease.objects.get(uuid=self.release_uuid)
        self.assertEqual(br.projects, "kamailio,lua-ngcp-kamailio,ngcp-panel")
        data = {}
        url = reverse("build:detail", args=[br.id])
        response = self.client.patch(url, data, format="json")
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
        br = models.BuildRelease.objects.get(uuid=self.release_uuid)
        self.assertEqual(br.projects, "kamailio,lua-ngcp-kamailio,ngcp-panel")


class TestCheckConfig(APISimpleTestCase):
    url = reverse("check-config")

    def setUp(self):
        self.data = {
            "jenkins-jobs": {},
            "distris": ["release-trunk-buster"],
            "release-trunk-buster": [],
        }

    def test_empty(self):
        response = self.client.post(self.url, {}, format="json")
        self.assertEqual(response.status_code, status.HTTP_406_NOT_ACCEPTABLE)

    def test_clean(self):
        response = self.client.post(self.url, self.data, format="json")
        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_circular_deps(self):
        self.data["jenkins-jobs"]["build_deps"] = {"A": ["A1", "A"]}
        response = self.client.post(self.url, self.data, format="json")
        self.assertEqual(response.status_code, status.HTTP_406_NOT_ACCEPTABLE)


@override_settings(JBI_ALLOWED_HOSTS=["fake.local"])
class TestTrunkCleanup(APIAuthenticatedTestCase):
    fixtures = ["test_weekly"]

    def test_mrXX(self):
        with self.assertRaises(NoReverseMatch):
            reverse("build:trunk-cleanup", args=["mr10.1"])

    def test_none(self):
        url = reverse("build:trunk-cleanup", args=["release-trunk-wheezy"])
        response = self.client.delete(url, {}, format="json")
        self.assertEqual(response.status_code, status.HTTP_200_OK)

    def test_weekly_not_done(self):
        qs = models.BuildRelease.objects.filter(release="release-trunk-weekly")
        self.assertEqual(qs.count(), 1)
        self.assertFalse(qs.first().done)
        url = reverse("build:trunk-cleanup", args=["release-trunk-weekly"])
        response = self.client.delete(url, {}, format="json")
        self.assertEqual(response.status_code, status.HTTP_202_ACCEPTED)
        self.assertEqual(qs.count(), 0)

    def test_weekly_not_done_with_history(self):
        params = {
            "uuid": "dbe569f7-eab6-4532-a6d1-d31fb559649c",
            "branch": "master",
            "release": "release-trunk-weekly",
            "distribution": "bullseye",
            "projects": "asterisk-voicemail,backup-tools",
            "built_projects": "release-copy-debs-yml",
        }
        set_build_done(
            models.BuildRelease.objects.filter(release="release-trunk-weekly")
        )
        models.BuildRelease.objects.create(**params)
        qs = models.BuildRelease.objects.filter(release="release-trunk-weekly")
        self.assertEqual(qs.count(), 2)
        url = reverse("build:trunk-cleanup", args=["release-trunk-weekly"])
        response = self.client.delete(url, {}, format="json")
        self.assertEqual(response.status_code, status.HTTP_202_ACCEPTED)
        self.assertEqual(qs.count(), 1)
        self.assertEqual(
            qs.first().uuid, "dbe569f7-eab6-4532-a6d1-d31fb559649b"
        )

    def test_weekly_done(self):
        qs = models.BuildRelease.objects.filter(release="release-trunk-weekly")
        self.assertEqual(qs.count(), 1)
        set_build_done(qs)
        self.assertTrue(qs.first().done)
        url = reverse("build:trunk-cleanup", args=["release-trunk-weekly"])
        response = self.client.delete(url, {}, format="json")
        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertEqual(qs.count(), 1)