import org.junit.jupiter.api.Test;

public class RelevanceEvaluationResultTest extends UnitFessTestCase {

    @Test
    public void test_withRelevantDocs() {
        List<String> docIds = Arrays.asList("doc1", "doc2", "doc3");
        List<Integer> indexes = Arrays.asList(1, 2, 3);
        RelevanceEvaluationResult result = RelevanceEvaluationResult.withRelevantDocs(docIds, indexes);

        assertTrue(result.isHasRelevantResults());

        doc1.put("title", "Doc1");
        Map<String, Object> doc2 = new HashMap<>();
        doc2.put("title", "Doc2");
        newDocs.add(doc1);
        newDocs.add(doc2);

        assertTrue(qrList.addAll(newDocs));
        assertEquals(2, qrList.size());
        assertEquals(doc1, qrList.get(0));
        assertEquals(doc2, qrList.get(1));

        List<Map<String, Object>> moreDocs = new ArrayList<>();

        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);
        assertEquals(2, result.getDocumentList().size());

        doc1.put("title", "Test Document 1");

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

        docList.add(doc1);
        docList.add(doc2);

        assertEquals(2, docList.size());
        assertEquals(doc1, docList.get(0));
        assertEquals(doc2, docList.get(1));

        docList.addContentSize(500);

                doc1.put("title", "Document 1");
                callback.store(initParamMap, doc1);

                Map<String, Object> doc2 = new HashMap<>();
                doc2.put("id", "2");
                doc2.put("title", "Document 2");
                callback.store(initParamMap, doc2);

                // Commit the changes
                callback.commit();
            }

            @Override

        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);
        assertEquals(documentItems, searchRenderData.getDocumentItems());

        docs.add(doc1);

        final Map<String, Object> doc2 = new LinkedHashMap<>();
        doc2.put("url", "https://other.com/docs/guide");
        doc2.put("title", "Guide");
        doc2.put("content", "Guide content");
        doc2.put("lang", "ja");
        doc2.put("anchor", Arrays.asList("https://a.com", "https://b.com"));
        docs.add(doc2);

        setupMockComponents(docs);

        doc1.put("url", "http://example.com/install");
        msg.addSource(new ChatSource(1, doc1));
        final Map<String, Object> doc2 = new HashMap<>();
        doc2.put("title", "Quick Start");
        doc2.put("url", "http://example.com/start");
        msg.addSource(new ChatSource(2, doc2));

        final String result = chatClient.testBuildAssistantHistoryContent(msg, "source_titles", 500, 500);

        try (RankFusionProcessor processor = new RankFusionProcessor()) {
            processor.setSearcher(new TestSearcherWithFixedIds("doc1", "doc2", "doc3"));
            processor.register(new TestSearcherWithFixedIds("doc2", "doc3", "doc4")); // Some duplicates
            processor.init();

        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);

        final String result = client.testBuildContext(documents);
        assertTrue(result.contains("TitleContent 1"));