Added mainBranch field in projectPreview

Author: cx-sumit-morchhale

internal/commands/project.go

@@ -610,6 +610,7 @@ func toProjectView(model wrappers.ProjectResponseModel) projectView { //nolint:g
 		Tags:           model.Tags,
 		Groups:         model.Groups,
 		ApplicationIds: model.ApplicationIds,
+		MainBranch:     model.MainBranch,
 	}
 }
 
@@ -618,6 +619,7 @@ type projectView struct {
 	Name           string
 	CreatedAt      time.Time `format:"name:Created at;time:01-02-06 15:04:05"`
 	UpdatedAt      time.Time `format:"name:Updated at;time:01-02-06 15:04:05"`
+	MainBranch     string
 	Tags           map[string]string
 	Groups         []string
 	ApplicationIds []string

internal/commands/project_test.go

@@ -3,7 +3,9 @@
 package commands
 
 import (
+	"github.com/checkmarx/ast-cli/internal/wrappers"
 	"testing"
+	"time"
 
 	asserts "github.com/stretchr/testify/assert"
 
@@ -269,3 +271,30 @@ func TestSupportEmptyTags_whenOnlyKeysFlagHasEmptyValue_shouldNotChangeParams(t
 	assert.Equal(t, params["tags-values"], "value1")
 	assert.Equal(t, len(params), 4)
 }
+
+func TestToProjectView(t *testing.T) {
+	// Arrange: Create a sample ProjectResponseModel
+	input := wrappers.ProjectResponseModel{
+		ID:             "123",
+		Name:           "Test Project",
+		CreatedAt:      time.Now(),
+		UpdatedAt:      time.Now(),
+		MainBranch:     "main",
+		Tags:           map[string]string{"key1": "value1"},
+		Groups:         []string{"group1", "group2"},
+		ApplicationIds: []string{"app1", "app2"},
+	}
+
+	// Act: Call the toProjectView function
+	result := toProjectView(input)
+
+	// Assert: Verify the fields are correctly mapped
+	assert.Equal(t, result.ID, input.ID)
+	assert.Equal(t, result.Name, input.Name)
+	assert.Equal(t, result.CreatedAt, input.CreatedAt)
+	assert.Equal(t, result.UpdatedAt, input.UpdatedAt)
+	assert.Equal(t, result.MainBranch, input.MainBranch)
+	assert.DeepEqual(t, result.Tags, input.Tags)
+	assert.DeepEqual(t, result.Groups, input.Groups)
+	assert.DeepEqual(t, result.ApplicationIds, input.ApplicationIds)
+}

test/integration/project_test.go

@@ -365,3 +365,21 @@ func TestCreateProjectWithSSHKey(t *testing.T) {
 	fmt.Printf("New project created with id: %s \n", createdProject.ID)
 	deleteProject(t, createdProject.ID)
 }
+
+func TestProjectShow_MainBranch_Exist(t *testing.T) {
+
+	projectID, projectName := createProject(t, Tags, Groups)
+
+	args := []string{
+		"scan", "create",
+		flag(params.ProjectName), projectName,
+		flag(params.SourcesFlag), "data/insecure.zip",
+		flag(params.BranchFlag), "dummy_branch",
+		flag(params.BranchPrimaryFlag), "dummy_branch",
+	}
+	err, _ = executeCommand(t, args...)
+	assert.NilError(t, err)
+
+	project := showProject(t, createdProject.ID)
+	asserts.Contains(t, project.MainBranch, "dummy_branch", "Project main branch should be 'dummy_branch'")
+}