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']
示例#3
0
 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']
示例#5
0
 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']
示例#6
0
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"}
示例#7
0
 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"}
示例#8
0
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!")