Add tests for message