quay/util/config/validators/test/test_validate_gitlab_trigger.py

import json
import pytest

from httmock import urlmatch, HTTMock

from config import build_requests_session
from util.config import URLSchemeAndHostname
from util.config.validator import ValidatorContext
from util.config.validators import ConfigValidationException
from util.config.validators.validate_gitlab_trigger import GitLabTriggerValidator

from test.fixtures import *


@pytest.mark.parametrize(
    "unvalidated_config",
    [
        ({}),
        ({"GITLAB_TRIGGER_CONFIG": {"GITLAB_ENDPOINT": "foo"}}),
        ({"GITLAB_TRIGGER_CONFIG": {"GITLAB_ENDPOINT": "http://someendpoint", "CLIENT_ID": "foo"}}),
        (
            {
                "GITLAB_TRIGGER_CONFIG": {
                    "GITLAB_ENDPOINT": "http://someendpoint",
                    "CLIENT_SECRET": "foo",
                }
            }
        ),
    ],
)
def test_validate_invalid_gitlab_trigger_config(unvalidated_config, app):
    validator = GitLabTriggerValidator()

    with pytest.raises(ConfigValidationException):
        validator.validate(ValidatorContext(unvalidated_config))


def test_validate_gitlab_enterprise_trigger(app):
    url_hit = [False]

    @urlmatch(netloc=r"somegitlab", path="/oauth/token")
    def handler(_, __):
        url_hit[0] = True
        return {"status_code": 400, "content": json.dumps({"error": "invalid code"})}

    with HTTMock(handler):
        validator = GitLabTriggerValidator()

        url_scheme_and_hostname = URLSchemeAndHostname("http", "localhost:5000")

        unvalidated_config = ValidatorContext(
            {
                "GITLAB_TRIGGER_CONFIG": {
                    "GITLAB_ENDPOINT": "http://somegitlab",
                    "CLIENT_ID": "foo",
                    "CLIENT_SECRET": "bar",
                },
            },
            http_client=build_requests_session(),
            url_scheme_and_hostname=url_scheme_and_hostname,
        )

        validator.validate(unvalidated_config)

    assert url_hit[0]