final ChatSession session = new ChatSession();
        assertEquals(0, session.getMessageCount());

        session.addUserMessage("First");
        assertEquals(1, session.getMessageCount());

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

    @Test
    public void test_getMessageCountWithNullMessages() {