internal/testing/sample: add AddPackage

For golang/go#39629

Change-Id: I9dcf43ec3ec245ef1514b6bdb64afd1e8266a425
Reviewed-on: https://go-review.googlesource.com/c/pkgsite/+/258561
Trust: Julie Qiu <julie@golang.org>
Run-TryBot: Julie Qiu <julie@golang.org>
TryBot-Result: kokoro <noreply+kokoro@google.com>
Reviewed-by: Jonathan Amsterdam <jba@google.com>

1 files changed, 35 insertions, 0 deletions

diff --git a/internal/testing/sample/sample.go b/internal/testing/sample/sample.go
index 1d024bb4..969ce87d 100644
--- a/internal/testing/sample/sample.go
+++ b/internal/testing/sample/sample.go
@@ -93,6 +93,41 @@ func NowTruncated() time.Time {
 	return time.Now().Truncate(time.Microsecond)
 }
 
+func AddPackage(m *internal.Module, fullPath string) *internal.Module {
+	if m.ModulePath != stdlib.ModulePath && !strings.HasPrefix(fullPath, m.ModulePath) {
+		panic(fmt.Sprintf("package path %q not a prefix of module path %q",
+			fullPath, m.ModulePath))
+	}
+	AddUnit(m, &internal.Unit{
+		UnitMeta:        *UnitMeta(fullPath, m.ModulePath, m.Version, path.Base(fullPath), true),
+		Imports:         Imports,
+		LicenseContents: Licenses,
+		Documentation: &internal.Documentation{
+			Synopsis: Synopsis,
+			HTML:     DocumentationHTML,
+			GOOS:     GOOS,
+			GOARCH:   GOARCH,
+		},
+	})
+	minLen := len(m.ModulePath)
+	if m.ModulePath == stdlib.ModulePath {
+		minLen = 1
+	}
+	for pth := fullPath; len(pth) > minLen; pth = path.Dir(pth) {
+		found := false
+		for _, u := range m.Units {
+			if u.Path == pth {
+				found = true
+				break
+			}
+		}
+		if !found {
+			AddUnit(m, UnitEmpty(pth, m.ModulePath, m.Version))
+		}
+	}
+	return m
+}
+
 func PackageMeta(fullPath string) *internal.PackageMeta {
 	return &internal.PackageMeta{
 		Path:              fullPath,