From fa03708786f2dce02a5ad581e45e997017f72ddf Mon Sep 17 00:00:00 2001 From: Jonathan Amsterdam Date: Wed, 9 Jun 2021 13:49:25 -0400 Subject: internal/postgres: read from imports table Read from the imports table instead of package_imports, under an experiment. Change-Id: Id5b61285325f28a168faeb25651839663269feaf Reviewed-on: https://go-review.googlesource.com/c/pkgsite/+/326450 Trust: Jonathan Amsterdam Run-TryBot: Jonathan Amsterdam Reviewed-by: Jamal Carvalho Reviewed-by: Julie Qiu --- internal/postgres/unit_test.go | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'internal/postgres/unit_test.go') diff --git a/internal/postgres/unit_test.go b/internal/postgres/unit_test.go index 64e0b57f..c60941b7 100644 --- a/internal/postgres/unit_test.go +++ b/internal/postgres/unit_test.go @@ -743,7 +743,12 @@ func TestGetUnit(t *testing.T) { test.want.IsRedistributable, ) test.want.CommitTime = um.CommitTime - checkUnit(ctx, t, testDB, um, test.want) + t.Run("none", func(t *testing.T) { + checkUnit(ctx, t, testDB, um, test.want) + }) + t.Run(internal.ExperimentReadImports, func(t *testing.T) { + checkUnit(ctx, t, testDB, um, test.want, internal.ExperimentReadImports) + }) }) } } -- cgit v1.3