complete unit tests for the model