1 // Copyright 2018 The Go Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style
3 // license that can be found in the LICENSE file.
12 "golang.org/x/mod/module"
15 var addRequireTests = []struct {
50 require x.y/q/v2 v2.3.4
64 var setRequireTests = []struct {
87 {"x.y/a", "v1.2.3", false},
88 {"x.y/b", "v1.2.3", false},
89 {"x.y/c", "v1.2.3", false},
106 x.y/e v1.2.3 // indirect
107 x.y/f v1.2.3 //indirect
108 x.y/g v1.2.3 // indirect
116 {"x.y/a", "v1.2.3", true},
117 {"x.y/b", "v1.2.3", true},
118 {"x.y/c", "v1.2.3", true},
119 {"x.y/d", "v1.2.3", true},
120 {"x.y/e", "v1.2.3", true},
121 {"x.y/f", "v1.2.3", true},
122 {"x.y/g", "v1.2.3", true},
126 x.y/a v1.2.3 // indirect
127 x.y/b v1.2.3 // indirect
128 x.y/c v1.2.3 // indirect; c
129 x.y/d v1.2.3 // indirect; c
130 x.y/e v1.2.3 // indirect
131 x.y/f v1.2.3 //indirect
132 x.y/g v1.2.3 // indirect
138 var addGoTests = []struct {
160 `require x.y/a v1.2.3
161 module example.com/inverted
164 `require x.y/a v1.2.3
165 module example.com/inverted
170 `require x.y/a v1.2.3
173 `require x.y/a v1.2.3
179 func TestAddRequire(t *testing.T) {
180 for i, tt := range addRequireTests {
181 t.Run(fmt.Sprintf("#%d", i), func(t *testing.T) {
182 f, err := Parse("in", []byte(tt.in), nil)
186 g, err := Parse("out", []byte(tt.out), nil)
190 golden, err := g.Format()
195 if err := f.AddRequire(tt.path, tt.vers); err != nil {
198 out, err := f.Format()
202 if !bytes.Equal(out, golden) {
203 t.Errorf("have:\n%s\nwant:\n%s", out, golden)
209 func TestSetRequire(t *testing.T) {
210 for i, tt := range setRequireTests {
211 t.Run(fmt.Sprintf("#%d", i), func(t *testing.T) {
212 f, err := Parse("in", []byte(tt.in), nil)
216 g, err := Parse("out", []byte(tt.out), nil)
220 golden, err := g.Format()
225 for _, mod := range tt.mods {
226 mods = append(mods, &Require{
231 Indirect: mod.indirect,
236 out, err := f.Format()
240 if !bytes.Equal(out, golden) {
241 t.Errorf("have:\n%s\nwant:\n%s", out, golden)
245 if len(f.Require) != len(mods) {
246 t.Errorf("after Cleanup, len(Require) = %v; want %v", len(f.Require), len(mods))
252 func TestAddGo(t *testing.T) {
253 for i, tt := range addGoTests {
254 t.Run(fmt.Sprintf("#%d", i), func(t *testing.T) {
255 f, err := Parse("in", []byte(tt.in), nil)
259 g, err := Parse("out", []byte(tt.out), nil)
263 golden, err := g.Format()
268 if err := f.AddGoStmt(tt.version); err != nil {
271 out, err := f.Format()
275 if !bytes.Equal(out, golden) {
276 t.Errorf("have:\n%s\nwant:\n%s", out, golden)