test updateWorkfile

goplus · Feb 12, 2024 · 27d860a · 27d860a
1 parent bfcfcf4
commit 27d860a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 12 deletions.
diff --git a/modload/module.go b/modload/module.go
@@ -345,20 +345,20 @@ const (
 // SaveWithGopMod adds `require github.com/goplus/gop` and saves all
 // changes of this module.
 func (p Module) SaveWithGopMod(gop *env.Gop, flags int) (err error) {
-	gopVer := getGopVer(gop)
-	if old := p.checkGopDeps(); old != flags {
-		if p.Path() == gopMod { // don't change Go+ itself
-			return
-		}
-		p.requireGop(gop, gopVer, old, flags)
-		if err = p.Save(); err != nil {
-			return
-		}
+	if p.Path() == gopMod { // don't change Go+ itself
+		return
 	}
-	if (flags & FlagDepModGop) == 0 {
+	old := p.checkGopDeps()
+	if (flags &^ old) == 0 { // nothing to do
 		return
 	}
 
+	gopVer := getGopVer(gop)
+	p.requireGop(gop, gopVer, old, flags)
+	return p.Save()
+}
+
+func (p Module) updateWorkfile(gop *env.Gop, gopVer string) (err error) {
 	var work *gomodfile.WorkFile
 	var workFile = p.workFile()
 	b, err := os.ReadFile(workFile)
@@ -386,6 +386,7 @@ func (p Module) SaveWithGopMod(gop *env.Gop, flags int) (err error) {
 func (p Module) requireGop(gop *env.Gop, gopVer string, old, flags int) {
 	if (flags&FlagDepModGop) != 0 && (old&FlagDepModGop) == 0 {
 		p.File.AddRequire(gopMod, gopVer)
+		p.updateWorkfile(gop, gopVer)
 	}
 	if (flags&FlagDepModX) != 0 && (old&FlagDepModX) == 0 { // depends module github.com/qiniu/x
 		if x, xsum, ok := getXVer(gop); ok {

diff --git a/modload/module_test.go b/modload/module_test.go
@@ -248,8 +248,8 @@ require (
 		log.Fatal("mod.SaveWithGopMod 3:", err)
 	}
 
-	if err = mod.SaveWithGopMod(&env.Gop{Version: "v1.2.0 devel", Root: ".gop"}, FlagDepModX); err != nil {
-		log.Fatal("mod.SaveWithGopMod 4:", err)
+	if err = mod.updateWorkfile(&env.Gop{Version: "v1.2.0 devel", Root: ".gop"}, ""); err != nil {
+		log.Fatal("updateWorkfile:", err)
 	}
 
 	mod.Opt.Projects = append(mod.Opt.Projects, spxProject)