Function TestAutoscaler_MaxReplicas

pkg/autoscaler/autoscaler_test.go:391–461

(t *testing.T)

}

func TestAutoscaler_MaxReplicas(t *testing.T) {
	t.Parallel()
	log := newLogger(t)

	mux := sync.RWMutex{}
	var latestRequest int32

	minReplicas := int32(5)
	maxReplicas := int32(10)

	scalerMock := &ScalerFunc{
		ScaleFunc: func(apiName string, request int32) error {
			mux.Lock()
			defer mux.Unlock()

			latestRequest = request
			return nil
		},
		GetInFlightRequestsFunc: func(apiName string, window time.Duration) (*float64, error) {
			return pointer.Float64(20), nil
		},
		GetAutoscalingSpecFunc: func(apiName string) (*userconfig.Autoscaling, error) {
			return &userconfig.Autoscaling{
				MinReplicas:        minReplicas,
				MaxReplicas:        maxReplicas,
				InitReplicas:       minReplicas,
				TargetInFlight:     pointer.Float64(1),
				Window:             500 * time.Millisecond,
				MaxDownscaleFactor: 0.75,
				MaxUpscaleFactor:   1.5,
			}, nil
		},
		CurrentRequestedReplicasFunc: func(apiName string) (int32, error) {
			return minReplicas, nil
		},
	}

	autoScaler := &Autoscaler{
		logger:  log,
		crons:   make(map[string]cron.Cron),
		scalers: make(map[userconfig.Kind]Scaler),
		recs:    make(map[string]*recommendations),
	}
	autoScaler.AddScaler(scalerMock, userconfig.RealtimeAPIKind)

	apiName := "test"
	api := userconfig.Resource{
		Name: apiName,
		Kind: userconfig.RealtimeAPIKind,
	}

	autoscaleFn, err := autoScaler.autoscaleFn(api)
	require.NoError(t, err)

	ticker := time.NewTicker(250 * time.Millisecond)
	go func() {
		for {
			select {