moar

Author: gssbzn

internal/cli/clusters/indexes/create.go

@@ -31,6 +31,12 @@ import (
 	atlasv2 "go.mongodb.org/atlas-sdk/v20250312002/admin"
 )
 
+//go:generate mockgen -typed -destination=create_mock_test.go -package=indexes . IndexCreator
+
+type IndexCreator interface {
+	CreateIndex(string, string, *atlasv2.DatabaseRollingIndexRequest) error
+}
+
 type CreateOpts struct {
 	cli.ProjectOpts
 	clusterName string
@@ -40,7 +46,7 @@ type CreateOpts struct {
 	filename    string
 	keys        []string
 	sparse      bool
-	store       store.IndexCreator
+	store       IndexCreator
 	fs          afero.Fs
 }
 

internal/mocks/mock_indexes.go renamed to internal/cli/clusters/indexes/create_mock_test.go

@@ -19,14 +19,13 @@ package indexes
 import (
 	"testing"
 
-	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/mocks"
 	"github.com/spf13/afero"
 	"go.uber.org/mock/gomock"
 )
 
 func TestCreate_Run(t *testing.T) {
 	ctrl := gomock.NewController(t)
-	mockStore := mocks.NewMockIndexCreator(ctrl)
+	mockStore := NewMockIndexCreator(ctrl)
 
 	createOpts := &CreateOpts{
 		name:        "ProjectBar",
@@ -51,7 +50,7 @@ func TestCreate_Run(t *testing.T) {
 
 func TestCreateWithFile_Run(t *testing.T) {
 	ctrl := gomock.NewController(t)
-	mockStore := mocks.NewMockIndexCreator(ctrl)
+	mockStore := NewMockIndexCreator(ctrl)
 	appFS := afero.NewMemMapFs()
 	fileJSON := `
 {

internal/cli/clusters/indexes/create_test.go

@@ -31,10 +31,16 @@ const describeTemplate = `ID	IDENTITY PROVIDER ID	IDENTITY PROVIDER STATUS
 {{.Id}}	{{.IdentityProviderId}}	{{.IdentityProviderStatus}}
 `
 
+//go:generate mockgen -typed -destination=describe_mock_test.go -package=federationsettings . Describer
+
+type Describer interface {
+	FederationSetting(opts *atlasv2.GetFederationSettingsApiParams) (*atlasv2.OrgFederationSettings, error)
+}
+
 type DescribeOpts struct {
 	cli.OrgOpts
 	cli.OutputOpts
-	store store.FederationSettingsDescriber
+	store Describer
 }
 
 func (opts *DescribeOpts) initStore(ctx context.Context) func() error {

internal/cli/federatedauthentication/federationsettings/describe.go

@@ -21,15 +21,14 @@ import (
 	"testing"
 
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/cli"
-	"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.NewMockFederationSettingsDescriber(ctrl)
+	mockStore := NewMockDescriber(ctrl)
 
 	buf := new(bytes.Buffer)
 	describeOpts := &DescribeOpts{

internal/cli/federatedauthentication/federationsettings/describe_mock_test.go

@@ -24,15 +24,22 @@ 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=databases . ProcessDatabaseLister
+
+type ProcessDatabaseLister interface {
+	ProcessDatabases(string, string, int, *store.ListOptions) (*atlasv2.PaginatedDatabase, error)
+}
+
 type ListsOpts struct {
 	cli.ProjectOpts
 	cli.OutputOpts
 	cli.ListOpts
 	host  string
 	port  int
-	store store.ProcessDatabaseLister
+	store ProcessDatabaseLister
 }
 
 func (opts *ListsOpts) initStore(ctx context.Context) func() error {

internal/cli/federatedauthentication/federationsettings/describe_test.go

@@ -19,7 +19,6 @@ package databases
 import (
 	"testing"
 
-	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/mocks"
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/pointer"
 	"github.com/mongodb/mongodb-atlas-cli/atlascli/internal/test"
 	atlasv2 "go.mongodb.org/atlas-sdk/v20250312002/admin"
@@ -28,7 +27,7 @@ import (
 
 func TestDatabasesListsOpts_Run(t *testing.T) {
 	ctrl := gomock.NewController(t)
-	mockStore := mocks.NewMockProcessDatabaseLister(ctrl)
+	mockStore := NewMockProcessDatabaseLister(ctrl)
 
 	expected := &atlasv2.PaginatedDatabase{
 		Results: &[]atlasv2.MesurementsDatabase{

internal/cli/metrics/databases/list.go

@@ -30,14 +30,20 @@ import (
 
 const createTemplate = "Created new access list entry(s).\n"
 
+//go:generate mockgen -typed -destination=create_mock_test.go -package=accesslists . OrganizationAPIKeyAccessListCreator
+
+type OrganizationAPIKeyAccessListCreator interface {
+	CreateOrganizationAPIKeyAccessList(*admin.CreateApiKeyAccessListApiParams) (*admin.PaginatedApiUserAccessListResponse, error)
+}
+
 type CreateOpts struct {
 	cli.OrgOpts
 	cli.OutputOpts
 	apyKey    string
 	ips       []string
 	cidrs     []string
 	currentIP bool
-	store     store.OrganizationAPIKeyAccessListCreator
+	store     OrganizationAPIKeyAccessListCreator
 }
 
 func (opts *CreateOpts) initStore(ctx context.Context) func() error {