revert

bucketeer-io · Mar 6, 2025 · fbd1988 · fbd1988
1 parent eb73d46
commit fbd1988
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/environment/storage/v2/environment_test.go b/pkg/environment/storage/v2/environment_test.go
@@ -32,7 +32,7 @@ func TestNewEnvironmentStorage(t *testing.T) {
 	t.Parallel()
 	mockController := gomock.NewController(t)
 	defer mockController.Finish()
-	storage := NewEnvironmentStorage(mock.NewMockClient(mockController))
+	storage := NewEnvironmentStorage(mock.NewMockQueryExecer(mockController))
 	assert.IsType(t, &environmentStorage{}, storage)
 }
 

diff --git a/pkg/environment/storage/v2/project_test.go b/pkg/environment/storage/v2/project_test.go
@@ -32,7 +32,7 @@ func TestNewProjectStorage(t *testing.T) {
 	t.Parallel()
 	mockController := gomock.NewController(t)
 	defer mockController.Finish()
-	storage := NewProjectStorage(mock.NewMockClient(mockController))
+	storage := NewProjectStorage(mock.NewMockQueryExecer(mockController))
 	assert.IsType(t, &projectStorage{}, storage)
 }