def decorated_view(*args, **kwargs): travis_token = travis_utils.is_travis_user(current_user.github_token) if travis_token is None: return jsonify(dict(message=messages.NO_TRAVIS_ACCOUNT)) else: current_user.set_travis_token(travis_token) return func(*args, **kwargs)
def test_should_confirm_travis_user(self): # Given token_mock = Mock(return_value=TRAVIS_TOKEN) # When with patch('travis_utils.get_access_token', token_mock): token = travis_utils.is_travis_user(GH_TOKEN) # Then # Note: We don't check if token is equal to expected token, # since a failure would mean the token being exposed in the error # output self.assertTrue(isinstance(token, basestring))
def test_should_detect_bogus_user(self): self.assertIsNone(travis_utils.is_travis_user(BOGUS))