Skip to content

Commit 051a825

Browse files
committed
internal/frontend: remove experiment middleware from newTestServer
No experiments are actually set in the callers to newTestServer or testServer. For golang/go#61399 Change-Id: Iae58a670cc17356bd24634e6ff17ee9f70ca5f69 Reviewed-on: https://go-review.googlesource.com/c/pkgsite/+/514520 Reviewed-by: Hyang-Ah Hana Kim <[email protected]> Run-TryBot: Michael Matloob <[email protected]> kokoro-CI: kokoro <[email protected]> TryBot-Result: Gopher Robot <[email protected]>
1 parent 30cc364 commit 051a825

File tree

2 files changed

+6
-39
lines changed

2 files changed

+6
-39
lines changed

internal/frontend/frontend_test.go

Lines changed: 3 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@ import (
1212

1313
"github.com/google/safehtml/template"
1414
"golang.org/x/pkgsite/internal"
15-
"golang.org/x/pkgsite/internal/middleware"
1615
"golang.org/x/pkgsite/internal/postgres"
1716
"golang.org/x/pkgsite/internal/proxy/proxytest"
1817
"golang.org/x/pkgsite/internal/queue"
@@ -45,13 +44,13 @@ type testPackage struct {
4544
docs []*internal.Documentation
4645
}
4746

48-
func newTestServer(t *testing.T, proxyModules []*proxytest.Module, cacher Cacher, experimentNames ...string) (*Server, http.Handler, func()) {
47+
func newTestServer(t *testing.T, proxyModules []*proxytest.Module, cacher Cacher) (*Server, http.Handler, func()) {
4948
t.Helper()
5049
proxyClient, teardown := proxytest.SetupTestClient(t, proxyModules)
5150
sourceClient := source.NewClient(sourceTimeout)
5251
ctx := context.Background()
5352

54-
q := queue.NewInMemory(ctx, 1, experimentNames,
53+
q := queue.NewInMemory(ctx, 1, nil,
5554
func(ctx context.Context, mpath, version string) (int, error) {
5655
return FetchAndUpdateState(ctx, mpath, version, proxyClient, sourceClient, testDB)
5756
})
@@ -73,16 +72,7 @@ func newTestServer(t *testing.T, proxyModules []*proxytest.Module, cacher Cacher
7372
mux := http.NewServeMux()
7473
s.Install(mux.Handle, cacher, nil)
7574

76-
var exps []*internal.Experiment
77-
for _, n := range experimentNames {
78-
exps = append(exps, &internal.Experiment{Name: n, Rollout: 100})
79-
}
80-
exp, err := middleware.NewExperimenter(ctx, time.Hour, func(context.Context) ([]*internal.Experiment, error) { return exps, nil }, nil)
81-
if err != nil {
82-
t.Fatal(err)
83-
}
84-
mw := middleware.Experiment(exp)
85-
return s, mw(mux), func() {
75+
return s, mux, func() {
8676
teardown()
8777
postgres.ResetTestDB(testDB, t)
8878
}

internal/frontend/server_test.go

Lines changed: 3 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,6 @@ import (
3030
"golang.org/x/pkgsite/internal"
3131
"golang.org/x/pkgsite/internal/cookie"
3232
"golang.org/x/pkgsite/internal/derrors"
33-
"golang.org/x/pkgsite/internal/experiment"
3433
"golang.org/x/pkgsite/internal/middleware"
3534
"golang.org/x/pkgsite/internal/postgres"
3635
"golang.org/x/pkgsite/internal/testing/htmlcheck"
@@ -62,8 +61,6 @@ type serverTestCase struct {
6261
wantLocation string
6362
// if non-nil, call the checker on the HTML root node
6463
want htmlcheck.Checker
65-
// list of experiments that must be enabled for this test to run
66-
requiredExperiments *experiment.Set
6764
}
6865

6966
// Units with this prefix will be marked as excluded.
@@ -1102,7 +1099,6 @@ func TestServer(t *testing.T) {
11021099
for _, test := range []struct {
11031100
name string
11041101
testCasesFunc func() []serverTestCase
1105-
experiments []string
11061102
}{
11071103
{
11081104
name: "no experiments",
@@ -1114,32 +1110,23 @@ func TestServer(t *testing.T) {
11141110
},
11151111
} {
11161112
t.Run(test.name, func(t *testing.T) {
1117-
testServer(t, test.testCasesFunc(), test.experiments...)
1113+
testServer(t, test.testCasesFunc())
11181114
})
11191115
}
11201116
}
11211117

1122-
func testServer(t *testing.T, testCases []serverTestCase, experimentNames ...string) {
1118+
func testServer(t *testing.T, testCases []serverTestCase) {
11231119
ctx, cancel := context.WithTimeout(context.Background(), testTimeout)
11241120
defer cancel()
11251121
defer postgres.ResetTestDB(testDB, t)
11261122

1127-
// Experiments need to be set in the context, for DB work, and as a
1128-
// middleware, for request handling.
1129-
ctx = experiment.NewContext(ctx, experimentNames...)
11301123
insertTestModules(ctx, t, testModules)
11311124
if err := testDB.InsertExcludedPrefix(ctx, excludedModulePath, "testuser", "testreason"); err != nil {
11321125
t.Fatal(err)
11331126
}
1134-
_, handler, _ := newTestServer(t, nil, nil, experimentNames...)
1135-
1136-
experimentsSet := experiment.NewSet(experimentNames...)
1127+
_, handler, _ := newTestServer(t, nil, nil)
11371128

11381129
for _, test := range testCases {
1139-
if !isSubset(test.requiredExperiments, experimentsSet) {
1140-
continue
1141-
}
1142-
11431130
t.Run(test.name, func(t *testing.T) { // remove initial '/' for name
11441131
w := httptest.NewRecorder()
11451132
handler.ServeHTTP(w, httptest.NewRequest("GET", test.urlPath, nil))
@@ -1170,16 +1157,6 @@ func testServer(t *testing.T, testCases []serverTestCase, experimentNames ...str
11701157
}
11711158
}
11721159

1173-
func isSubset(subset, set *experiment.Set) bool {
1174-
for _, e := range subset.Active() {
1175-
if !set.IsActive(e) {
1176-
return false
1177-
}
1178-
}
1179-
1180-
return true
1181-
}
1182-
11831160
func TestServerErrors(t *testing.T) {
11841161
_, handler, _ := newTestServer(t, nil, nil)
11851162
for _, test := range []struct {

0 commit comments

Comments
 (0)