}

        return needsRotation;
    }

    /**
     * Reset key rotation tracking after new keys are established
     */
    public void resetKeyRotationTracking() {
        bytesEncrypted.set(0);
        encryptionStartTime = System.currentTimeMillis();
        log.debug("Key rotation tracking reset");
    }

    /**

        // Given
        Smb2EncryptionContext context = new Smb2EncryptionContext(1, DialectVersion.SMB311, testEncryptionKey, testDecryptionKey);

        // When
        context.resetKeyRotationTracking();

        // Then
        assertFalse(context.needsKeyRotation(), "Should not need rotation after reset");
    }

    @Test