diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py index 7b3788d..82932a2 100644 --- a/tests/unit/conftest.py +++ b/tests/unit/conftest.py @@ -18,9 +18,12 @@ from threading import Thread from typing import Generator -import pytest from aiofiles.tempfile import TemporaryDirectory -from mocks import FakeAlloyDBClient, FakeCredentials, FakeInstance, metadata_exchange +import pytest +from mocks import FakeAlloyDBClient +from mocks import FakeCredentials +from mocks import FakeInstance +from mocks import metadata_exchange from google.cloud.alloydb.connector.utils import _write_to_file diff --git a/tests/unit/mocks.py b/tests/unit/mocks.py index f12ba45..4dc0b25 100644 --- a/tests/unit/mocks.py +++ b/tests/unit/mocks.py @@ -13,23 +13,27 @@ # limitations under the License. import asyncio +from datetime import datetime +from datetime import timedelta +from datetime import timezone import io import ipaddress import json import ssl import struct -from datetime import datetime, timedelta, timezone from typing import Any, Callable, Literal, Optional from cryptography import x509 -from cryptography.hazmat.primitives import hashes, serialization +from cryptography.hazmat.primitives import hashes +from cryptography.hazmat.primitives import serialization from cryptography.hazmat.primitives.asymmetric import rsa from cryptography.x509.oid import NameOID -from google.auth.credentials import TokenState, _helpers +from google.auth.credentials import _helpers +from google.auth.credentials import TokenState from google.auth.transport import requests -import google.cloud.alloydb_connectors_v1.proto.resources_pb2 as connectorspb from google.cloud.alloydb.connector.connection_info import ConnectionInfo +import google.cloud.alloydb_connectors_v1.proto.resources_pb2 as connectorspb class FakeCredentials: diff --git a/tests/unit/test_async_connector.py b/tests/unit/test_async_connector.py index 3d5b171..e2da0ee 100644 --- a/tests/unit/test_async_connector.py +++ b/tests/unit/test_async_connector.py @@ -15,17 +15,16 @@ import asyncio from typing import Union -import pytest from aiohttp import ClientResponseError from mock import patch -from mocks import ( - FakeAlloyDBClient, - FakeConnectionInfo, - FakeCredentials, - write_static_info, -) +from mocks import FakeAlloyDBClient +from mocks import FakeConnectionInfo +from mocks import FakeCredentials +from mocks import write_static_info +import pytest -from google.cloud.alloydb.connector import AsyncConnector, IPTypes +from google.cloud.alloydb.connector import AsyncConnector +from google.cloud.alloydb.connector import IPTypes from google.cloud.alloydb.connector.exceptions import IPTypeNotFoundError from google.cloud.alloydb.connector.instance import RefreshAheadCache diff --git a/tests/unit/test_connector.py b/tests/unit/test_connector.py index 359036e..c7660d1 100644 --- a/tests/unit/test_connector.py +++ b/tests/unit/test_connector.py @@ -16,12 +16,15 @@ from threading import Thread from typing import Union -import pytest from aiohttp import ClientResponseError from mock import patch -from mocks import FakeAlloyDBClient, FakeCredentials, write_static_info +from mocks import FakeAlloyDBClient +from mocks import FakeCredentials +from mocks import write_static_info +import pytest -from google.cloud.alloydb.connector import Connector, IPTypes +from google.cloud.alloydb.connector import Connector +from google.cloud.alloydb.connector import IPTypes from google.cloud.alloydb.connector.exceptions import IPTypeNotFoundError from google.cloud.alloydb.connector.instance import RefreshAheadCache from google.cloud.alloydb.connector.utils import generate_keys