implement unit tests with mock store