t.Errorf(fmt.Sprintf("Count should work, but got err %v", err))
	}
	if count1 != 1 {
		t.Errorf("Count with group should be 1, but got count: %v", count1)
	}

	var count2 int64
	if err := DB.Model(&Company{}).Where("name in ?", []string{"company_count_group_b", "company_count_group_c"}).Group("name").Count(&count2).Error; err != nil {
		t.Errorf(fmt.Sprintf("Count should work, but got err %v", err))
	}
	if count2 != 2 {

        response.setDataCount(5);

        // With 0 entries, should return dataCount
        int result = response.readDataWireFormat(buffer, 0, 0);
        // Should return dataCount
        assertEquals(5, result);
    }

    @Test
    void testGetSid() {
        // Test the getSid method
        // By default, it should be 0
        assertEquals(0, response.getSid());
    }

    @Test

        assertTrue(config.isUseRDMA(), "RDMA should be enabled");
        assertEquals("disni", config.getRdmaProvider(), "Provider should be disni");
        assertEquals(16384, config.getRdmaReadWriteThreshold(), "Threshold should be 16KB");
        assertEquals(131072, config.getRdmaMaxSendSize(), "Max send size should be 128KB");
        assertEquals(131072, config.getRdmaMaxReceiveSize(), "Max receive size should be 128KB");

    }

    @Test
    void testNtlmNegotiateOem() {
        assertEquals(0x00000002, NtlmFlags.NTLMSSP_NEGOTIATE_OEM, "NTLMSSP_NEGOTIATE_OEM should be 0x00000002");
    }

    @Test
    void testNtlmRequestTarget() {
        assertEquals(0x00000004, NtlmFlags.NTLMSSP_REQUEST_TARGET, "NTLMSSP_REQUEST_TARGET should be 0x00000004");
    }

    @Test

        // Then - Buffer should have expected size
        assertNotNull(buffer, "Buffer should not be null");
        assertEquals(SmbComTransaction.TRANSACTION_BUF_SIZE, buffer.length, "Buffer should have TRANSACTION_BUF_SIZE");

        // The validation in getBuffer() ensures size is within bounds (0 < size <= 1MB)
        assertTrue(buffer.length > 0, "Buffer size should be positive");

        String hostName = mockAddress.getHostName();

        // Then
        assertEquals(expectedHostName, hostName, "Should return configured hostname");
        assertNotNull(hostName, "Host name should not be null");
    }

    @Test
    @DisplayName("getHostAddress should return valid IP address string")
    void testGetHostAddressContract() {
        // Given
        String expectedAddress = "192.168.1.100";

            assertTrue(methodNames.contains("getConfig"), "Should contain getConfig method");
            assertTrue(methodNames.contains("unwrap"), "Should contain unwrap method");
            assertTrue(methodNames.contains("getContext"), "Should contain getContext method");
        }

        @Test
        @DisplayName("Should be a public interface")

        // Manually unblock
        assertTrue(rateLimiter.unblockIp(ip), "Manual unblock should succeed");

        // Should be allowed again
        assertTrue(rateLimiter.checkAttempt("user7", ip), "Should be allowed after manual unblock");
    }

    @Test
    public void testNullUsername() throws Exception {
        // Null username should be handled gracefully

    }

    @Test
    @DisplayName("Should handle empty byte array")
    void testToHexStringEmpty() {
        // Given
        byte[] data = {};

        // When
        String result = Hexdump.toHexString(data);

        // Then
        assertNotNull(result);
        assertEquals("", result);
    }

    @Test
    @DisplayName("Should handle null byte array")
    void testToHexStringNull() {

        }

        @Test
        @DisplayName("Should throw exception on null buffer")
        void testEncodeWithNullBuffer() {
            assertThrows(IllegalArgumentException.class, () -> testRequest.encode(null, 0), "Should throw exception for null buffer");
        }

        @Test
        @DisplayName("Should throw exception on negative index")
        void testEncodeWithNegativeIndex() {