@Test
    public void test_update() {
        loadTestData();

        StopwordsItem item = stopwordsFile.get(1).get();
        item.setNewInput("updated");
        stopwordsFile.update(item);

        // Verify the item was updated
        OptionalEntity<StopwordsItem> result = stopwordsFile.get(1);
        assertTrue(result.isPresent());
        assertEquals("updated", result.get().getInput());
    }

        assertTrue(result.isHasRelevantResults());
        assertEquals("doc-with-dash", result.getRelevantDocIds().get(0));
        assertEquals("doc_with_underscore", result.getRelevantDocIds().get(1));
        assertEquals("doc.with.dot", result.getRelevantDocIds().get(2));
    }

    @Test
    public void test_toString() {
        List<String> docIds = Arrays.asList("doc1", "doc2");

        final PagingList<KuromojiItem> itemList2 = kuromojiFile.selectList(0,
                20);
        assertEquals(2, itemList2.size());

        final KuromojiItem kuromojiItem2 = itemList2.get(1);
        kuromojiFile.delete(kuromojiItem2);
        final PagingList<KuromojiItem> itemList3 = kuromojiFile.selectList(0,
                20);
        assertEquals(1, itemList3.size());

        assertEquals(4, history.size());
        assertEquals("user", history.get(0).getRole());
        assertEquals("Q1", history.get(0).getContent());
        assertEquals("assistant", history.get(1).getRole());
        assertEquals("A1", history.get(1).getContent());
        assertEquals("user", history.get(2).getRole());
        assertEquals("Q2", history.get(2).getContent());
        assertEquals("assistant", history.get(3).getRole());

        assertEquals("BoolQueryBuilder", queryBuilderList.get(1).getClass().getSimpleName());
        final List<FieldSortBuilder> fieldSortBuilderList = cb.query().getFieldSortBuilderList();
        assertEquals(2, fieldSortBuilderList.size());
        assertEquals("sortOrder", fieldSortBuilderList.get(0).getFieldName());
        assertEquals("name", fieldSortBuilderList.get(1).getFieldName());
    }

    @Test

        // Get an existing item
        ProtwordsItem item = protwordsFile.get(1).get();
        item.setNewInput("updatedWord");

        // Update the item
        protwordsFile.update(item);

        // Verify the update
        protwordsFile.reload(null);
        ProtwordsItem updatedItem = protwordsFile.get(1).get();
        assertEquals("updatedWord", updatedItem.getInput());
    }

    @Test

        qrList.calculatePageInfo();
        pnList = qrList.getPageNumberList();
        assertEquals(4, pnList.size());
        assertEquals("1", pnList.get(0));
        assertEquals("2", pnList.get(1));
        assertEquals("3", pnList.get(2));
        assertEquals("4", pnList.get(3));

        qrList = new QueryResponseList(null, 0, 20, 0) {
            @Override
            public int size() {

        // Test onPhaseComplete
        callback.onPhaseComplete(ChatPhaseCallback.PHASE_INTENT);
        assertEquals(2, events.size());
        assertEquals("complete:intent", events.get(1));

        // Test onChunk
        callback.onChunk("test content", false);
        assertEquals(3, events.size());
        assertEquals("chunk:test content:false", events.get(2));

        assertTrue(queryContext.hasSorts());
        List<SortBuilder<?>> sorts = queryContext.sortBuilders();
        assertEquals(2, sorts.size());
        assertEquals(sort1, sorts.get(0));
        assertEquals(sort2, sorts.get(1));
    }

    // Test addSorts with no sorts
    @Test
    public void test_addSorts_empty() {
        queryContext = new QueryContext("test", false);
        queryContext.addSorts();

        assertEquals(ChatMessage.ROLE_USER, messages.get(0).getRole());
        assertEquals("User 2", messages.get(0).getContent());
        assertEquals(ChatMessage.ROLE_ASSISTANT, messages.get(1).getRole());
        assertEquals("Assistant 2", messages.get(1).getContent());
        assertEquals(ChatMessage.ROLE_USER, messages.get(2).getRole());
        assertEquals("User 3", messages.get(2).getContent());