trw.WriteHeader(123)
	if !trw.headerWritten {
		t.Fatal("headerWritten was not set by WriteHeader call")
	}

	_, err := trw.Write([]byte("hello"))
	if err != nil {
		t.Fatalf("Write unexpectedly failed: %v", err)
	}

	// Check that WriteHeader and Write were called on the underlying response writer
	resp := rw.Result()
	if resp.StatusCode != 123 {

    }

    @Test
    public void test_crawlingInfoPager_setSessionId() {
        final CrawlingInfoPager pager = new CrawlingInfoPager();
        pager.sessionId = "test-session-123";
        assertEquals("test-session-123", pager.sessionId);
    }

    @Test
    public void test_crawlingInfoPager_setId() {
        final CrawlingInfoPager pager = new CrawlingInfoPager();
        pager.id = "crawling-id-456";

        // Setup is handled by MockitoExtension
    }

    @Test
    void constructor_shouldOpenPolicySuccessfully() throws IOException {
        // Arrange
        String server = "testServer";
        int access = 123;

        // Mock the behavior of sendrecv for MsrpcLsarOpenPolicy2
        doAnswer(invocation -> {
            MsrpcLsarOpenPolicy2 rpc = invocation.getArgument(0);
            rpc.retval = 0; // Simulate success

    }

    /**
     * Edge case – constructor using the winerr flag.
     */
    @Test
    @DisplayName("Winerr code 123 – message equals 123")
    void testConstructorWithWinerr() {
        SmbException ex = new SmbException(123, true);
        assertEquals("123", ex.getMessage());
        assertEquals(123, ex.getNtStatus());
    }

    /**
     * Parameterised test covering many error codes.

            // When: Creating connect2 message
            samr.SamrConnect2 message = new samr.SamrConnect2("system", 123, mockPolicyHandle);

            // Then: Should have correct values
            assertEquals(0x39, message.getOpnum());
            assertEquals("system", message.system_name);
            assertEquals(123, message.access_mask);
            assertEquals(mockPolicyHandle, message.handle);
        }

        @Test

    }

    @Test
    public void test_createSuccessResponse_basic() {
        final Map<String, Object> response = chatApiManager.createSuccessResponse("session-123", "Hello, world!", null);

        assertEquals("ok", response.get("status"));
        assertEquals("session-123", response.get("sessionId"));
        assertEquals("Hello, world!", response.get("content"));
        assertFalse(response.containsKey("sources"));
    }

    }

    @Test
    public void test_type_conversion_from_string() {
        Map<String, Object> doc = new HashMap<>();

        doc.put("intStr", "123");
        assertEquals(Integer.valueOf(123), DocumentUtil.getValue(doc, "intStr", Integer.class));

        doc.put("longStr", "123456789");
        assertEquals(Long.valueOf(123456789L), DocumentUtil.getValue(doc, "longStr", Long.class));

    @Test
    public void test_toString() {
        // Test toString method
        ProtwordsItem item = new ProtwordsItem(123, "testword");
        assertEquals("ProtwordsItem [id=123, inputs=testword, newInputs=null]", item.toString());

        item.setNewInput("newword");
        assertEquals("ProtwordsItem [id=123, inputs=testword, newInputs=newword]", item.toString());

        ProtwordsItem item2 = new ProtwordsItem(0, "createword");

        SID unknownSid = new SID(rpcSid, SID.SID_TYPE_UNKNOWN, null, null, false);
        unknownSid.sub_authority_count = 1;
        unknownSid.sub_authority = new int[] { 123 };
        assertEquals("123", unknownSid.getAccountName());
    }

    /**
     * Test getting the SID type as text.
     */
    @Test
    void testGetTypeText() {
        rpc.sid_t rpcSid = new rpc.sid_t();

        assertEquals(0, crawlingConfigHelper.getExcludedUrlList("123").size());
        systemProperties.setProperty(Constants.FAILURE_COUNT_THRESHOLD_PROPERTY, "0");
        assertEquals(0, crawlingConfigHelper.getExcludedUrlList("123").size());
        errorCount.set(5);
        assertEquals(5, crawlingConfigHelper.getExcludedUrlList("123").size());