def enable_ci_for_repo(repo, gh_token):
""" Enable the travis hook for the given repo. """
access_token = travis_utils.get_access_token(gh_token)
repo_id = travis_utils.get_repo_id(repo, access_token)
enabled = travis_utils.enable_hook(repo_id, access_token)
if enabled:
print('Enabled GitHub/Travis hook for %s' % repo)
else:
print('Failed to enable GitHub/Travis hook for %s' % repo)
def get_gh_token(bogus):
""" Returns the GH token to use. """
# fixme: Change this to check if we are rate limited, an use BOGUS if so.
token = os.environ.get('GH_TOKEN', bogus)
return token
BOGUS = 'this-is-a-bogus-token'
THIS_REPO = 'punchagan/statiki'
GH_TOKEN = get_gh_token(BOGUS)
TRAVIS_TOKEN = (
BOGUS if GH_TOKEN == BOGUS else
travis_utils.get_access_token(GH_TOKEN)
)
@unittest.skipIf(TRAVIS_TOKEN == BOGUS, 'Need a real GitHubtoken...')
class TestTravisUtils(unittest.TestCase):
def test_should_find_existing_hook(self):
self.assertTrue(
travis_utils.hook_exists(THIS_REPO, TRAVIS_TOKEN)
)
def test_should_not_find_bogus_hook(self):
self.assertFalse(
travis_utils.hook_exists(THIS_REPO+BOGUS, TRAVIS_TOKEN)
)
