openFunction = OPEN_FN_CREATE | OPEN_FN_OPEN;
            }
        } else {
            openFunction = OPEN_FN_OPEN;
        }
    }

    @Override
    int getBatchLimit(final byte command) {
        return command == SMB_COM_READ_ANDX ? BATCH_LIMIT : 0;
    }

    @Override
    int writeParameterWordsWireFormat(final byte[] dst, int dstIndex) {
        final int start = dstIndex;

    }

    @Test
    @DisplayName("constructor sets correct command value")
    void constructorSetsCorrectCommand() {
        SmbComLogoffAndX msg = new SmbComLogoffAndX(null);
        assertNotNull(msg, "Message should not be null");
        // SMB_COM_LOGOFF_ANDX = 0x74
        assertEquals((byte) 0x74, msg.command, "Command should be SMB_COM_LOGOFF_ANDX");
    }

    @Test

    @Override
    protected <T> Callable<T> wrapTask(Callable<T> callable) {
      return new WrappedCallable<T>(callable);
    }

    @Override
    protected Runnable wrapTask(Runnable command) {
      return new WrappedRunnable(command);
    }
  }

  private static final class MockExecutor implements ScheduledExecutorService {
    String lastMethodCalled = "";
    long lastInitialDelay;
    long lastDelay;

    void testToStringUnknownCommand() {
        smb.command = (byte) 0xFF; // Unknown command
        smb.errorCode = 0;
        smb.mid = 1;
        String str = smb.toString();
        assertTrue(str.contains("command=UNKNOWN"));
    }

        // Scenario 1: Command not found
        CommandExecutionException cmdNotFound = new CommandExecutionException("Command 'xyz' not found");
        assertTrue(cmdNotFound.getMessage().contains("not found"));

        // Scenario 2: Command timeout
        CommandExecutionException timeout =

        result = response.readParametersWireFormat(smallBuffer, 0, -1);
        assertEquals(0, result);
    }

    @Test
    @DisplayName("Test command type is correct")
    void testCommandType() {
        // Verify that the command is not set in constructor (defaults to 0)
        assertEquals(0, response.getCommand());

        // Verify we can change the subcommand

        private byte[] rawPayload = null;

        public TestAndXServerMessageBlock(Configuration config, byte command, ServerMessageBlock andx) {
            super(config, command, andx);
        }

        public TestAndXServerMessageBlock(Configuration config, byte command) {
            super(config, command);
        }

        public TestAndXServerMessageBlock(Configuration config) {
            super(config);
        }

                "Command should change to SMB_COM_TRANSACTION_SECONDARY");
    }

    @Test
    @DisplayName("Test NT transaction command handling")
    void testNtTransactionCommand() {
        transaction.command = ServerMessageBlock.SMB_COM_NT_TRANSACT;

        // First call
        transaction.nextElement();

        // Second call should change to NT_TRANSACT_SECONDARY

    }

    @Test
    public void testConstructor() {
        // Test if the constructor sets the file name and command correctly
        assertEquals(TEST_FILE_NAME, smbComQueryInformation.path);
        assertEquals(ServerMessageBlock.SMB_COM_QUERY_INFORMATION, smbComQueryInformation.command);
    }

    @Test
    public void testWriteParameterWordsWireFormat() {

    void testConstructor() {
        assertEquals(testDomain, netServerEnum2.domain);
        assertEquals(testServerTypes, netServerEnum2.serverTypes);
        assertEquals(ServerMessageBlock.SMB_COM_TRANSACTION, netServerEnum2.command);
        assertEquals(SmbComTransaction.NET_SERVER_ENUM2, netServerEnum2.subCommand);
        assertEquals("\\PIPE\\LANMAN", netServerEnum2.name);
        assertEquals(8, netServerEnum2.maxParameterCount);