Add tests for TextMessage