internal/vulncheck/internal/buildinfo/additions_scan_test.go

// Copyright 2021 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.

//go:build go1.18
// +build go1.18

package buildinfo

import (
	"os"
	"testing"

	"github.com/google/go-cmp/cmp"
	"golang.org/x/vuln/internal/test"
)

func TestExtractPackagesAndSymbols(t *testing.T) {
	unsupported := map[string]bool{
		"darwin/386": true,
		"darwin/arm": true,
	}

	for _, g := range []string{"linux", "darwin", "windows", "freebsd"} {
		for _, a := range []string{"amd64", "386", "arm", "arm64"} {
			goos := g
			goarch := a

			ga := goos + "/" + goarch
			if unsupported[ga] {
				continue
			}

			t.Run(ga, func(t *testing.T) {
				binary, done := test.GoBuild(t, "testdata", "", "GOOS", goos, "GOARCH", goarch)
				defer done()

				f, err := os.Open(binary)
				if err != nil {
					t.Fatal(err)
				}
				defer f.Close()
				_, syms, _, err := ExtractPackagesAndSymbols(f)
				if err != nil {
					t.Fatal(err)
				}
				got := syms["main"]
				want := []string{"main", "f", "g"}
				if !cmp.Equal(got, want) {
					t.Errorf("\ngot  %q\nwant %q", got, want)
				}
			})
		}
	}
}