Add tests for session storage