Fix mocks

Author: jakubno

packages/orchestrator/cmd/inspect-data/main.go

@@ -20,6 +20,7 @@ func main() {
 	flag.Parse()
 
 	template := storage.NewTemplateFiles(
+		storage.BucketStorage,
 		"",
 		*buildId,
 		"",

packages/orchestrator/cmd/inspect-header/main.go

@@ -19,6 +19,7 @@ func main() {
 	flag.Parse()
 
 	template := storage.NewTemplateFiles(
+		storage.BucketStorage,
 		"",
 		*buildId,
 		"",

packages/orchestrator/cmd/mock-snapshot/mock.go

@@ -187,6 +187,7 @@ func mockSnapshot(
 	snapshotTime := time.Now()
 
 	snapshotTemplateFiles, err := storage.NewTemplateFiles(
+		storage.BucketStorage,
 		"snapshot-template",
 		"f0370054-b669-eee4-b33b-573d5287c6ef",
 		sbx.Config.KernelVersion,

packages/orchestrator/cmd/simulate-headers-merge/main.go

@@ -7,10 +7,11 @@ import (
 	"log"
 	"os"
 
+	"github.com/google/uuid"
+
 	"github.com/e2b-dev/infra/packages/shared/pkg/storage"
 	"github.com/e2b-dev/infra/packages/shared/pkg/storage/gcs"
 	"github.com/e2b-dev/infra/packages/shared/pkg/storage/header"
-	"github.com/google/uuid"
 )
 
 func main() {
@@ -22,6 +23,7 @@ func main() {
 	flag.Parse()
 
 	baseTemplate := storage.NewTemplateFiles(
+		storage.BucketStorage,
 		"",
 		*baseBuildId,
 		"",
@@ -30,6 +32,7 @@ func main() {
 	)
 
 	diffTemplate := storage.NewTemplateFiles(
+		storage.BucketStorage,
 		"",
 		*diffBuildId,
 		"",