62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
import pytest
|
|
|
|
from httmock import urlmatch, HTTMock
|
|
|
|
from util.config.validators import ConfigValidationException
|
|
from util.config.validators.validate_github import GitHubLoginValidator, GitHubTriggerValidator
|
|
|
|
@pytest.fixture(params=[GitHubLoginValidator, GitHubTriggerValidator])
|
|
def github_validator(request):
|
|
return request.param
|
|
|
|
|
|
@pytest.mark.parametrize('github_config', [
|
|
({}),
|
|
({'GITHUB_ENDPOINT': 'foo'}),
|
|
({'GITHUB_ENDPOINT': 'http://github.com'}),
|
|
({'GITHUB_ENDPOINT': 'http://github.com', 'CLIENT_ID': 'foo'}),
|
|
({'GITHUB_ENDPOINT': 'http://github.com', 'CLIENT_SECRET': 'foo'}),
|
|
({
|
|
'GITHUB_ENDPOINT': 'http://github.com',
|
|
'CLIENT_ID': 'foo',
|
|
'CLIENT_SECRET': 'foo',
|
|
'ORG_RESTRICT': True
|
|
}),
|
|
({
|
|
'GITHUB_ENDPOINT': 'http://github.com',
|
|
'CLIENT_ID': 'foo',
|
|
'CLIENT_SECRET': 'foo',
|
|
'ORG_RESTRICT': True,
|
|
'ALLOWED_ORGANIZATIONS': [],
|
|
}),
|
|
])
|
|
def test_validate_invalid_github_config(github_config, github_validator):
|
|
with pytest.raises(ConfigValidationException):
|
|
unvalidated_config = {}
|
|
unvalidated_config[github_validator.config_key] = github_config
|
|
github_validator.validate(unvalidated_config, None, None)
|
|
|
|
def test_validate_github(github_validator):
|
|
url_hit = [False, False]
|
|
|
|
@urlmatch(netloc=r'somehost')
|
|
def handler(url, request):
|
|
url_hit[0] = True
|
|
return {'status_code': 200, 'content': '', 'headers': {'X-GitHub-Request-Id': 'foo'}}
|
|
|
|
@urlmatch(netloc=r'somehost', path=r'/api/v3/applications/foo/tokens/foo')
|
|
def app_handler(url, request):
|
|
url_hit[1] = True
|
|
return {'status_code': 404, 'content': '', 'headers': {'X-GitHub-Request-Id': 'foo'}}
|
|
|
|
with HTTMock(app_handler, handler):
|
|
github_validator.validate({
|
|
github_validator.config_key: {
|
|
'GITHUB_ENDPOINT': 'http://somehost',
|
|
'CLIENT_ID': 'foo',
|
|
'CLIENT_SECRET': 'bar',
|
|
},
|
|
}, None, None)
|
|
|
|
assert url_hit[0]
|
|
assert url_hit[1]
|