assertEquals(1, session.getMessageCount());

        session.addAssistantMessage("Second");
        assertEquals(2, session.getMessageCount());
    }

    @Test
    public void test_getMessageCountWithNullMessages() {
        final ChatSession session = new ChatSession();
        session.setMessages(null);
        assertEquals(0, session.getMessageCount());
    }

    @Test