samypr100
77 lines
2.7 KiB
Python
77 lines
2.7 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
# The OpenSearch Contributors require contributions made to
|
|
# this file be licensed under the Apache-2.0 license or a
|
|
# compatible open source license.
|
|
#
|
|
# Modifications Copyright OpenSearch Contributors. See
|
|
# GitHub history for details.
|
|
import os
|
|
from typing import Type
|
|
|
|
import pytest
|
|
from pytest import MarkDecorator
|
|
|
|
from opensearchpy import (
|
|
AIOHttpConnection,
|
|
AsyncConnection,
|
|
AsyncHttpConnection,
|
|
AsyncOpenSearch,
|
|
)
|
|
from opensearchpy._async.helpers.test import get_test_client
|
|
|
|
pytestmark: MarkDecorator = pytest.mark.asyncio
|
|
|
|
|
|
class TestAIOHttp:
|
|
|
|
def test_default(self) -> None:
|
|
client = AsyncOpenSearch()
|
|
assert client.transport.connection_class == AIOHttpConnection
|
|
assert client.transport.pool_maxsize is None
|
|
|
|
def test_connection_class(self) -> None:
|
|
client = AsyncOpenSearch(connection_class=AsyncHttpConnection)
|
|
assert client.transport.connection_class == AsyncHttpConnection
|
|
assert client.transport.pool_maxsize is None
|
|
|
|
def test_pool_maxsize(self) -> None:
|
|
client = AsyncOpenSearch(connection_class=AsyncHttpConnection, pool_maxsize=42)
|
|
assert client.transport.connection_class == AsyncHttpConnection
|
|
assert client.transport.pool_maxsize == 42
|
|
|
|
@pytest.mark.parametrize( # type: ignore[misc]
|
|
"connection_class", [AIOHttpConnection, AsyncHttpConnection]
|
|
)
|
|
async def test_default_limit(self, connection_class: Type[AsyncConnection]) -> None:
|
|
client = await get_test_client(
|
|
connection_class=connection_class,
|
|
verify_certs=False,
|
|
http_auth=("admin", os.getenv("OPENSEARCH_PASSWORD", "admin")),
|
|
)
|
|
assert isinstance(
|
|
client.transport.connection_pool.connections[0], connection_class
|
|
)
|
|
assert (
|
|
client.transport.connection_pool.connections[0].session.connector.limit # type: ignore[attr-defined]
|
|
== 10
|
|
)
|
|
|
|
@pytest.mark.parametrize( # type: ignore[misc]
|
|
"connection_class", [AIOHttpConnection, AsyncHttpConnection]
|
|
)
|
|
async def test_custom_limit(self, connection_class: Type[AsyncConnection]) -> None:
|
|
client = await get_test_client(
|
|
connection_class=connection_class,
|
|
verify_certs=False,
|
|
pool_maxsize=42,
|
|
http_auth=("admin", os.getenv("OPENSEARCH_PASSWORD", "admin")),
|
|
)
|
|
assert isinstance(
|
|
client.transport.connection_pool.connections[0], connection_class
|
|
)
|
|
assert (
|
|
client.transport.connection_pool.connections[0].session.connector.limit # type: ignore[attr-defined]
|
|
== 42
|
|
)
|