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

import pytest

from httmock import urlmatch, HTTMock

from config import build_requests_session
from util.config.validator import ValidatorContext
from util.config.validators import ConfigValidationException
from util.config.validators.validate_github import GitHubLoginValidator, GitHubTriggerValidator

from test.fixtures import *

@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, app):
  with pytest.raises(ConfigValidationException):
    unvalidated_config = {}
    unvalidated_config[github_validator.config_key] = github_config
    github_validator.validate(ValidatorContext(unvalidated_config))

def test_validate_github(github_validator, app):
  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):
    unvalidated_config = ValidatorContext({
      github_validator.config_key: {
        'GITHUB_ENDPOINT': 'http://somehost',
        'CLIENT_ID': 'foo',
        'CLIENT_SECRET': 'bar',
      },
    })

    unvalidated_config.http_client = build_requests_session()
    github_validator.validate(unvalidated_config)

  assert url_hit[0]
  assert url_hit[1]