# 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 rest_framework import status

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")