Add unit tests for model validations