long seq1 = registration.getNextSequenceNumber();
        long seq2 = registration.getNextSequenceNumber();
        long seq3 = registration.getNextSequenceNumber();

        assertEquals(1, seq1);
        assertEquals(2, seq2);
        assertEquals(3, seq3);
        assertTrue(seq2 > seq1);
        assertTrue(seq3 > seq2);
    }

    @Test
    void testHeartbeatUpdate() throws InterruptedException {

    assertFalse(registration.isExpired(60000));
    
    // Test sequence numbers
    long seq1 = registration.getNextSequenceNumber();
    long seq2 = registration.getNextSequenceNumber();
    assertEquals(seq1 + 1, seq2);
}

@Test
public void testWitnessNotification() {
    WitnessNotification notification = new WitnessNotification(
        WitnessEventType.CLIENT_MOVE, "TestResource");

        Smb2TreeDisconnectRequest req1 = new Smb2TreeDisconnectRequest(mockConfig);
        Smb2TreeDisconnectRequest req2 = new Smb2TreeDisconnectRequest(mockConfig);

        // Then - all should have same command
        int cmd1 = (int) commandField.get(req1);
        int cmd2 = (int) commandField.get(req2);

        assertEquals(cmd1, cmd2);
        assertEquals(SMB2_TREE_DISCONNECT, cmd1);
    }

    @Test

  @Test
  fun serverDisconnectsBeforeSecondRequestHttp1() {
    enableProtocol(Protocol.HTTP_1_1)

    server.enqueue(MockResponse(code = 200, body = "Req1"))
    server.enqueue(MockResponse(code = 200, body = "Req2"))

    val eventListener =
      object : EventListener() {
        var socket: SSLSocket? = null
        var closed = false

        override fun connectionAcquired(
          call: Call,