Unified/Blob: Allow direct access to UIDs (#101339)

pkg/storage/unified/sql/test/integration_test.go

@@ -31,39 +31,43 @@ func TestMain(m *testing.M) {
 
 // TestStorageBackend is a test for the StorageBackend interface.
 func TestIntegrationSQLStorageBackend(t *testing.T) {
-	unitest.RunStorageBackendTest(t, func(ctx context.Context) resource.StorageBackend {
-		dbstore := infraDB.InitTestDB(t)
-		eDB, err := dbimpl.ProvideResourceDB(dbstore, setting.NewCfg(), nil)
-		require.NoError(t, err)
-		require.NotNil(t, eDB)
+	t.Run("IsHA (polling notifier)", func(t *testing.T) {
+		unitest.RunStorageBackendTest(t, func(ctx context.Context) resource.StorageBackend {
+			dbstore := infraDB.InitTestDB(t)
+			eDB, err := dbimpl.ProvideResourceDB(dbstore, setting.NewCfg(), nil)
+			require.NoError(t, err)
+			require.NotNil(t, eDB)
 
-		backend, err := sql.NewBackend(sql.BackendOptions{
-			DBProvider: eDB,
-			IsHA:       true,
-		})
-		require.NoError(t, err)
-		require.NotNil(t, backend)
-		err = backend.Init(testutil.NewDefaultTestContext(t))
-		require.NoError(t, err)
-		return backend
-	}, nil)
-	// Run single instance tests with in-process notifier.
-	unitest.RunStorageBackendTest(t, func(ctx context.Context) resource.StorageBackend {
-		dbstore := infraDB.InitTestDB(t)
-		eDB, err := dbimpl.ProvideResourceDB(dbstore, setting.NewCfg(), nil)
-		require.NoError(t, err)
-		require.NotNil(t, eDB)
+			backend, err := sql.NewBackend(sql.BackendOptions{
+				DBProvider: eDB,
+				IsHA:       true,
+			})
+			require.NoError(t, err)
+			require.NotNil(t, backend)
+			err = backend.Init(testutil.NewDefaultTestContext(t))
+			require.NoError(t, err)
+			return backend
+		}, nil)
+	})
 
-		backend, err := sql.NewBackend(sql.BackendOptions{
-			DBProvider: eDB,
-			IsHA:       false,
-		})
-		require.NoError(t, err)
-		require.NotNil(t, backend)
-		err = backend.Init(testutil.NewDefaultTestContext(t))
-		require.NoError(t, err)
-		return backend
-	}, nil)
+	t.Run("NotHA (in process notifier)", func(t *testing.T) {
+		unitest.RunStorageBackendTest(t, func(ctx context.Context) resource.StorageBackend {
+			dbstore := infraDB.InitTestDB(t)
+			eDB, err := dbimpl.ProvideResourceDB(dbstore, setting.NewCfg(), nil)
+			require.NoError(t, err)
+			require.NotNil(t, eDB)
+
+			backend, err := sql.NewBackend(sql.BackendOptions{
+				DBProvider: eDB,
+				IsHA:       false,
+			})
+			require.NoError(t, err)
+			require.NotNil(t, backend)
+			err = backend.Init(testutil.NewDefaultTestContext(t))
+			require.NoError(t, err)
+			return backend
+		}, nil)
+	})
 }
 
 func TestClientServer(t *testing.T) {