// When - navigate through chain
        CommonServerMessageBlockResponse second = response.getNextResponse();
        CommonServerMessageBlockResponse third = second.getNextResponse();
        CommonServerMessageBlockResponse end = third.getNextResponse();

        // Then
        assertEquals(nextResponse, second);
        assertEquals(thirdResponse, third);
        assertNull(end);
    }

    @Test

			}

			// If we are doing batching, we should wait
			// at least for a second, before sending.
			// Even if there is nothing in the queue.
			if h.batchSize > 1 && time.Since(lastBatchProcess) < time.Second {
				continue
			}
		}

		// if we have reached the count send at once
		// or we have crossed last second before batch was sent, send at once
		lastBatchProcess = time.Now()

		var retries int

    taskFaker.runTasks()
    assertThat(client.closed).isTrue()
    server.listener.assertExhausted() // Client should not have sent second close.
    client.listener.assertExhausted() // Server should not have sent second close.
  }

  @Test
  fun serverCloseBreaksReadMessageLoop() {
    server.webSocket!!.send("Hello!")
    server.webSocket!!.close(1000, "Bye!")
    assertThat(client.processNextFrame()).isTrue()

        when(mockFile.openUnshared(anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenReturn(mockFileHandle);

        when(mockTreeHandle.send(any(Smb2WriteRequest.class), any())).thenReturn(mockWriteResponse);
        when(mockWriteResponse.getCount()).thenReturn(6, 5); // First write returns 6, second returns 5

        outputStream = new SmbFileOutputStream(mockFile, mockTreeHandle, mockFileHandle,

        c.acquire();
        verify(tree, times(1)).acquire();

        // Act: second acquire does not call tree.acquire again
        c.acquire();
        verify(tree, times(1)).acquire();

        // Act: first release does not release yet (usage -> 1)
        c.release();
        verify(tree, never()).release();

        // Act: second release (usage -> 0) releases the tree
        c.release();

	// dRWMutexUnlockTimeout - default timeout for the unlock call
	drwMutexUnlockCallTimeout = 30 * time.Second

	// dRWMutexForceUnlockTimeout - default timeout for the unlock call
	drwMutexForceUnlockCallTimeout = 30 * time.Second

	// dRWMutexRefreshInterval - default the interval between two refresh calls
	drwMutexRefreshInterval = 10 * time.Second

	drwMutexInfinite = 1<<63 - 1
)

        when(fileHandle.acquire()).thenReturn(fileHandle);
        when(fileHandle.getTree()).thenReturn(tree);

        // Mock response and send behavior
        Smb2IoctlResponse resp = mock(Smb2IoctlResponse.class);
        when(resp.getOutputLength()).thenReturn(42);
        when(tree.send(any(Smb2IoctlRequest.class), any())).thenReturn(resp);

        byte[] out = new byte[10];

	readBufferSize     = 32 << 10 // 32 KiB is the most optimal on Linux
	writeBufferSize    = 32 << 10 // 32 KiB is the most optimal on Linux
	defaultDialTimeout = 2 * time.Second
	connPingInterval   = 10 * time.Second
	connWriteTimeout   = 3 * time.Second
)

type connectionParams struct {
	ctx           context.Context
	id            uuid.UUID
	local, remote string
	handlers      *handlers

            TestServerMessageBlock2 second = new TestServerMessageBlock2(mockConfig);
            TestServerMessageBlock2 third = new TestServerMessageBlock2(mockConfig);

            assertTrue(testMessage.chain(second));
            assertTrue(testMessage.chain(third));

            assertEquals(second, testMessage.getNext());
            assertEquals(third, second.getNext());
        }

        @Test

        lenient().when(fh.isValid()).thenReturn(true);
        lenient().when(fh.getTree()).thenReturn(tree);
        when(fh.getFid()).thenReturn(99);

        // Mock send to just echo the provided response instance
        when(tree.send(any(TransTransactNamedPipe.class), any(TransTransactNamedPipeResponse.class), eq(RequestParam.NO_RETRY)))
                .thenAnswer(inv -> inv.getArgument(1));