/**
     * Constructor for ESSourceReader.
     * @param client The OpenSearch client.
     * @param settings The SuggestSettings instance.
     * @param indexName The name of the index to read from.
     */
    public ESSourceReader(final Client client, final SuggestSettings settings, final String indexName) {
        this.client = client;
        this.settings = settings;
        this.indexName = indexName;

import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.opensearch.index.query.QueryBuilders;
import org.opensearch.transport.client.Client;

public class SuggestUtilTest {

    private static OpenSearchRunner runner;
    private static Client client;
    private static SuggestSettings settings;
    private static final String TEST_INDEX = "test_suggest_util";

    @BeforeClass

  @Test
  fun forciblyStopDispatcher() {
    client =
      client
        .newBuilder()
        .fastFallback(true)
        .build()

    val callFailure = CompletableFuture<Exception>()

    server.enqueue(
      MockResponse()
        .setSocketPolicy(SocketPolicy.STALL_SOCKET_AT_START),
    )
    server.start()
    val call =
      client.newCall(
        Request
          .Builder()

client = TestClient(app)


def test_get():
    response = client.post(
        "/invoices/", json={"id": "fooinvoice", "customer": "John", "total": 5.3}
    )
    assert response.status_code == 200, response.text
    assert response.json() == {"msg": "Invoice received"}


def test_openapi_schema():
    with client:
        response = client.get("/openapi.json")

        Files.writeString(jcifsPropertiesPath, "jcifs.smb.client.nativeOs=FileOS");
        System.setProperty("jcifs.properties", jcifsPropertiesPath.toString());

        System.setProperty("jcifs.smb.client.nativeOs", "SystemOS");

        Properties customProps = new Properties();
        customProps.setProperty("jcifs.smb.client.nativeOs", "CustomOS");

        SingletonContext.init(customProps);

    return {"message": "Hello Users!", "params": commons}


client = TestClient(app)


async def override_dependency(q: Union[str, None] = None):
    return {"q": q, "skip": 5, "limit": 10}


app.dependency_overrides[common_parameters] = override_dependency


def test_override_in_items():
    response = client.get("/items/")
    assert response.status_code == 200
    assert response.json() == {

        # Test that the fix also works for custom pydantic classes
        return SomeCustomClass(a_uuid=MyUuid("b8799909-f914-42de-91bc-95c819218d01"))

    client = TestClient(app)

    with client:
        response_simple = client.get("/fast_uuid")
        response_pydantic = client.get("/get_custom_class")

    assert response_simple.json() == {
        "fast_uuid": "a10ff360-3b1e-4984-a26f-d3ab460bdb51"
    }

    fake_answer_to_everything_ml_model,
    ml_models,
)


def test_events():
    assert not ml_models, "ml_models should be empty"
    with TestClient(app) as client:
        assert ml_models["answer_to_everything"] == fake_answer_to_everything_ml_model
        response = client.get("/predict", params={"x": 2})
        assert response.status_code == 200, response.text
        assert response.json() == {"result": 84.0}

from docs_src.response_model.tutorial003_02_py39 import app

client = TestClient(app)


def test_get_portal():
    response = client.get("/portal")
    assert response.status_code == 200, response.text
    assert response.json() == {"message": "Here's your interdimensional portal."}


def test_get_redirect():
    response = client.get("/portal", params={"teleport": True}, follow_redirects=False)

    return request.param


@pytest.fixture(name="client")
def get_client(mod_name: str) -> TestClient:
    mod = importlib.import_module(f"docs_src.path_operation_configuration.{mod_name}")
    return TestClient(mod.app)


def test_post_items(client: TestClient):
    response = client.post(
        "/items/",
        json={
            "name": "Foo",