Add tests for storage connection handling