Add a test case for UnifiedDataStore