update groups when creating a new project(AST-64011)

internal/services/groups.go

@@ -8,6 +8,20 @@ import (
 	"github.com/pkg/errors"
 )
 
+func GetGroupMap(groupsWrapper wrappers.GroupsWrapper, projectGroups string, projModelResp *wrappers.ProjectResponseModel,
+	featureFlagsWrapper wrappers.FeatureFlagsWrapper) ([]*wrappers.Group, []string, error) {
+	groupsMap, groupErr := CreateGroupsMap(projectGroups, groupsWrapper)
+	if groupErr != nil {
+		return nil, nil, errors.Errorf("%s: %v", failedUpdatingProj, groupErr)
+	}
+	groups := getGroupsForRequest(groupsMap, featureFlagsWrapper)
+	if projModelResp != nil {
+		groups = append(getGroupsForRequest(groupsMap, featureFlagsWrapper), projModelResp.Groups...)
+		return groupsMap, groups, nil
+	}
+	return groupsMap, groups, nil
+}
+
 func CreateGroupsMap(groupsStr string, groupsWrapper wrappers.GroupsWrapper) ([]*wrappers.Group, error) {
 	groups := strings.Split(groupsStr, ",")
 	var groupsMap []*wrappers.Group

internal/services/projects.go

@@ -87,6 +87,16 @@ func createProject(
 	var projModel = wrappers.Project{}
 	projModel.Name = projectName
 	projModel.ApplicationIds = applicationID
+	var groupsMap []*wrappers.Group
+	if projectGroups != "" {
+		var groups []string
+		var groupErr error
+		groupsMap, groups, groupErr = GetGroupMap(groupsWrapper, projectGroups, nil, featureFlagsWrapper)
+		if groupErr != nil {
+			return "", groupErr
+		}
+		projModel.Groups = groups
+	}
 
 	if projectPrivatePackage != "" {
 		projModel.PrivatePackage, _ = strconv.ParseBool(projectPrivatePackage)
@@ -108,7 +118,7 @@ func createProject(
 		}
 
 		if projectGroups != "" {
-			err = UpsertProjectGroups(groupsWrapper, &projModel, projectsWrapper, accessManagementWrapper, nil, projectGroups, projectID, projectName, featureFlagsWrapper)
+			err = UpsertProjectGroups(&projModel, projectsWrapper, accessManagementWrapper, projectID, projectName, featureFlagsWrapper, groupsMap)
 			if err != nil {
 				return projectID, err
 			}
@@ -215,27 +225,28 @@ func updateProject(
 	}
 
 	if projectGroups != "" {
-		err = UpsertProjectGroups(groupsWrapper, &projModel, projectsWrapper, accessManagementWrapper, projModelResp, projectGroups, projectID, projectName, featureFlagsWrapper)
+		err = UpsertProjectGroupsByUpdateFlow(groupsWrapper, &projModel, projectsWrapper, accessManagementWrapper, projModelResp, projectGroups, projectID, projectName, featureFlagsWrapper)
 		if err != nil {
 			return projectID, err
 		}
 	}
 	return projectID, nil
 }
 
-func UpsertProjectGroups(groupsWrapper wrappers.GroupsWrapper, projModel *wrappers.Project, projectsWrapper wrappers.ProjectsWrapper,
+func UpsertProjectGroupsByUpdateFlow(groupsWrapper wrappers.GroupsWrapper, projModel *wrappers.Project, projectsWrapper wrappers.ProjectsWrapper,
 	accessManagementWrapper wrappers.AccessManagementWrapper, projModelResp *wrappers.ProjectResponseModel,
 	projectGroups string, projectID string, projectName string, featureFlagsWrapper wrappers.FeatureFlagsWrapper) error {
-	groupsMap, groupErr := CreateGroupsMap(projectGroups, groupsWrapper)
+	groupsMap, groups, groupErr := GetGroupMap(groupsWrapper, projectGroups, projModelResp, featureFlagsWrapper)
 	if groupErr != nil {
-		return errors.Errorf("%s: %v", failedUpdatingProj, groupErr)
-	}
-	projModel.Groups = getGroupsForRequest(groupsMap, featureFlagsWrapper)
-	if projModelResp != nil {
-		groups := append(getGroupsForRequest(groupsMap, featureFlagsWrapper), projModelResp.Groups...)
-		projModel.Groups = groups
+		return groupErr
 	}
+	projModel.Groups = groups
+	return UpsertProjectGroups(projModel, projectsWrapper, accessManagementWrapper, projectID, projectName, featureFlagsWrapper, groupsMap)
+}
 
+func UpsertProjectGroups(projModel *wrappers.Project, projectsWrapper wrappers.ProjectsWrapper,
+	accessManagementWrapper wrappers.AccessManagementWrapper, projectID string, projectName string,
+	featureFlagsWrapper wrappers.FeatureFlagsWrapper, groupsMap []*wrappers.Group) error {
 	err := AssignGroupsToProjectNewAccessManagement(projectID, projectName, groupsMap, accessManagementWrapper, featureFlagsWrapper)
 	if err != nil {
 		return err

internal/services/projects_test.go

@@ -106,6 +106,15 @@ func Test_createProject(t *testing.T) {
 			projectGroups:           "",
 			featureFlagsWrapper:     &mock.FeatureFlagsMockWrapper{},
 		}, want: "ID-new-project-name", wantErr: false},
+		{name: "When called with a new project name and existing project groups return the Id of the newly created project", args: args{
+			projectName:             "new-project-name",
+			cmd:                     &cobra.Command{},
+			projectsWrapper:         &mock.ProjectsMockWrapper{},
+			groupsWrapper:           &mock.GroupsMockWrapper{},
+			accessManagementWrapper: &mock.AccessManagementMockWrapper{},
+			projectGroups:           "existsGroup1,existsGroup2",
+			featureFlagsWrapper:     &mock.FeatureFlagsMockWrapper{},
+		}, want: "ID-new-project-name", wantErr: false},
 		{name: "When called with a new project name and non existing project groups return error", args: args{
 			projectName:             "new-project-name",
 			cmd:                     &cobra.Command{},
@@ -114,7 +123,7 @@ func Test_createProject(t *testing.T) {
 			accessManagementWrapper: &mock.AccessManagementMockWrapper{},
 			projectGroups:           "grp1,grp2",
 			featureFlagsWrapper:     &mock.FeatureFlagsMockWrapper{},
-		}, want: "ID-new-project-name", wantErr: true},
+		}, want: "", wantErr: true},
 		{name: "When called with mock fake error model return fake error from project create", args: args{
 			projectName:             "mock-some-error-model",
 			cmd:                     &cobra.Command{},
@@ -132,7 +141,7 @@ func Test_createProject(t *testing.T) {
 			accessManagementWrapper: &mock.AccessManagementMockWrapper{},
 			projectGroups:           "fake-group-error",
 			featureFlagsWrapper:     &mock.FeatureFlagsMockWrapper{},
-		}, want: "ID-new-project-name", wantErr: true},
+		}, want: "", wantErr: true},
 		{name: "When called with a new project name and projectPrivatePackage set to true return the Id of the newly created project", args: args{
 			projectName:             "new-project-name",
 			cmd:                     &cobra.Command{},

internal/wrappers/mock/groups-mock.go

@@ -12,5 +12,8 @@ func (g *GroupsMockWrapper) Get(groupName string) ([]wrappers.Group, error) {
 	if groupName == "fake-group-error" {
 		return nil, errors.Errorf("fake grroup error")
 	}
+	if groupName == "existsGroup1" || groupName == "existsGroup2" {
+		return []wrappers.Group{{ID: "1", Name: groupName}}, nil
+	}
 	return []wrappers.Group{{ID: "1", Name: "group"}}, nil
 }