Add unit test to product model