示例#1
0
 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()
示例#2
0
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())))