CLOUDP-60875: List host databases, OM (#105)

Author: gssbzn

Makefile

@@ -74,6 +74,8 @@ gen-mocks: ## Generate mocks
 	mockgen -source=internal/store/processes.go -destination=internal/mocks/mock_processes.go -package=mocks
 	mockgen -source=internal/store/logs.go -destination=internal/mocks/mock_logs.go -package=mocks
 	mockgen -source=internal/store/hosts.go -destination=internal/mocks/mock_hosts.go -package=mocks
+	mockgen -source=internal/store/host_databases.go -destination=internal/mocks/mock_host_databases.go -package=mocks
+	mockgen -source=internal/store/host_disks.go -destination=internal/mocks/mock_host_disks.go -package=mocks
 
 .PHONY: build
 build: ## Generate a binary in ./bin

internal/cli/atlas_measurements_databases_list.go

@@ -52,11 +52,11 @@ func (opts *atlasMeasurementsDatabasesListsOpts) Run() error {
 	return json.PrettyPrint(result)
 }
 
-// mongocli atlas measurements process(es) disks lists host:port
+// mongocli atlas measurements process(es) disks lists [host:port]
 func AtlasMeasurementsDatabasesListBuilder() *cobra.Command {
 	opts := &atlasMeasurementsDatabasesListsOpts{}
 	cmd := &cobra.Command{
-		Use:     "list",
+		Use:     "list [host:port]",
 		Short:   description.ListDatabases,
 		Aliases: []string{"ls"},
 		Args:    cobra.ExactArgs(1),

internal/cli/ops_manager_measurements.go

@@ -27,5 +27,7 @@ func OpsManagerMeasurementsBuilder() *cobra.Command {
 
 	cmd.AddCommand(OpsManagerMeasurementsProcessBuilder())
 	cmd.AddCommand(OpsManagerMeasurementsDisksBuilder())
+	cmd.AddCommand(OpsManagerMeasurementsDatabasesBuilder())
+
 	return cmd
 }

internal/cli/ops_manager_measurements_databases.go

@@ -0,0 +1,32 @@
+// Copyright 2020 MongoDB Inc
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package cli
+
+import (
+	"github.com/mongodb/mongocli/internal/description"
+	"github.com/spf13/cobra"
+)
+
+func OpsManagerMeasurementsDatabasesBuilder() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:     "databases",
+		Aliases: []string{"database"},
+		Short:   description.Databases,
+	}
+
+	cmd.AddCommand(OpsManagerMeasurementsDatabasesListBuilder())
+
+	return cmd
+}

internal/cli/ops_manager_measurements_databases_list.go

@@ -0,0 +1,77 @@
+// Copyright 2020 MongoDB Inc
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package cli
+
+import (
+	"github.com/mongodb/mongocli/internal/description"
+	"github.com/mongodb/mongocli/internal/flags"
+	"github.com/mongodb/mongocli/internal/json"
+	"github.com/mongodb/mongocli/internal/store"
+	"github.com/mongodb/mongocli/internal/usage"
+	"github.com/spf13/cobra"
+)
+
+type opsManagerMeasurementsDatabasesListsOpts struct {
+	globalOpts
+	listOpts
+	hostID string
+	store  store.HostDatabaseLister
+}
+
+func (opts *opsManagerMeasurementsDatabasesListsOpts) init() error {
+	if opts.ProjectID() == "" {
+		return errMissingProjectID
+	}
+
+	var err error
+	opts.store, err = store.New()
+	return err
+}
+
+func (opts *opsManagerMeasurementsDatabasesListsOpts) Run() error {
+	listOpts := opts.newListOptions()
+	result, err := opts.store.HostDatabases(opts.ProjectID(), opts.hostID, listOpts)
+
+	if err != nil {
+		return err
+	}
+
+	return json.PrettyPrint(result)
+}
+
+// mongocli om measurements process(es) disks lists [hostID]
+func OpsManagerMeasurementsDatabasesListBuilder() *cobra.Command {
+	opts := &opsManagerMeasurementsDatabasesListsOpts{}
+	cmd := &cobra.Command{
+		Use:     "list [hostID]",
+		Short:   description.ListDatabases,
+		Aliases: []string{"ls"},
+		Args:    cobra.ExactArgs(1),
+		PreRunE: func(cmd *cobra.Command, args []string) error {
+			return opts.init()
+		},
+		RunE: func(cmd *cobra.Command, args []string) error {
+			opts.hostID = args[0]
+
+			return opts.Run()
+		},
+	}
+
+	cmd.Flags().IntVar(&opts.pageNum, flags.Page, 0, usage.Page)
+	cmd.Flags().IntVar(&opts.itemsPerPage, flags.Limit, 0, usage.Limit)
+	cmd.Flags().StringVar(&opts.projectID, flags.ProjectID, "", usage.ProjectID)
+
+	return cmd
+}

internal/cli/ops_manager_measurements_databases_list_test.go

@@ -0,0 +1,46 @@
+// Copyright 2020 MongoDB Inc
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+package cli
+
+import (
+	"testing"
+
+	"github.com/golang/mock/gomock"
+	"github.com/mongodb/mongocli/internal/fixtures"
+	"github.com/mongodb/mongocli/internal/mocks"
+)
+
+func TestOpsManagerMeasurementsDatabasesListsOpts_Run(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	mockStore := mocks.NewMockHostDatabaseLister(ctrl)
+
+	defer ctrl.Finish()
+
+	expected := fixtures.ProcessDatabases()
+
+	listOpts := &opsManagerMeasurementsDatabasesListsOpts{
+		hostID: "1",
+		store:  mockStore,
+	}
+
+	opts := listOpts.newListOptions()
+	mockStore.
+		EXPECT().HostDatabases(listOpts.projectID, listOpts.hostID, opts).
+		Return(expected, nil).
+		Times(1)
+
+	if err := listOpts.Run(); err != nil {
+		t.Fatalf("Run() unexpected error: %v", err)
+	}
+}

internal/mocks/mock_host_databases.go

@@ -0,0 +1,39 @@
+// Copyright 2020 MongoDB Inc
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package store
+
+import (
+	"context"
+	"fmt"
+
+	atlas "github.com/mongodb/go-client-mongodb-atlas/mongodbatlas"
+	om "github.com/mongodb/go-client-mongodb-ops-manager/opsmngr"
+	"github.com/mongodb/mongocli/internal/config"
+)
+
+type HostDatabaseLister interface {
+	HostDatabases(string, string, *atlas.ListOptions) (*atlas.ProcessDatabasesResponse, error)
+}
+
+// HostDatabases encapsulate the logic to manage different cloud providers
+func (s *Store) HostDatabases(groupID, hostID string, opts *atlas.ListOptions) (*atlas.ProcessDatabasesResponse, error) {
+	switch s.service {
+	case config.OpsManagerService, config.CloudManagerService:
+		result, _, err := s.client.(*om.Client).HostDatabases.List(context.Background(), groupID, hostID, opts)
+		return result, err
+	default:
+		return nil, fmt.Errorf("unsupported service: %s", s.service)
+	}
+}