add unit test for model