added new model tests and added mock as a test dependency