def get_temp_creds(context): creds = read_integration_creds() if "certificate" in creds: return temp_creds = utils.create_temp_creds( creds["clientId"], creds["accessToken"], expires=arrow.utcnow().shift(minutes=10).datetime, scopes=[ "queue:cancel-task:test-dummy-scheduler/*", "queue:claim-work:test-dummy-provisioner/dummy-worker-*", "queue:create-task:lowest:test-dummy-provisioner/dummy-worker-*", "queue:scheduler-id:test-dummy-scheduler", "queue:define-task:test-dummy-provisioner/dummy-worker-*", "queue:get-artifact:SampleArtifacts/_/X.txt", "queue:schedule-task:test-dummy-scheduler/*", "queue:task-group-id:test-dummy-scheduler/*", "queue:worker-id:test-dummy-workers/dummy-worker-*", ], ) if temp_creds: context.credentials = temp_creds print("Using temp creds!") else: raise Exception("Can't get temp_creds!")
async def test_temp_creds(context_function): async with context_function(None) as context: async with remember_cwd(): os.chdir(os.path.dirname(context.config['work_dir'])) context.temp_credentials = utils.create_temp_creds( context.credentials['clientId'], context.credentials['accessToken'], expires=arrow.utcnow().replace(minutes=10).datetime) result = await context.temp_queue.ping() assert result['alive']
def test_temp_creds(self, event_loop, context_function): with context_function(None) as context: with remember_cwd(): os.chdir("integration") context.temp_credentials = utils.create_temp_creds( context.credentials['clientId'], context.credentials['accessToken'], expires=arrow.utcnow().replace(minutes=10).datetime ) result = event_loop.run_until_complete(context.temp_queue.ping()) assert result['alive']
async def test_temp_creds(context_function): async with context_function(None) as context: async with remember_cwd(): os.chdir(os.path.dirname(context.config['work_dir'])) context.temp_credentials = utils.create_temp_creds( context.credentials['clientId'], context.credentials['accessToken'], expires=arrow.utcnow().replace(minutes=10).datetime ) result = await context.temp_queue.ping() assert result['alive']
def test_temp_creds(self, event_loop, context_function): with context_function(None) as context: with remember_cwd(): os.chdir("integration") context.temp_credentials = utils.create_temp_creds( context.credentials['clientId'], context.credentials['accessToken'], expires=arrow.utcnow().replace(minutes=10).datetime) result = event_loop.run_until_complete( context.temp_queue.ping()) assert result['alive']
def test_create_temp_creds(): with mock.patch.object(utils, "createTemporaryCredentials") as p: p.return_value = {"one": b"one", "two": "two"} creds = utils.create_temp_creds("clientId", "accessToken", "start", "expires") assert p.called_once_with( "clientId", "accessToken", "start", "expires", ["assume:project:taskcluster:worker-test-scopes"], name=None) assert creds == {"one": "one", "two": "two"}
def test_temp_creds(self): with mock.patch.object(utils, 'createTemporaryCredentials') as p: p.return_value = { "one": b"one", "two": "two", } creds = utils.create_temp_creds('clientId', 'accessToken', 'start', 'expires') assert p.called_once_with( 'clientId', 'accessToken', 'start', 'expires', ['assume:project:taskcluster:worker-test-scopes', ], name=None ) assert creds == {"one": "one", "two": "two"}
def test_create_temp_creds(): with mock.patch.object(utils, 'createTemporaryCredentials') as p: p.return_value = { "one": b"one", "two": "two", } creds = utils.create_temp_creds('clientId', 'accessToken', 'start', 'expires') assert p.called_once_with( 'clientId', 'accessToken', 'start', 'expires', ['assume:project:taskcluster:worker-test-scopes', ], name=None ) assert creds == {"one": "one", "two": "two"}
def get_temp_creds(context): if 'certificate' in context.credentials: return temp_creds = utils.create_temp_creds( context.credentials['clientId'], context.credentials['accessToken'], expires=arrow.utcnow().replace(minutes=10).datetime, ) if temp_creds: context.credentials = temp_creds print("Using temp creds!") else: raise Exception("Can't get temp_creds!")
def get_temp_creds(context): if 'certificate' in context.credentials: return temp_creds = utils.create_temp_creds( context.credentials['clientId'], context.credentials['accessToken'], expires=arrow.utcnow().replace(minutes=10).datetime, ) if temp_creds: context.credentials = temp_creds print("Using temp creds!") else: raise Exception("Can't get temp_creds!")