#
# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
#

import logging

import pytest
from source_gitlab import SourceGitlab
from source_gitlab.streams import GitlabStream


def test_streams(config, requests_mock):
    requests_mock.get("/api/v4/groups", json=[{"id": "g1"}, {"id": "g256"}])
    source = SourceGitlab()
    streams = source.streams(config)
    assert len(streams) == 22
    assert all([isinstance(stream, GitlabStream) for stream in streams])
    groups, projects, *_ = streams
    assert groups.group_ids == ["g1", "g256"]
    assert projects.project_ids == []


@pytest.mark.parametrize(
    "url_mocks",
    (
        (
            {"url": "/api/v4/groups", "json": [{"id": "g1"}]},
            {"url": "/api/v4/groups/g1", "json": [{"id": "g1", "projects": [{"id": "p1", "path_with_namespace": "p1"}]}]},
            {"url": "/api/v4/projects/p1", "json": {"id": "p1"}}
        ),
        (
            {"url": "/api/v4/groups", "json": []},
        ),
    )
)
def test_connection_success(config, requests_mock, url_mocks):
    for url_mock in url_mocks:
        requests_mock.get(**url_mock)
    source = SourceGitlab()
    status, msg = source.check_connection(logging.getLogger(), config)
    assert (status, msg) == (True, None)


def test_connection_fail_due_to_api_error(config, mocker, requests_mock):
    mocker.patch("time.sleep")
    requests_mock.get("/api/v4/groups", status_code=500)
    source = SourceGitlab()
    status, msg = source.check_connection(logging.getLogger(), config)
    assert status is False, msg.startswith('Unable to connect to Gitlab API with the provided credentials - "DefaultBackoffException"')


@pytest.mark.parametrize(
    "api_url, deployment_env, expected_message",
    (
        ("http://gitlab.my.company.org", "CLOUD", "Http scheme is not allowed in this environment. Please use `https` instead."),
        ("https://gitlab.com/api/v4", "CLOUD", "Invalid API resource locator.")
    )
)
def test_connection_fail_due_to_config_error(mocker, api_url, deployment_env, expected_message):
    mocker.patch("os.environ", {"DEPLOYMENT_MODE": deployment_env})
    source = SourceGitlab()
    config = {
        "start_date": "2021-01-01T00:00:00Z",
        "api_url": api_url,
        "credentials": {
            "auth_type": "access_token",
            "access_token": "token"
        }
    }
    status, msg = source.check_connection(logging.getLogger(), config)
    assert (status, msg) == (False, expected_message)