Add unit tests for message module