Add unit tests for product