Function TestActivatorHandler_ServeHTTP

pkg/activator/handler_test.go:30–66

(t *testing.T)

)

func TestActivatorHandler_ServeHTTP(t *testing.T) {
	t.Parallel()

	log := newLogger(t)

	apiName := "test"
	act := &activator{
		autoscalerClient: autoscalerClientMock{},
		apiActivators: map[string]*apiActivator{
			apiName: newAPIActivator(1, 1),
		},
		readinessTrackers: map[string]*readinessTracker{
			apiName: {ready: true},
		},
		logger: log,
	}

	ah := NewHandler(act, log)

	var callCount int
	server := httptest.NewServer(
		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			callCount++
			w.WriteHeader(http.StatusOK)
		}),
	)

	w := httptest.NewRecorder()
	r := httptest.NewRequest("POST", "http://fake.cortex.dev/api", nil)
	r.Header.Set(consts.CortexAPINameHeader, apiName)
	r.Header.Set(consts.CortexTargetServiceHeader, server.URL)

	ah.ServeHTTP(w, r)

	require.Equal(t, http.StatusOK, w.Code)
	require.Equal(t, 1, callCount)
}