Merge pull request #19 from Vonage/feat/mongo-new

MongoDB support

Author: dacheng293

pkg/api/deployment.go

@@ -401,3 +401,100 @@ func (c *DeploymentClient) RemoveSecret(ctx context.Context, name string) error
 	}
 	return nil
 }
+
+type pluginsRequest struct {
+	Plugin  string                 `json:"plugin"`
+	Version string                 `json:"version"`
+	Action  string                 `json:"action"`
+	Options map[string]interface{} `json:"options"`
+}
+
+type MongoInfoResponse struct {
+	Username         string `json:"username"`
+	Password         string `json:"password"`
+	Database         string `json:"database"`
+	ConnectionString string `json:"connectionString"`
+}
+
+const pluginsPath = "/plugins"
+
+func (c *DeploymentClient) CreateMongoDatabase(ctx context.Context, version string) (MongoInfoResponse, error) {
+	var result MongoInfoResponse
+
+	resp, err := c.httpClient.R().
+		SetContext(ctx).
+		SetResult(&result).
+		SetBody(pluginsRequest{Plugin: "mongo", Version: version, Action: "createDB"}).
+		Post(c.baseURL + pluginsPath)
+	if err != nil {
+		return MongoInfoResponse{}, fmt.Errorf("%w: trace_id = %s", err, traceIDFromHTTPResponse(resp))
+	}
+	if resp.StatusCode() == http.StatusNotFound {
+		return MongoInfoResponse{}, ErrNotFound
+	}
+	if resp.IsError() {
+		return MongoInfoResponse{}, NewErrorFromHTTPResponse(resp)
+	}
+
+	return result, nil
+}
+
+func (c *DeploymentClient) DeleteMongoDatabase(ctx context.Context, version, database string) error {
+	resp, err := c.httpClient.R().
+		SetContext(ctx).
+		SetBody(pluginsRequest{Plugin: "mongo", Version: version, Action: "deleteDB", Options: map[string]interface{}{"database": database}}).
+		Post(c.baseURL + pluginsPath)
+	if err != nil {
+		return fmt.Errorf("%w: trace_id = %s", err, traceIDFromHTTPResponse(resp))
+	}
+	if resp.StatusCode() == http.StatusNotFound {
+		return ErrNotFound
+	}
+	if resp.IsError() {
+		return NewErrorFromHTTPResponse(resp)
+	}
+
+	return nil
+}
+
+func (c *DeploymentClient) GetMongoDatabase(ctx context.Context, version string, database string) (MongoInfoResponse, error) {
+	var result MongoInfoResponse
+
+	resp, err := c.httpClient.R().
+		SetContext(ctx).
+		SetResult(&result).
+		SetBody(pluginsRequest{Plugin: "mongo", Version: version, Action: "getDB", Options: map[string]interface{}{"database": database}}).
+		Post(c.baseURL + pluginsPath)
+	if err != nil {
+		return MongoInfoResponse{}, fmt.Errorf("%w: trace_id = %s", err, traceIDFromHTTPResponse(resp))
+	}
+	if resp.StatusCode() == http.StatusNotFound {
+		return MongoInfoResponse{}, ErrNotFound
+	}
+	if resp.IsError() {
+		return MongoInfoResponse{}, NewErrorFromHTTPResponse(resp)
+	}
+
+	return result, nil
+}
+
+func (c *DeploymentClient) ListMongoDatabases(ctx context.Context, version string) ([]string, error) {
+	var result []string
+
+	resp, err := c.httpClient.R().
+		SetContext(ctx).
+		SetResult(&result).
+		SetBody(pluginsRequest{Plugin: "mongo", Version: version, Action: "listDBs"}).
+		Post(c.baseURL + pluginsPath)
+	if err != nil {
+		return nil, fmt.Errorf("%w: trace_id = %s", err, traceIDFromHTTPResponse(resp))
+	}
+	if resp.StatusCode() == http.StatusNotFound {
+		return nil, ErrNotFound
+	}
+	if resp.IsError() {
+		return nil, NewErrorFromHTTPResponse(resp)
+	}
+
+	return result, nil
+}

pkg/cmdutil/factory.go

