Add more tests for chat messages