long timeout = RESPONSE_TIMEOUT;
                            resp.expiration = System.currentTimeMillis() + timeout;
                            while (resp.hasMoreElements()) {
                                wait(timeout);
                                timeout = resp.expiration - System.currentTimeMillis();
                                if (timeout <= 0) {

        long serverTimeZoneOffset = 3600000L; // 1 hour
        response = new SmbComQueryInformationResponse(mockConfig, serverTimeZoneOffset);

        // Set a test time using reflection
        long testTime = System.currentTimeMillis();
        setFieldValue(response, "lastWriteTime", testTime);

        // All time methods should return the same converted time
        long expectedTime = testTime + serverTimeZoneOffset;

        // Max write size
        SMBUtil.writeInt4(1048576, buffer, offset + 36);

        // System time
        SMBUtil.writeTime(System.currentTimeMillis(), buffer, offset + 40);

        // Server start time
        SMBUtil.writeTime(System.currentTimeMillis() - 3600000, buffer, offset + 48);

        // Security buffer offset and length
        SMBUtil.writeInt2(128, buffer, offset + 56); // Offset

        if (logger.isInfoEnabled()) {
            logger.info("Restoring elevate words: index={}", index);
        }
        final long start = System.currentTimeMillis();
        int numberOfSuggestDocs = 0;
        int numberOfInputDocs = 0;

        final ElevateWord[] elevateWords = settings.elevateWord().get();

        source.put(FieldNames.LANGUAGES, Arrays.asList("en"));
        source.put(FieldNames.KINDS, Arrays.asList("document"));
        source.put(FieldNames.TIMESTAMP, System.currentTimeMillis());

        SuggestItem item = SuggestItemSerializer.fromSource(source);

        assertNotNull(item);
        assertEquals("test text", item.getText());
        assertEquals(10L, item.getDocFreq());

    public static final int WITNESS_RESOURCE_STATE_UNAVAILABLE = 0x000000FF;

    /**
     * Creates a new empty witness notification.
     */
    public WitnessNotification() {
        this.timestamp = System.currentTimeMillis();
        this.newIPAddresses = new ArrayList<>();
        this.oldIPAddresses = new ArrayList<>();
        this.flags = WITNESS_RESOURCE_STATE_UNKNOWN;
    }

    /**
     * Creates a new witness notification.

            }

            if (!exception.isRetryable()) {
                return false;
            }

            // Check if we've exceeded time limits
            long totalElapsed = System.currentTimeMillis() - exception.getOperationStartTime();
            if (totalElapsed > maxDelayMs * maxAttempts) {
                log.debug("Retry time limit exceeded for operation");
                return false;
            }

        dataStoreFactory = new DataStoreFactory();
        systemHelper = new SystemHelper() {
            @Override
            public long getCurrentTimeAsLong() {
                return System.currentTimeMillis();
            }
        };
        ComponentUtil.register(systemHelper, "systemHelper");

        // Create temporary directory for test JAR files

        rawKeys.put(sessionId, keyClone);

        // Track creation time for rotation (only for non-archived keys)
        if (!sessionId.contains(".v")) {
            keyCreationTimes.put(sessionId, System.currentTimeMillis());
            keyVersions.putIfAbsent(sessionId, 0);
        }

        // Optionally store in KeyStore
        if (keyStore != null) {
            try {

        super(message);
        this.errorCode = errorCode;
        this.severity = severity;
        this.category = category;
        this.context = new HashMap<>();
        this.timestamp = System.currentTimeMillis();
    }

    /**
     * Creates an SMB exception with cause
     *
     * @param message the error message
     * @param errorCode the SMB error code
     * @param severity the error severity