ba773ce4ce
* Connector health: source hubspot, gitlab, snapchat-marketing: fix builds * #26254 source Gitlab: disallow enpoints in URLs, disallow bare http in cloud * source gitlab: update changelog * #26254 source gitlab: rm http from spec
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
#
|
|
# 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)
|