private static class TestSessionServicePacket extends SessionServicePacket {
        int trailerLength = 10; // Default trailer length for testing
        int trailerBytesRead = 0;

        @Override
        int writeTrailerWireFormat(byte[] dst, int dstIndex) {
            // Simple mock implementation
            for (int i = 0; i < trailerLength && dstIndex + i < dst.length; i++) {
                dst[dstIndex + i] = (byte) i;

        @ParameterizedTest
        @ValueSource(ints = { NegTokenInit.DELEGATION, NegTokenInit.MUTUAL_AUTHENTICATION, NegTokenInit.REPLAY_DETECTION,
                NegTokenInit.SEQUENCE_CHECKING, NegTokenInit.ANONYMITY, NegTokenInit.CONFIDENTIALITY, NegTokenInit.INTEGRITY })
        @DisplayName("Individual flag values are correctly set and retrieved")
        void testIndividualFlagValues(int flag) {
            NegTokenInit init = new NegTokenInit();

        assertEquals(1, pool.failCounts.get("10.0.0.1"));
        assertEquals(1, pool.failCounts.get("10.0.0.2"));
    }

    @ParameterizedTest
    @ValueSource(ints = { 0, -1 })
    @DisplayName("Should default to port 445 when port <= 0")
    void testDefaultPort(int invalidPort) throws Exception {
        // Given: Create a new pool for this test to ensure isolation
        SmbTransportPoolImpl testPool = new SmbTransportPoolImpl();

        void testSizeCalculation(int hashAlgoCount) {
            // Arrange
            int[] hashAlgos = new int[hashAlgoCount];
            byte[] salt = new byte[16];
            context = new PreauthIntegrityNegotiateContext(mockConfig, hashAlgos, salt);

            // Act
            int size = context.size();

            // Assert
            assertEquals(4 + (2 * hashAlgoCount) + 16, size);

            for (int i = 0; i < 100; i++) {
                int bytesRead = response.readBytesWireFormat(buffer, 0);
                // Then
                assertEquals(4, bytesRead, "Failed at iteration " + i);
            }

            // When - perform multiple write operations
            for (int i = 0; i < 100; i++) {
                int bytesWritten = response.writeBytesWireFormat(buffer, 0);

        // Given - large buffer with many notifications
        byte[] buffer = new byte[65536];
        int offset = 0;

        setHeaderStart(response, 64);

        // Calculate space for notifications
        int notificationSize = 32; // Each notification
        int notificationCount = 100;
        int totalSize = notificationSize * notificationCount;

        // Write structure

        String shareName = "SHARE1";
        int shareType = 0x00000000; // Disk share
        String remark = "Test share";

        // Calculate buffer size
        int shareNameSize = 14; // 13 bytes + null terminator (fixed field)
        int typeSize = 2;
        int offsetSize = 4;
        int remarkSize = remark.length() + 1;
        int totalSize = shareNameSize + typeSize + offsetSize + remarkSize;

  @SuppressWarnings("JdkObsolete")
  private static Enumeration<Integer> enumerate(int... ints) {
    Vector<Integer> vector = new Vector<>(Ints.asList(ints));
    return vector.elements();
  }

  public void testToString() {
    Iterator<String> iterator = Lists.newArrayList("yam", "bam", "jam", "ham").iterator();
    assertThat(Iterators.toString(iterator)).isEqualTo("[yam, bam, jam, ham]");
  }

        byte[] buffer = new byte[256];
        int result = request.readBytesWireFormat(buffer, 0);

        assertEquals(0, result);
    }

    @ParameterizedTest
    @ValueSource(ints = { 0, 10, 50, 100 })
    @DisplayName("writeBytesWireFormat should work at different offsets")
    void testWriteBytesWireFormatAtDifferentOffsets(int offset) {
        byte[] buffer = new byte[512];

    void testWriteParametersWireFormatNetServerEnum2() throws Exception {
        String domain = "TESTDOMAIN";
        int serverTypes = NetServerEnum2.SV_TYPE_ALL;
        netServerEnum2 = new NetServerEnum2(realConfig, domain, serverTypes);

        byte[] dst = new byte[1024];
        int dstIndex = 0;

        int bytesWritten = netServerEnum2.writeParametersWireFormat(dst, dstIndex);

        // Verify bytes written