Function TestOverridesAPIHardLimits

integration/overrides_test.go:202–288

(t *testing.T)

}

func TestOverridesAPIHardLimits(t *testing.T) {
	s, err := e2e.NewScenario(networkName)
	require.NoError(t, err)
	defer s.Close()

	minio := e2edb.NewMinio(9001, "cortex")
	require.NoError(t, s.StartAndWaitReady(minio))

	// Runtime config with hard limits
	runtimeConfig := map[string]interface{}{
		"overrides": map[string]interface{}{},
		"hard_overrides": map[string]interface{}{
			"user1": map[string]interface{}{
				"ingestion_rate":             10000,
				"max_global_series_per_user": 50000,
			},
		},
		"api_allowed_limits": []string{
			"ingestion_rate",
			"max_global_series_per_user",
		},
	}
	uploadRuntimeConfig(t, minio, runtimeConfig)

	cortexSvc := e2ecortex.NewSingleBinary("cortex-overrides-hard-limits", defaultArgsOverrides(minio), "")
	require.NoError(t, s.StartAndWaitReady(cortexSvc))

	t.Run("POST overrides within hard limits", func(t *testing.T) {
		overrides := map[string]interface{}{
			"ingestion_rate":             5000,  // Within hard limit of 10000
			"max_global_series_per_user": 25000, // Within hard limit of 50000
		}
		requestBody, err := json.Marshal(overrides)
		require.NoError(t, err)

		req, err := http.NewRequest("POST", "http://"+cortexSvc.HTTPEndpoint()+"/api/v1/user-overrides", bytes.NewReader(requestBody))
		require.NoError(t, err)
		req.Header.Set("X-Scope-OrgID", "user1")
		req.Header.Set("Content-Type", "application/json")

		resp, err := http.DefaultClient.Do(req)
		require.NoError(t, err)
		defer resp.Body.Close()

		assert.Equal(t, http.StatusOK, resp.StatusCode)
	})

	t.Run("POST overrides exceeding hard limits", func(t *testing.T) {
		overrides := map[string]interface{}{
			"ingestion_rate": 15000, // Exceeds hard limit of 10000
		}
		requestBody, err := json.Marshal(overrides)
		require.NoError(t, err)

		req, err := http.NewRequest("POST", "http://"+cortexSvc.HTTPEndpoint()+"/api/v1/user-overrides", bytes.NewReader(requestBody))
		require.NoError(t, err)
		req.Header.Set("X-Scope-OrgID", "user1")
		req.Header.Set("Content-Type", "application/json")