Add unit tests for Store