Function TestWithCacheShardLevelPaths

pkg/cache/server/handler_test.go:51–131

(t *testing.T)

}

func TestWithCacheShardLevelPaths(t *testing.T) {
	t.Parallel()
	tests := []struct {
		name           string
		path           string
		shard          request.Shard
		cluster        *request.Cluster
		wantStatus     int
		wantNextCalled bool
	}{
		{
			name:           "non-shard path passes through",
			path:           "/apis/apis.kcp.io/v1alpha1/apiexports",
			shard:          request.Shard("amber"),
			cluster:        &request.Cluster{Name: logicalcluster.Name("ws-1234")},
			wantStatus:     http.StatusOK,
			wantNextCalled: true,
		},
		{
			name:           "metrics with no shard and no cluster passes through",
			path:           "/metrics",
			wantStatus:     http.StatusOK,
			wantNextCalled: true,
		},
		{
			name:           "metrics with shard set is rejected with 501",
			path:           "/metrics",
			shard:          request.Shard("amber"),
			wantStatus:     http.StatusNotImplemented,
			wantNextCalled: false,
		},
		{
			name:           "metrics with cluster set is rejected with 501",
			path:           "/metrics",
			cluster:        &request.Cluster{Name: logicalcluster.Name("ws-1234")},
			wantStatus:     http.StatusNotImplemented,
			wantNextCalled: false,
		},
		{
			name:           "metrics with both shard and cluster set is rejected",
			path:           "/metrics",
			shard:          request.Shard("amber"),
			cluster:        &request.Cluster{Name: logicalcluster.Name("ws-1234")},
			wantStatus:     http.StatusNotImplemented,
			wantNextCalled: false,
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			t.Parallel()
			nextCalled := false
			next := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				nextCalled = true
				w.WriteHeader(http.StatusOK)
			})

			h := WithCacheShardLevelPaths(next)