def test_bad_jwt_key(self): fname = pkg_resources.resource_filename(__name__, 'jwt-test-user.json') with open(fname) as f: userdata = json.loads(f.read()) token = hj.JWTClient(hj.JWTClient.generate_key()).encode(userdata) session = aiohttp.ClientSession( raise_for_status=True, timeout=aiohttp.ClientTimeout(total=60)) bc = BatchClient(session, url=os.environ.get('BATCH_URL'), token=token) try: b = bc.create_batch() j = b.create_job('alpine', ['false']) b.submit() assert False, j except aiohttp.ClientResponseError as e: if e.status == 401: pass else: assert False, e finally: bc.close()
import json import sys import hailtop.gear.auth as hj with open(sys.argv[1], 'rb') as f: c = hj.JWTClient(f.read()) sys.stdout.write(c.encode(json.loads(sys.stdin.read())))