| // Copyright 2022 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package worker |
| |
| import ( |
| "context" |
| "fmt" |
| "testing" |
| |
| "github.com/google/go-cmp/cmp" |
| "golang.org/x/pkgsite-metrics/internal/config" |
| "golang.org/x/pkgsite-metrics/internal/govulncheck" |
| "golang.org/x/pkgsite-metrics/internal/queue" |
| "golang.org/x/pkgsite-metrics/internal/scan" |
| ) |
| |
| func TestCreateQueueTasks(t *testing.T) { |
| vreq := func(path, version, mode string, importedBy int) *govulncheck.Request { |
| return &govulncheck.Request{ |
| ModuleURLPath: scan.ModuleURLPath{Module: path, Version: version}, |
| QueryParams: govulncheck.QueryParams{Mode: mode, ImportedBy: importedBy}, |
| } |
| } |
| |
| params := &govulncheck.EnqueueQueryParams{Min: 8, File: "testdata/modules.txt"} |
| gotTasks, err := createGovulncheckQueueTasks(context.Background(), &config.Config{}, params, []string{ModeGovulncheck}) |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| wantTasks := []queue.Task{ |
| vreq("github.com/pkg/errors", "v0.9.1", ModeGovulncheck, 10), |
| vreq("golang.org/x/net", "v0.4.0", ModeGovulncheck, 20), |
| } |
| if diff := cmp.Diff(wantTasks, gotTasks, cmp.AllowUnexported(govulncheck.Request{})); diff != "" { |
| t.Errorf("mismatch (-want, +got):\n%s", diff) |
| } |
| |
| allModes, err := listModes("", true) |
| if err != nil { |
| t.Fatal(err) |
| } |
| gotTasks, err = createGovulncheckQueueTasks(context.Background(), &config.Config{}, params, allModes) |
| if err != nil { |
| t.Fatal(err) |
| } |
| // cfg.BinaryBucket is empty, so no binary-mode tasks are created. |
| wantTasks = []queue.Task{ |
| vreq("github.com/pkg/errors", "v0.9.1", ModeGovulncheck, 10), |
| vreq("golang.org/x/net", "v0.4.0", ModeGovulncheck, 20), |
| } |
| |
| if diff := cmp.Diff(wantTasks, gotTasks, cmp.AllowUnexported(govulncheck.Request{})); diff != "" { |
| t.Errorf("mismatch (-want, +got):\n%s", diff) |
| } |
| } |
| |
| func TestListModes(t *testing.T) { |
| for _, test := range []struct { |
| param string |
| all bool |
| want []string |
| wantErr bool |
| }{ |
| {"", true, []string{ModeGovulncheck}, false}, |
| {"", false, []string{ModeGovulncheck}, false}, |
| {"imports", true, nil, true}, |
| } { |
| t.Run(fmt.Sprintf("%q,%t", test.param, test.all), func(t *testing.T) { |
| got, err := listModes(test.param, test.all) |
| if err != nil && !test.wantErr { |
| t.Fatal(err) |
| } |
| if err == nil && !cmp.Equal(got, test.want) { |
| t.Errorf("got %v, want %v", got, test.want) |
| } |
| }) |
| } |
| } |