Skip to content

Commit

Permalink
Fix lint errors for imports
Browse files Browse the repository at this point in the history
  • Loading branch information
rhatgadkar-goog committed Jan 31, 2025
1 parent aac315c commit d17da68
Show file tree
Hide file tree
Showing 4 changed files with 26 additions and 17 deletions.
7 changes: 5 additions & 2 deletions tests/unit/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
12 changes: 8 additions & 4 deletions tests/unit/mocks.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
15 changes: 7 additions & 8 deletions tests/unit/test_async_connector.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
9 changes: 6 additions & 3 deletions tests/unit/test_connector.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit d17da68

Please sign in to comment.