public class DocumentHelperTest extends UnitFessTestCase {

    @Override
    protected void setUp(TestInfo testInfo) throws Exception {
        super.setUp(testInfo);
    }

    @Test
    public void test_getContent() {
        DocumentHelper documentHelper = new DocumentHelper();

        ResponseData responseData = new ResponseData();
        Map<String, Object> dataMap = new HashMap<>();

        assertEquals("(memory)", source.getLocation());

        source = new StringSource("Hello World!", "LOCATION");
        assertEquals("LOCATION", source.getLocation());
    }

    @Test
    void testGetContent() {
        StringSource source = new StringSource(null);
        assertEquals("", source.getContent());

        source = new StringSource("Hello World!", "LOCATION");

        assertEquals(Integer.valueOf(150), response.getTotalTokens());
        assertEquals("gpt-4", response.getModel());
    }

    @Test
    public void test_setContent() {
        final LlmChatResponse response = new LlmChatResponse();
        response.setContent("Test content");
        assertEquals("Test content", response.getContent());