#
# Copyright (C) 2019-2022 Mathieu Parent <math.parent@gmail.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser 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 Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from collections.abc import Generator
from contextlib import contextmanager
from tempfile import NamedTemporaryFile
from unittest.mock import call, patch

from gitlab import Gitlab

from gitlabracadabra.gitlab.connections import GitlabConnections
from gitlabracadabra.tests.case import TestCase


class TestGitlabConnections(TestCase):
    """Test GitlabConnections class."""

    @contextmanager
    def temp_config(
        self,
        call_count: int = 1,
    ) -> Generator[GitlabConnections, None, None]:
        """Fake configuration.

        Args:
            call_count: Number of Gitlab objects created.

        Yields:
            The Gitlab connections singleton.
        """
        config = """
            [global]
            default = gitlab

            [gitlab]
            url = https://gitlab.com
            private_token = T0k3n

            [internal]
            url = https://gitlab.example.com
            private_token = n3k0T
        """
        with NamedTemporaryFile(mode="w") as tmp:
            tmp.write(config)
            tmp.flush()
            singleton = GitlabConnections()
            singleton.load(None, [tmp.name], debug=False)
            with patch.object(Gitlab, "auth") as auth_mock:
                yield singleton
                assert auth_mock.mock_calls == [call() for _ in range(call_count)]

    def test_singleton(self) -> None:
        """Ensure singleton pattern."""
        singleton1 = GitlabConnections()
        singleton2 = GitlabConnections()
        assert id(singleton1) == id(singleton2)

    def test_get_connection_none(self) -> None:
        """Get default Gitlab connection."""
        with self.temp_config() as singleton:
            gl1 = singleton.get_connection()
            assert gl1.pygitlab.api_url == "https://gitlab.com/api/v4"
            assert gl1.pygitlab.private_token == "T0k3n"
            gl2 = singleton.get_connection(None)
            assert id(gl1) == id(gl2)

    def test_get_connection_internal(self) -> None:
        """Get another Gitlab connection."""
        with self.temp_config() as singleton:
            gl1 = singleton.get_connection("internal")
            assert gl1.pygitlab.api_url == "https://gitlab.example.com/api/v4"
            assert gl1.pygitlab.private_token == "n3k0T"
            gl2 = singleton.get_connection("internal")
            assert id(gl1) == id(gl2)

    def test_get_connection_both(self) -> None:
        """Get several Gitlab connections."""
        with self.temp_config(2) as singleton:
            gl1 = singleton.get_connection()
            gl2 = singleton.get_connection("internal")
            assert gl1.pygitlab.api_url == "https://gitlab.com/api/v4"
            assert gl1.pygitlab.private_token == "T0k3n"
            assert gl2.pygitlab.api_url == "https://gitlab.example.com/api/v4"
            assert gl2.pygitlab.private_token == "n3k0T"
