diff options
Diffstat (limited to 'internal/postgres/unit_test.go')
| -rw-r--r-- | internal/postgres/unit_test.go | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/internal/postgres/unit_test.go b/internal/postgres/unit_test.go index 53603643..b0096353 100644 --- a/internal/postgres/unit_test.go +++ b/internal/postgres/unit_test.go @@ -23,7 +23,7 @@ import ( func TestGetUnit(t *testing.T) { ctx, cancel := context.WithTimeout(context.Background(), testTimeout) defer cancel() - + ctx = experiment.NewContext(ctx, internal.ExperimentInsertPackageSource) defer ResetTestDB(testDB, t) InsertSampleDirectoryTree(ctx, t, testDB) @@ -208,6 +208,7 @@ func TestGetUnitFieldSet(t *testing.T) { ctx, cancel := context.WithTimeout(context.Background(), testTimeout) defer cancel() + ctx = experiment.NewContext(ctx, internal.ExperimentInsertPackageSource) defer ResetTestDB(testDB, t) // Add a module that has READMEs in a directory and a package. |