Add message model tests