Function TestOverridesAPIWithRunningCortex

integration/overrides_test.go:50–200

(t *testing.T)

}

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

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

	runtimeConfig := map[string]interface{}{
		"overrides": map[string]interface{}{
			"user1": map[string]interface{}{
				"ingestion_rate": 5000,
			},
		},
		"api_allowed_limits": []string{
			"ingestion_rate",
			"max_global_series_per_user",
			"max_global_series_per_metric",
			"ingestion_burst_size",
			"ruler_max_rules_per_rule_group",
			"ruler_max_rule_groups_per_tenant",
		},
	}
	uploadRuntimeConfig(t, minio, runtimeConfig)

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

	t.Run("GET overrides for existing user", func(t *testing.T) {
		req, err := http.NewRequest("GET", "http://"+cortexSvc.HTTPEndpoint()+"/api/v1/user-overrides", nil)
		require.NoError(t, err)
		req.Header.Set("X-Scope-OrgID", "user1")

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

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

		var overrides map[string]interface{}
		err = json.NewDecoder(resp.Body).Decode(&overrides)
		require.NoError(t, err)

		assert.Equal(t, float64(5000), overrides["ingestion_rate"])
	})

	t.Run("GET overrides for non-existing user", func(t *testing.T) {
		req, err := http.NewRequest("GET", "http://"+cortexSvc.HTTPEndpoint()+"/api/v1/user-overrides", nil)
		require.NoError(t, err)
		req.Header.Set("X-Scope-OrgID", "user2")

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

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