Add some tests for storage

Test Plan:
- Run tests