more

Author: gssbzn

internal/cli/backup/exports/buckets/create.go

@@ -29,13 +29,19 @@ import (
 	atlasv2 "go.mongodb.org/atlas-sdk/v20250312002/admin"
 )
 
+//go:generate mockgen -typed -destination=create_mock_test.go -package=buckets . ExportBucketsCreator
+
+type ExportBucketsCreator interface {
+	CreateExportBucket(string, *atlasv2.DiskBackupSnapshotExportBucketRequest) (*atlasv2.DiskBackupSnapshotExportBucketResponse, error)
+}
+
 type CreateOpts struct {
 	cli.ProjectOpts
 	cli.OutputOpts
 	iamRoleID     string
 	bucketName    string
 	cloudProvider string
-	store         store.ExportBucketsCreator
+	store         ExportBucketsCreator
 }
 
 func (opts *CreateOpts) initStore(ctx context.Context) func() error {

internal/cli/backup/exports/buckets/create_mock_test.go

@@ -19,14 +19,13 @@ package buckets
 import (
 	"testing"
 
-	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/mocks"
 	atlasv2 "go.mongodb.org/atlas-sdk/v20250312002/admin"
 	"go.uber.org/mock/gomock"
 )
 
 func TestCreateOpts_Run(t *testing.T) {
 	ctrl := gomock.NewController(t)
-	mockStore := mocks.NewMockExportBucketsCreator(ctrl)
+	mockStore := NewMockExportBucketsCreator(ctrl)
 
 	expected := &atlasv2.DiskBackupSnapshotExportBucketResponse{}
 

internal/cli/backup/exports/buckets/create_test.go

@@ -27,11 +27,17 @@ import (
 	"github.com/spf13/cobra"
 )
 
+//go:generate mockgen -typed -destination=delete_mock_test.go -package=buckets . ExportBucketsDeleter
+
+type ExportBucketsDeleter interface {
+	DeleteExportBucket(string, string) error
+}
+
 type DeleteOpts struct {
 	cli.ProjectOpts
 	*cli.DeleteOpts
 	bucketID string
-	store    store.ExportBucketsDeleter
+	store    ExportBucketsDeleter
 }
 
 func (opts *DeleteOpts) initStore(ctx context.Context) func() error {

internal/cli/backup/exports/buckets/delete.go

@@ -20,13 +20,12 @@ import (
 	"testing"
 
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/cli"
-	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/mocks"
 	"go.uber.org/mock/gomock"
 )
 
 func TestDelete_Run(t *testing.T) {
 	ctrl := gomock.NewController(t)
-	mockStore := mocks.NewMockExportBucketsDeleter(ctrl)
+	mockStore := NewMockExportBucketsDeleter(ctrl)
 
 	deleteOpts := &DeleteOpts{
 		DeleteOpts: &cli.DeleteOpts{

internal/cli/backup/exports/buckets/delete_mock_test.go

@@ -25,17 +25,24 @@ import (
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/store"
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/usage"
 	"github.com/spf13/cobra"
+	atlasv2 "go.mongodb.org/atlas-sdk/v20250312002/admin"
 )
 
 var describeTemplate = `ID	BUCKET NAME	CLOUD PROVIDER	IAM ROLE ID
 {{.Id}}	{{.BucketName}}	{{.CloudProvider}}	{{.IamRoleId}}
 `
 
+//go:generate mockgen -typed -destination=describe_mock_test.go -package=buckets . ExportBucketsDescriber
+
+type ExportBucketsDescriber interface {
+	DescribeExportBucket(string, string) (*atlasv2.DiskBackupSnapshotExportBucketResponse, error)
+}
+
 type DescribeOpts struct {
 	cli.ProjectOpts
 	cli.OutputOpts
 	bucketID string
-	store    store.ExportBucketsDescriber
+	store    ExportBucketsDescriber
 }
 
 func (opts *DescribeOpts) initStore(ctx context.Context) func() error {

internal/cli/backup/exports/buckets/delete_test.go

@@ -19,15 +19,14 @@ package buckets
 import (
 	"testing"
 
-	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/mocks"
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/test"
 	atlasv2 "go.mongodb.org/atlas-sdk/v20250312002/admin"
 	"go.uber.org/mock/gomock"
 )
 
 func TestDescribe_Run(t *testing.T) {
 	ctrl := gomock.NewController(t)
-	mockStore := mocks.NewMockExportBucketsDescriber(ctrl)
+	mockStore := NewMockExportBucketsDescriber(ctrl)
 
 	var expected atlasv2.DiskBackupSnapshotExportBucketResponse
 

internal/cli/backup/exports/buckets/describe.go

@@ -24,13 +24,20 @@ import (
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/store"
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/usage"
 	"github.com/spf13/cobra"
+	atlasv2 "go.mongodb.org/atlas-sdk/v20250312002/admin"
 )
 
+//go:generate mockgen -typed -destination=list_mock_test.go -package=buckets . ExportBucketsLister
+
+type ExportBucketsLister interface {
+	ExportBuckets(string, *store.ListOptions) (*atlasv2.PaginatedBackupSnapshotExportBuckets, error)
+}
+
 type ListOpts struct {
 	cli.ProjectOpts
 	cli.OutputOpts
 	cli.ListOpts
-	store store.ExportBucketsLister
+	store ExportBucketsLister
 }
 
 func (opts *ListOpts) initStore(ctx context.Context) func() error {