diff --git a/src/integration-tests/integration-test.py b/src/integration-tests/integration-test.py index 361512d0f..95e916b7b 100755 --- a/src/integration-tests/integration-test.py +++ b/src/integration-tests/integration-test.py @@ -25,15 +25,14 @@ from enum import Enum from typing import Dict, List, Optional, Set, Tuple from uuid import UUID, uuid4 +from onefuzz.api import Command, Onefuzz +from onefuzz.backend import ContainerWrapper, wait +from onefuzz.cli import execute_api from onefuzztypes.enums import OS, ContainerType, TaskState, VmState from onefuzztypes.models import Job, Pool, Repro, Scaleset from onefuzztypes.primitives import Directory, File from pydantic import BaseModel, Field -from onefuzz.api import Command, Onefuzz -from onefuzz.backend import ContainerWrapper, wait -from onefuzz.cli import execute_api - LINUX_POOL = "linux-test" WINDOWS_POOL = "linux-test" BUILD = "0" diff --git a/src/utils/ado-testing/check-ado.py b/src/utils/ado-testing/check-ado.py index 7f5b1249b..4f3275ecd 100644 --- a/src/utils/ado-testing/check-ado.py +++ b/src/utils/ado-testing/check-ado.py @@ -7,8 +7,9 @@ import argparse import os import sys import time + from azure.devops.connection import Connection -from azure.devops.credentials import BasicTokenAuthentication, BasicAuthentication +from azure.devops.credentials import BasicAuthentication, BasicTokenAuthentication from azure.devops.v6_0.work_item_tracking.models import Wiql @@ -17,11 +18,11 @@ def main() -> None: formatter_class=argparse.ArgumentDefaultsHelpFormatter ) parser.add_argument( - "--url", default="https://dev.azure.com/your-instance-name", help="ADO Instance URL" - ) - parser.add_argument( - "--areapath", default="OneFuzz-Test-Project", help="areapath" + "--url", + default="https://dev.azure.com/your-instance-name", + help="ADO Instance URL", ) + parser.add_argument("--areapath", default="OneFuzz-Test-Project", help="areapath") parser.add_argument("--title", help="work item title") parser.add_argument( "--expected", type=int, help="expected number of work items", default=1 @@ -65,9 +66,11 @@ def main() -> None: time.sleep(2) print("trying again", flush=True) - assert len(work_items) >= args.expected, ( - "unexpected work items (got %d, expected at least %d)" - % (len(work_items), args.expected) + assert ( + len(work_items) >= args.expected + ), "unexpected work items (got %d, expected at least %d)" % ( + len(work_items), + args.expected, ) diff --git a/src/utils/ado-testing/check-notifications.py b/src/utils/ado-testing/check-notifications.py index 2a9ae3467..37f917cea 100644 --- a/src/utils/ado-testing/check-notifications.py +++ b/src/utils/ado-testing/check-notifications.py @@ -5,6 +5,7 @@ import json from subprocess import check_call + from onefuzz.api import Onefuzz