fos = new FileOutputStream(newFile);
                writer = new BufferedWriter(new OutputStreamWriter(fos, Constants.UTF_8));
                fos = null; // Successfully wrapped, no need to close explicitly
            } catch (final Exception e) {
                if (fos != null) {
                    try {
                        fos.close();
                    } catch (final IOException ioe) {

     * Check global rate limit
     */
    private boolean checkGlobalRateLimit() {
        long now = System.currentTimeMillis();
        long windowStart = currentWindowStart.get();

        // Check if we need to reset the window (1 minute window)
        if (now - windowStart > 60000) {
            currentWindowStart.compareAndSet(windowStart, now);
            globalAttemptsInCurrentWindow.set(0);
        }

            }
        }
    }

    /**
     * Polls the next URL queue entry for the specified session.
     * This method manages local caches and fetches from OpenSearch when needed.
     *
     * @param sessionId The session ID.
     * @return The next URL queue entry, or null if none available.
     */
    @Override
    public OpenSearchUrlQueue poll(final String sessionId) {

    class EdgeCaseTests {

        @Test
        @DisplayName("Should handle minimum valid buffer size")
        void testMinimumBufferSize() throws Exception {
            byte[] buffer = new byte[24]; // Exact size needed
            SMBUtil.writeInt2(24, buffer, 0);
            buffer[2] = 0x01;

            int bytesRead = notification.readBytesWireFormat(buffer, 0);
            assertEquals(24, bytesRead);
        }

    private SmbTransport.ServerData serverData;
    private SmbComNegotiateResponse response;

    @BeforeEach
    void setUp() throws UnknownHostException {
        // To instantiate the non-static inner class ServerData, we need an instance of the outer class SmbTransport.
        UniAddress uniAddress = new UniAddress(InetAddress.getByName("127.0.0.1"));
        SmbTransport transport = new SmbTransport(uniAddress, 445, null, 0);

@ExtendWith(MockitoExtension.class)
@DisplayName("Type3Message Tests")
class Type3MessageTest {

    @Mock
    private CIFSContext mockContext;

    @BeforeEach
    void setUp() {
        // Basic setup for mock context if needed
    }

    /**
     * Helper method to create properly configured mock context
     */
    private CIFSContext createMockContext() {
        CIFSContext mockCtx = mock(CIFSContext.class);

                        response.wait(timeout);

                        /*
                         * JetDirect printer can respond to regular broadcast query
                         * with node status so we need to check to make sure that
                         * the record type matches the question type and if not,
                         * loop around and try again.
                         */

                && this.dataDisplacement + this.dataCount >= this.totalDataCount) {
            this.hasMore = false;
        }
        return this;
    }

    /**
     * Calculates padding needed for the given offset
     * @param offset
     *            the current offset
     * @return padding size in bytes
     */
    protected int pad(final int offset) {
        final int p = offset % getPadding();

    @DisplayName("Should handle buffer boundary conditions")
    void testBufferBoundaryConditions() throws Exception {
        // Given - notification exactly at buffer end
        byte[] buffer = new byte[104]; // Exact size needed
        int offset = 0;

        setHeaderStart(response, 64);

        // Write structure
        SMBUtil.writeInt2(9, buffer, offset);
        SMBUtil.writeInt2(80 - 64, buffer, offset + 2);

                fos = new FileOutputStream(newFile);
                writer = new BufferedWriter(new OutputStreamWriter(fos, Constants.UTF_8));
                fos = null; // Successfully wrapped, no need to close explicitly
            } catch (final Exception e) {
                if (fos != null) {
                    try {
                        fos.close();
                    } catch (final IOException ioe) {