}

    @Test
    void testTrans2QueryPathUnsupportedLevel() throws Exception {
        Trans2QueryPathInformationResponse trans = new Trans2QueryPathInformationResponse(9999) {
        };
        assertEquals(0, trans.readDataWireFormat(new byte[10], 0, 10));
        java.lang.reflect.Field infoField = Trans2QueryPathInformationResponse.class.getDeclaredField("info");
        infoField.setAccessible(true);

        super.setUp();
        labelTypeHelper = new LabelTypeHelper();
        ComponentUtil.register(new SystemHelper(), "systemHelper");
        ComponentUtil.register(new MockLabelTypeService(), LabelTypeService.class.getCanonicalName());
        ComponentUtil.register(new MockVirtualHostHelper(), "virtualHostHelper");
        ComponentUtil.register(new MockRoleQueryHelper(), "roleQueryHelper");
    }

    }

    public void test_load_virtualHosts() {
        List<RelatedQuery> testData = new ArrayList<>();
        testData.add(createRelatedQuery("term1", new String[] { "query1" }, "host1"));
        testData.add(createRelatedQuery("term1", new String[] { "query2" }, "host2"));
        testData.add(createRelatedQuery("term2", new String[] { "query3" }, ""));
        mockBhv.setTestData(testData);

    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockContext.getConfig()).thenReturn(mockConfig);

        testFileId = new byte[16];
        new SecureRandom().nextBytes(testFileId);
        outputBuffer = new byte[4096];
        request = new Smb2ReadRequest(mockConfig, testFileId, outputBuffer, 0);
    }

    @Nested
    @DisplayName("Constructor Tests")
    class ConstructorTests {

            context = new PreauthIntegrityNegotiateContext(mockConfig, new int[3], new byte[32]);
            assertEquals(4 + 6 + 32, context.size()); // header + 3*2 + 32

            context = new PreauthIntegrityNegotiateContext(mockConfig, null, new byte[16]);
            assertEquals(4 + 16, context.size());

            context = new PreauthIntegrityNegotiateContext(mockConfig, new int[5], null);

        Map<String, String> andParams = new HashMap<>();
        andParams.put("q", query.replace("OR", "AND"));
        String andResponse = checkMethodBase(new HashMap<>()).params(andParams).get("/api/v1/documents").asString();
        int andRecordCount = JsonPath.from(andResponse).getInt("record_count");

        Map<String, String> params = new HashMap<>();
        params.put("q", query);

        NtlmContext context = new NtlmContext(mockAuth, true);
        context.initSecContext(new byte[0], 0, 0); // state -> 2

        byte[] serverChallenge = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08 };
        int type2Flags = NtlmFlags.NTLMSSP_NEGOTIATE_NTLM | NtlmFlags.NTLMSSP_REQUEST_TARGET;
        Type2Message type2Message = new Type2Message(type2Flags, serverChallenge, domain);

            // Given: SAM array with one entry
            samr.SamrSamArray array = new samr.SamrSamArray();
            array.count = 1;
            samr.SamrSamEntry entry = new samr.SamrSamEntry();
            entry.name = new rpc.unicode_string();
            array.entries = new samr.SamrSamEntry[] { entry };

            // When: Encoding array
            array.encode(mockNdrBuffer);

        verifyTokenKeep(this::asIndexHtml);

        final String diagnosticId = "log" + new SimpleDateFormat("yyyyMMddHHmm").format(ComponentUtil.getSystemHelper().getCurrentTime());
        return asStream(diagnosticId + ".zip").contentTypeOctetStream().stream(out -> {
            try (ZipOutputStream zos = new ZipOutputStream(out.stream())) {
                writeLogFiles(zos, diagnosticId);

    public void setUp() {
        MockitoAnnotations.openMocks(this);
        reconnector = new HandleReconnector(mockManager, 2, 50); // 2 retries, 50ms delay

        HandleGuid guid = new HandleGuid();
        byte[] fileId = new byte[16];
        for (int i = 0; i < 16; i++) {
            fileId[i] = (byte) (i + 1);
        }

        testHandle = new HandleInfo("/test/file.txt", guid, fileId, HandleType.DURABLE_V2, 120000, null);
    }