def make_task_graph(*args, **kwargs): args = list(args) mocked_get_json_rev = args.pop() mocked_get_json_rev.return_value = {"pushid": 78123} return make_task_graph_orig(*args, public_key=DUMMY_PUBLIC_KEY, balrog_username="******", balrog_password="******", beetmover_aws_access_key_id="baz", beetmover_aws_secret_access_key="norf", running_tests=True, **kwargs)
def make_task_graph(*args, **kwargs): args = list(args) mocked_get_resultsets = args.pop() mocked_get_resultsets.return_value = [{"revision_hash": "abcdefgh1234567"}] return make_task_graph_orig(*args, public_key=DUMMY_PUBLIC_KEY, balrog_username="******", balrog_password="******", beetmover_aws_access_key_id="baz", beetmover_aws_secret_access_key="norf", running_tests=True, **kwargs)