def test_security_scopes_sub_dependency_caching(
    client: TestClient, call_counts: dict[str, int]
):
    response = client.get("/")

    assert response.status_code == 200
    assert call_counts["get_db_session"] == 1
    assert call_counts["get_current_user"] == 2
    assert call_counts["get_user_me"] == 2
    assert call_counts["get_user_items"] == 1
    assert response.json() == {
        "user_me": {

from fastapi.testclient import TestClient


@pytest.fixture(name="call_counter")
def call_counter_fixture():
    return {"count": 0}


@pytest.fixture(name="app")
def app_fixture(call_counter: dict[str, int]):
    def get_db():
        call_counter["count"] += 1
        return f"db_{call_counter['count']}"

    def get_user(db: Annotated[str, Depends(get_db)]):
        return "user"

        SearchLogHelper mockSearchLogHelper = new SearchLogHelper() {
            private int callCount = 0;

            @Override
            public void storeSearchLog() {
                callCount++;
                throw new RuntimeException("Error " + callCount);
            }
        };
        ComponentUtil.register(mockSearchLogHelper, "searchLogHelper");

    @Test
    public void test_evaluate_concurrency() {
        queryRescorer = new QueryRescorer() {
            private int callCount = 0;

            @Override
            public synchronized RescorerBuilder<?> evaluate(Map<String, Object> params) {
                callCount++;
                if (callCount % 2 == 0) {
                    return createMockRescorerBuilder();
                }
                return null;

              public Object load(String key) throws InterruptedException {
                callCount.incrementAndGet();
                startSignal.await();
                return result;
              }
            });

    List<Object> resultArray = doConcurrentGet(cache, "bar", count, startSignal);

    assertThat(callCount.get()).isEqualTo(1);
    for (int i = 0; i < count; i++) {

    @Test
    public void test_addShutdownListener_replace() {
        // Test replacing shutdown listener
        AtomicInteger callCount = new AtomicInteger(0);
        ShutdownListener listener1 = new ShutdownListener() {
            @Override
            public void onShutdown() {
                callCount.incrementAndGet();
            }
        };

        ShutdownListener listener2 = new ShutdownListener() {
            @Override

              public Object load(String key) throws InterruptedException {
                callCount.incrementAndGet();
                startSignal.await();
                return result;
              }
            });

    List<Object> resultArray = doConcurrentGet(cache, "bar", count, startSignal);

    assertThat(callCount.get()).isEqualTo(1);
    for (int i = 0; i < count; i++) {

        // Test storing various data types
        final AtomicInteger callCount = new AtomicInteger(0);

        IndexUpdateCallback callback = new IndexUpdateCallback() {
            @Override
            public void store(DataStoreParams paramMap, Map<String, Object> dataMap) {
                assertNotNull(dataMap);
                callCount.incrementAndGet();
            }

            @Override

            ThumbnailManager mockThumbnailManager = new ThumbnailManager() {
                private int callCount = 0;

                @Override
                public int generate(ExecutorService executorService, boolean cleanup) {
                    // Return 0 to exit the loop
                    if (callCount++ == 0) {
                        return 1;
                    }
                    return 0;

        SmbWatchHandle realHandle = new SmbWatchHandle() {
            private int callCount = 0;

            @Override
            public List<FileNotifyInformation> watch() throws CIFSException {
                callCount++;
                if (callCount == 1) {
                    return Arrays.asList(fileNotifyInfo1);
                }
                return Collections.emptyList();