}

        if (url) {
            var ext = url.split('.').pop().toLowerCase().split('?')[0];
            switch (ext) {
                case 'pdf': return 'fa-file-pdf-o';
                case 'doc': case 'docx': return 'fa-file-word-o';
                case 'xls': case 'xlsx': return 'fa-file-excel-o';
                case 'ppt': case 'pptx': return 'fa-file-powerpoint-o';

            // Expected
        }
    }

    @Test
    public void test_withRelevantDocs_manyDocs() {
        List<String> docIds = Arrays.asList("doc1", "doc2", "doc3", "doc4", "doc5", "doc6", "doc7", "doc8", "doc9", "doc10");
        List<Integer> indexes = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9, 10);
        RelevanceEvaluationResult result = RelevanceEvaluationResult.withRelevantDocs(docIds, indexes);

    @Test
    public void test_searchWithQuery_returnsConfiguredResults() {
        final List<Map<String, Object>> docs = new ArrayList<>();
        final Map<String, Object> doc = new HashMap<>();
        doc.put("title", "Test Doc");
        doc.put("doc_id", "doc1");
        docs.add(doc);
        chatClient.setSearchResults(docs);
        chatClient.resetSearchDocumentsCalled();

        Map<String, Object> doc1 = new HashMap<>();
        doc1.put("title", "Test Document 1");
        doc1.put("content", "Test content 1");
        documentItems.add(doc1);

        Map<String, Object> doc2 = new HashMap<>();
        doc2.put("title", "Test Document 2");
        doc2.put("content", "Test content 2");
        documentItems.add(doc2);

        searchRenderData.setDocumentItems(documentItems);

        final Map<String, Object> doc1 = new HashMap<>();
        doc1.put("_id", "doc1");
        doc1.put("title", "Document 1");

        final Map<String, Object> doc2 = new HashMap<>();
        doc2.put("_id", "doc2");
        doc2.put("title", "Document 2");

        final SearchResult result = SearchResult.create().addDocument(doc1).addDocument(doc2).allRecordCount(2).build();

        assertNotNull(result);

Shinsuke Sugaya <******@****.***> 1773561175 +0900

Shinsuke Sugaya <******@****.***> 1773561175 +0900

Shinsuke Sugaya <******@****.***> 1773561175 +0900

        final Map<String, Object> doc1 = new HashMap<>();
        doc1.put("title", "Doc 1");
        doc1.put("content", "<h1>Title</h1><p>Content 1</p>");
        documents.add(doc1);

        final Map<String, Object> doc2 = new HashMap<>();
        doc2.put("title", "Doc 2");
        doc2.put("content", "<ul><li>Item A</li><li>Item B</li></ul>");
        documents.add(doc2);

        /**
         * The document object.
         */
        protected Object doc;

        /**
         * Sets the document object.
         * @param doc The document object.
         * @return The ApiDocResponse instance.
         */
        public ApiDocResponse doc(final Object doc) {
            this.doc = doc;
            return this;
        }

        @Override