aboutsummaryrefslogtreecommitdiffhomepage
path: root/tools/run_tests
diff options
context:
space:
mode:
authorGravatar Lidi Zheng <lidiz@google.com>2018-12-14 10:08:22 -0800
committerGravatar Lidi Zheng <lidiz@google.com>2018-12-14 10:08:22 -0800
commit5a38d1956fb7007108a310e0280df58c9c824cec (patch)
tree43bf0f624a45172e05fa54c7d5bafd20cf1fdf7f /tools/run_tests
parent0f0822d53f37ea9da8bf838e5b3c62b05afe2f10 (diff)
Make yapf happy
Diffstat (limited to 'tools/run_tests')
-rw-r--r--tools/run_tests/python_utils/check_on_pr.py78
1 files changed, 49 insertions, 29 deletions
diff --git a/tools/run_tests/python_utils/check_on_pr.py b/tools/run_tests/python_utils/check_on_pr.py
index 96418bf2d3..935d59713e 100644
--- a/tools/run_tests/python_utils/check_on_pr.py
+++ b/tools/run_tests/python_utils/check_on_pr.py
@@ -25,47 +25,57 @@ _GITHUB_API_PREFIX = 'https://api.github.com'
_GITHUB_REPO = 'lidizheng/grpc'
_GITHUB_APP_ID = 22288
_INSTALLATION_ID = 516307
-_GITHUB_APP_KEY = open(os.environ['HOME'] + '/.ssh/google-grpc-checker.2018-12-13.private-key.pem', 'r').read()
+_GITHUB_APP_KEY = open(
+ os.environ['HOME'] + '/.ssh/google-grpc-checker.2018-12-13.private-key.pem',
+ 'r').read()
_ACCESS_TOKEN_CACHE = None
+
def _jwt_token():
- return jwt.encode({
- 'iat': int(time.time()),
- 'exp': int(time.time() + 60 * 10), # expire in 10 minutes
- 'iss': _GITHUB_APP_ID,
- }, _GITHUB_APP_KEY, algorithm='RS256')
+ return jwt.encode(
+ {
+ 'iat': int(time.time()),
+ 'exp': int(time.time() + 60 * 10), # expire in 10 minutes
+ 'iss': _GITHUB_APP_ID,
+ },
+ _GITHUB_APP_KEY,
+ algorithm='RS256')
+
def _access_token():
global _ACCESS_TOKEN_CACHE
if _ACCESS_TOKEN_CACHE == None or _ACCESS_TOKEN_CACHE['exp'] < time.time():
resp = requests.post(
- url='https://api.github.com/app/installations/%s/access_tokens' % _INSTALLATION_ID,
+ url='https://api.github.com/app/installations/%s/access_tokens' %
+ _INSTALLATION_ID,
headers={
'Authorization': 'Bearer %s' % _jwt_token().decode('ASCII'),
'Accept': 'application/vnd.github.machine-man-preview+json',
- }
- )
- _ACCESS_TOKEN_CACHE = {'token': resp.json()['token'], 'exp': time.time()+60}
+ })
+ _ACCESS_TOKEN_CACHE = {
+ 'token': resp.json()['token'],
+ 'exp': time.time() + 60
+ }
return _ACCESS_TOKEN_CACHE['token']
+
def _call(url, method='GET', json=None):
if not url.startswith('https://'):
url = _GITHUB_API_PREFIX + url
- headers={
+ headers = {
'Authorization': 'Bearer %s' % _access_token(),
'Accept': 'application/vnd.github.antiope-preview+json',
}
- return requests.request(
- method=method,
- url=url,
- headers=headers,
- json=json)
+ return requests.request(method=method, url=url, headers=headers, json=json)
+
def _latest_commit():
- resp = _call('/repos/%s/pulls/%s/commits' % (_GITHUB_REPO, os.environ['ghprbPullId']))
+ resp = _call('/repos/%s/pulls/%s/commits' % (_GITHUB_REPO,
+ os.environ['ghprbPullId']))
return resp.json()[-1]
+
def check_on_pr(name, summary, success=True):
"""Create/Update a check on current pull request.
@@ -84,15 +94,25 @@ def check_on_pr(name, summary, success=True):
print('Missing ghprbPullId env var: not commenting')
return
commit = _latest_commit()
- resp = _call('/repos/%s/check-runs' % _GITHUB_REPO, method='POST', json={
- 'name': name,
- 'head_sha': commit['sha'],
- 'status': 'completed',
- 'completed_at': '%sZ' % datetime.datetime.utcnow().replace(microsecond=0).isoformat(),
- 'conclusion': 'success' if success else 'failure',
- 'output': {
- 'title': name,
- 'summary': summary,
- }
- })
- print('Result of Creating/Updating Check on PR:', json.dumps(resp.json(), indent=2))
+ resp = _call(
+ '/repos/%s/check-runs' % _GITHUB_REPO,
+ method='POST',
+ json={
+ 'name':
+ name,
+ 'head_sha':
+ commit['sha'],
+ 'status':
+ 'completed',
+ 'completed_at':
+ '%sZ' %
+ datetime.datetime.utcnow().replace(microsecond=0).isoformat(),
+ 'conclusion':
+ 'success' if success else 'failure',
+ 'output': {
+ 'title': name,
+ 'summary': summary,
+ }
+ })
+ print('Result of Creating/Updating Check on PR:',
+ json.dumps(resp.json(), indent=2))