Adding projectOwnerId to Projects.Create (#251)

Author: andreaangiolillo

mongodbatlas/projects.go

@@ -42,7 +42,7 @@ type ProjectsService interface {
 	GetAllProjects(context.Context, *ListOptions) (*Projects, *Response, error)
 	GetOneProject(context.Context, string) (*Project, *Response, error)
 	GetOneProjectByName(context.Context, string) (*Project, *Response, error)
-	Create(context.Context, *Project) (*Project, *Response, error)
+	Create(context.Context, *Project, *CreateProjectOptions) (*Project, *Response, error)
 	Delete(context.Context, string) (*Response, error)
 	GetProjectTeamsAssigned(context.Context, string) (*TeamsAssigned, *Response, error)
 	AddTeamsToProject(context.Context, string, []*ProjectTeam) (*TeamsAssigned, *Response, error)
@@ -99,6 +99,10 @@ type TeamsAssigned struct {
 	TotalCount int       `json:"totalCount"`
 }
 
+type CreateProjectOptions struct {
+	ProjectOwnerID string `url:"projectOwnerId,omitempty"` // Unique 24-hexadecimal digit string that identifies the Atlas user account to be granted the Project Owner role on the specified project.
+}
+
 // GetAllProjects gets all project.
 //
 // See more: https://docs.atlas.mongodb.com/reference/api/project-get-all/
@@ -177,12 +181,17 @@ func (s *ProjectsServiceOp) GetOneProjectByName(ctx context.Context, projectName
 // Create creates a project.
 //
 // See more: https://docs.atlas.mongodb.com/reference/api/project-create-one/
-func (s *ProjectsServiceOp) Create(ctx context.Context, createRequest *Project) (*Project, *Response, error) {
+func (s *ProjectsServiceOp) Create(ctx context.Context, createRequest *Project, opts *CreateProjectOptions) (*Project, *Response, error) {
 	if createRequest == nil {
 		return nil, nil, NewArgError("createRequest", "cannot be nil")
 	}
 
-	req, err := s.Client.NewRequest(ctx, http.MethodPost, projectBasePath, createRequest)
+	path, err := setListOptions(projectBasePath, opts)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	req, err := s.Client.NewRequest(ctx, http.MethodPost, path, createRequest)
 	if err != nil {
 		return nil, nil, err
 	}

mongodbatlas/projects_test.go

@@ -219,7 +219,56 @@ func TestProject_Create(t *testing.T) {
 		}`)
 	})
 
-	project, _, err := client.Projects.Create(ctx, createRequest)
+	opts := &CreateProjectOptions{ProjectOwnerID: "1"}
+
+	project, _, err := client.Projects.Create(ctx, createRequest, opts)
+	if err != nil {
+		t.Fatalf("Projects.Create returned error: %v", err)
+	}
+
+	expected := &Project{
+		ClusterCount: 2,
+		Created:      "2016-07-14T14:19:33Z",
+		ID:           "5a0a1e7e0f2912c554080ae6",
+		Links: []*Link{
+			{
+				Href: "https://cloud.mongodb.com/api/atlas/v1.0/groups/5a0a1e7e0f2912c554080ae6",
+				Rel:  "self",
+			},
+		},
+		Name:  "ProjectFoobar",
+		OrgID: "5a0a1e7e0f2912c554080adc",
+	}
+
+	if diff := deep.Equal(project, expected); diff != nil {
+		t.Error(diff)
+	}
+}
+
+func TestProject_Create_without_opts(t *testing.T) {
+	client, mux, teardown := setup()
+	defer teardown()
+
+	createRequest := &Project{
+		OrgID: "5a0a1e7e0f2912c554080adc",
+		Name:  "ProjectFoobar",
+	}
+
+	mux.HandleFunc("/api/atlas/v1.0/groups", func(w http.ResponseWriter, r *http.Request) {
+		fmt.Fprint(w, `{
+			"clusterCount": 2,
+			"created": "2016-07-14T14:19:33Z",
+			"id": "5a0a1e7e0f2912c554080ae6",
+			"links": [{
+				"href": "https://cloud.mongodb.com/api/atlas/v1.0/groups/5a0a1e7e0f2912c554080ae6",
+				"rel": "self"
+			}],
+			"name": "ProjectFoobar",
+			"orgId": "5a0a1e7e0f2912c554080adc"
+		}`)
+	})
+
+	project, _, err := client.Projects.Create(ctx, createRequest, nil)
 	if err != nil {
 		t.Fatalf("Projects.Create returned error: %v", err)
 	}