add unit test cases

Author: almas1992

pkgmanager/helper.go

@@ -42,7 +42,13 @@ func AddConfigSpec(path, name, statement string) pkgmanager.GoNodeModifier {
 	return ModifyGoNode{
 		Matchers: buildConfigMatchers(strings.Split(path, ".")),
 		Action: func(cursor *dstutil.Cursor) {
-			mv := cursor.Node().(*dst.KeyValueExpr).Value.(*dst.CompositeLit)
+			var mv *dst.CompositeLit
+			switch node := cursor.Node().(type) {
+			case *dst.KeyValueExpr:
+				mv = node.Value.(*dst.CompositeLit)
+			case *dst.CallExpr:
+				mv = node.Args[1].(*dst.CompositeLit)
+			}
 			key := WrapNewline(&dst.BasicLit{Kind: token.STRING, Value: strconv.Quote(name)})
 			if KeyExists(key, mv.Elts) {
 				color.Warningln(fmt.Sprintf("Key [%s] already exists in [%s]", name, path))

pkgmanager/helper_test.go

@@ -0,0 +1,154 @@
+package pkgmanager
+
+import (
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/suite"
+
+	"github.com/goravel/framework/contracts/pkgmanager"
+	"github.com/goravel/framework/support/file"
+)
+
+type HelperTestSuite struct {
+	suite.Suite
+	content string
+}
+
+func (s *HelperTestSuite) SetupTest() {
+	s.content = `package config
+
+import (
+	"github.com/goravel/framework/auth"
+	"github.com/goravel/framework/contracts/foundation"
+	"github.com/goravel/framework/crypt"
+	"github.com/goravel/framework/facades"
+)
+
+func Boot() {}
+
+func init() {
+	config := facades.Config()
+	config.Add("app", map[string]any{
+		"name": config.Env("APP_NAME", "Goravel"),
+		"providers": []foundation.ServiceProvider{
+			&auth.AuthServiceProvider{},
+			&crypt.ServiceProvider{},
+		},
+	})
+}`
+}
+
+func (s *HelperTestSuite) TearDownTest() {}
+
+func TestHelperTestSuite(t *testing.T) {
+	suite.Run(t, new(HelperTestSuite))
+}
+
+func (s *HelperTestSuite) TestHelper() {
+	cases := []struct {
+		name      string
+		modifiers []pkgmanager.GoNodeModifier
+
+		assert func(filename string)
+	}{
+		{
+			name: "AddConfigSpec",
+			modifiers: []pkgmanager.GoNodeModifier{
+				AddConfigSpec("app", "key", `"value"`),
+			},
+			assert: func(content string) {
+				s.Contains(content, `"key": "value",`)
+			},
+		},
+		{
+			name: "AddImportSpec",
+			modifiers: []pkgmanager.GoNodeModifier{
+				AddImportSpec("github.com/goravel/test"),
+			},
+			assert: func(content string) {
+				s.Contains(content, `"github.com/goravel/test"`)
+			},
+		},
+		{
+			name: "AddProviderSpec",
+			modifiers: []pkgmanager.GoNodeModifier{
+				AddProviderSpec("&test.ServiceProvider{}"),
+			},
+			assert: func(content string) {
+				s.Contains(content, "&crypt.ServiceProvider{},\n\t\t\t&test.ServiceProvider{}")
+			},
+		},
+		{
+			name: "AddProviderSpecAfter",
+			modifiers: []pkgmanager.GoNodeModifier{
+				AddProviderSpecAfter("&test.ServiceProvider{}", "&auth.AuthServiceProvider{}"),
+			},
+			assert: func(content string) {
+				s.Contains(content, "&auth.AuthServiceProvider{},\n\t\t\t&test.ServiceProvider{}")
+			},
+		},
+		{
+			name: "AddProviderSpecBefore",
+			modifiers: []pkgmanager.GoNodeModifier{
+				AddProviderSpecBefore("&test.ServiceProvider{}", "&auth.AuthServiceProvider{}"),
+			},
+			assert: func(content string) {
+				s.Contains(content, "&test.ServiceProvider{},\n\t\t\t&auth.AuthServiceProvider{}")
+			},
+		},
+		{
+			name: "RemoveConfigSpec",
+			modifiers: []pkgmanager.GoNodeModifier{
+				RemoveConfigSpec("app.providers"),
+			},
+			assert: func(content string) {
+				s.NotContains(content, "providers")
+			},
+		},
+		{
+			name: "RemoveImportSpec",
+			modifiers: []pkgmanager.GoNodeModifier{
+				RemoveImportSpec("github.com/goravel/framework/auth"),
+			},
+			assert: func(content string) {
+				s.NotContains(content, `"github.com/goravel/framework/auth"`)
+			},
+		},
+		{
+			name: "RemoveProviderSpec",
+			modifiers: []pkgmanager.GoNodeModifier{
+				RemoveProviderSpec("&auth.AuthServiceProvider{}"),
+			},
+			assert: func(content string) {
+				s.NotContains(content, "&auth.AuthServiceProvider{}")
+			},
+		},
+		{
+			name: "ReplaceConfigSpec",
+			modifiers: []pkgmanager.GoNodeModifier{
+				ReplaceConfigSpec("app.name", `"Goravel"`),
+			},
+			assert: func(content string) {
+				s.Contains(content, `"name": "Goravel"`)
+				s.NotContains(content, `config.Env("APP_NAME", "Goravel")`)
+			},
+		},
+	}
+
+	for _, tc := range cases {
+		s.Run(tc.name, func() {
+			dir := s.T().TempDir()
+			filename := "test.go"
+			s.Require().NoError(file.PutContent(filepath.Join(dir, filename), s.content))
+			mg := ModifyGoFile{
+				File:      filename,
+				Modifiers: tc.modifiers,
+			}
+			s.Require().NoError(mg.Apply(dir))
+			content, err := file.GetContent(filepath.Join(dir, filename))
+			s.Require().NoError(err)
+			tc.assert(content)
+		})
+	}
+}

pkgmanager/matcher_test.go

@@ -34,7 +34,7 @@ func init() {
 	config.Add("app", map[string]any{
 		"name": config.Env("APP_NAME", "Goravel"),
 		"providers": []foundation.ServiceProvider{
-			auth.AuthServiceProvider{},
+			&auth.AuthServiceProvider{},
 		},
 	})
 }`)