@@ -52,6 +52,10 @@ type DeploymentInterface interface {
 	CreateSecret(ctx context.Context, s config.Secret) error
 	UpdateSecret(ctx context.Context, s config.Secret) error
 	RemoveSecret(ctx context.Context, name string) error
+	CreateMongoDatabase(ctx context.Context, version string) (api.MongoInfoResponse, error)
+	DeleteMongoDatabase(ctx context.Context, version, database string) error
+	GetMongoDatabase(ctx context.Context, version, database string) (api.MongoInfoResponse, error)
+	ListMongoDatabases(ctx context.Context, version string) ([]string, error)
 }
 
 type DatastoreInterface interface {

testutil/mocks/factory.go

@@ -0,0 +1,65 @@
+package create
+
+import (
+	"context"
+	"fmt"
+	"vonage-cloud-runtime-cli/pkg/cmdutil"
+
+	"github.com/MakeNowJust/heredoc"
+	"github.com/spf13/cobra"
+)
+
+type Options struct {
+	cmdutil.Factory
+
+	Version string
+}
+
+func NewCmdMongoCreate(f cmdutil.Factory) *cobra.Command {
+
+	opts := Options{
+		Factory: f,
+	}
+
+	cmd := &cobra.Command{
+		Use:     "create",
+		Short:   "Create a database and user credentials",
+		Example: heredoc.Doc(`$ vcr mongo create`),
+		Args:    cobra.MaximumNArgs(0),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			ctx, cancel := context.WithDeadline(context.Background(), opts.Deadline())
+			defer cancel()
+
+			return runCreate(ctx, &opts)
+		},
+	}
+
+	cmd.Flags().StringVarP(&opts.Version, "version", "v", "v0.1", "API version (default is v0.1)")
+
+	return cmd
+}
+
+func runCreate(ctx context.Context, opts *Options) error {
+	io := opts.IOStreams()
+	c := opts.IOStreams().ColorScheme()
+
+	spinner := cmdutil.DisplaySpinnerMessageWithHandle(" Creating database")
+	result, err := opts.DeploymentClient().CreateMongoDatabase(ctx, opts.Version)
+	spinner.Stop()
+	if err != nil {
+		return fmt.Errorf("failed to create database: %w", err)
+	}
+	fmt.Fprintf(io.Out, heredoc.Doc(`
+						%s Database created:
+							username: %s
+							password: %s
+							database: %s
+							connectionString: %s
+						`),
+		c.SuccessIcon(),
+		result.Username,
+		result.Password,
+		result.Database,
+		result.ConnectionString)
+	return nil
+}

vcr/mongo/create/create.go

@@ -0,0 +1,117 @@
+package create
+
+import (
+	"bytes"
+	"errors"
+	"io"
+	"testing"
+	"vonage-cloud-runtime-cli/pkg/api"
+	"vonage-cloud-runtime-cli/testutil"
+	"vonage-cloud-runtime-cli/testutil/mocks"
+
+	"github.com/MakeNowJust/heredoc"
+	"github.com/cli/cli/v2/pkg/iostreams"
+	"github.com/golang/mock/gomock"
+	"github.com/google/shlex"
+	"github.com/stretchr/testify/require"
+)
+
+func TestMongoCreate(t *testing.T) {
+	type mock struct {
+		CreateTimes          int
+		CreateReturnResponse api.MongoInfoResponse
+		CreateReturnErr      error
+		CreateVersion        string
+	}
+	type want struct {
+		errMsg string
+		stdout string
+	}
+
+	tests := []struct {
+		name string
+		cli  string
+		mock mock
+		want want
+	}{
+		{
+			name: "happy-path",
+			cli:  "",
+			mock: mock{
+				CreateVersion: "v0.1",
+				CreateTimes:   1,
+				CreateReturnResponse: api.MongoInfoResponse{
+					Username:         "test",
+					Password:         "test",
+					Database:         "TestDB",
+					ConnectionString: "mongodb://test:test@localhost:27017/TestDB",
+				},
+				CreateReturnErr: nil,
+			},
+			want: want{
+				stdout: heredoc.Doc(`
+				✓ Database created:
+					username: test
+					password: test
+					database: TestDB
+					connectionString: mongodb://test:test@localhost:27017/TestDB
+				`),
+			},
+		},
+		{
+			name: "create-api-error",
+			cli:  "",
+			mock: mock{
+				CreateVersion:        "v0.1",
+				CreateTimes:          1,
+				CreateReturnResponse: api.MongoInfoResponse{},
+				CreateReturnErr:      errors.New("api error"),
+			},
+			want: want{
+				errMsg: "failed to create database: api error",
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+
+			ctrl := gomock.NewController(t)
+
+			deploymentMock := mocks.NewMockDeploymentInterface(ctrl)
+			deploymentMock.EXPECT().
+				CreateMongoDatabase(gomock.Any(), tt.mock.CreateVersion).
+				Times(tt.mock.CreateTimes).
+				Return(tt.mock.CreateReturnResponse, tt.mock.CreateReturnErr)
+
+			ios, _, stdout, stderr := iostreams.Test()
+
+			argv, err := shlex.Split(tt.cli)
+			if err != nil {
+				t.Fatal(err)
+			}
+
+			f := testutil.DefaultFactoryMock(t, ios, nil, nil, nil, deploymentMock, nil, nil)
+
+			cmd := NewCmdMongoCreate(f)
+			cmd.SetArgs(argv)
+			cmd.SetIn(&bytes.Buffer{})
+			cmd.SetOut(io.Discard)
+			cmd.SetErr(io.Discard)
+
+			if _, err := cmd.ExecuteC(); err != nil && tt.want.errMsg != "" {
+				require.Error(t, err, "should throw error")
+				require.Equal(t, tt.want.errMsg, err.Error())
+				return
+			}
+			cmdOut := &testutil.CmdOut{
+				OutBuf: stdout,
+				ErrBuf: stderr,
+			}
+
+			require.NoError(t, err, "should not throw error")
+			require.Equal(t, tt.want.stdout, cmdOut.String())
+
+		})
+	}
+}