Add unit test for store class