Glue: Add DB migration & support provisioning for user-defined correlations config (#55560)

* Allow provisioning correlation config

* Simplify code

* Fix reading correlations test

* Fix linting errors

* Fix linting errors

* remove simpleJson

* Clean up

* Fix tests

* Update swagger docs

* Fix linting

* Fix linting

* Clean up swagger definitions

Co-authored-by: Elfo404 <me@giordanoricci.com>

pkg/tests/api/correlations/correlations_create_test.go

@@ -211,7 +211,7 @@ func TestIntegrationCreateCorrelation(t *testing.T) {
 		require.NoError(t, res.Body.Close())
 	})
 
-	t.Run("Should correctly create a correlation", func(t *testing.T) {
+	t.Run("Should correctly create a correlation without a config", func(t *testing.T) {
 		description := "a description"
 		label := "a label"
 		res := ctx.Post(PostParams{
@@ -240,4 +240,68 @@ func TestIntegrationCreateCorrelation(t *testing.T) {
 
 		require.NoError(t, res.Body.Close())
 	})
+
+	t.Run("Should correctly create a correlation with a correct config", func(t *testing.T) {
+		description := "a description"
+		label := "a label"
+		res := ctx.Post(PostParams{
+			url: fmt.Sprintf("/api/datasources/uid/%s/correlations", writableDs),
+			body: fmt.Sprintf(`{
+					"targetUID": "%s",
+					"description": "%s",
+					"label": "%s",
+					"config": {
+						"field": "fieldName",
+						"target": { "expr": "foo" }
+					}
+				}`, writableDs, description, label),
+			user: adminUser,
+		})
+		require.Equal(t, http.StatusOK, res.StatusCode)
+
+		responseBody, err := io.ReadAll(res.Body)
+		require.NoError(t, err)
+
+		var response correlations.CreateCorrelationResponseBody
+		err = json.Unmarshal(responseBody, &response)
+		require.NoError(t, err)
+
+		require.Equal(t, "Correlation created", response.Message)
+		require.Equal(t, writableDs, response.Result.SourceUID)
+		require.Equal(t, writableDs, response.Result.TargetUID)
+		require.Equal(t, description, response.Result.Description)
+		require.Equal(t, label, response.Result.Label)
+		require.Equal(t, "fieldName", response.Result.Config.Field)
+
+		require.NoError(t, res.Body.Close())
+	})
+
+	t.Run("Should not create a correlation with incorrect config", func(t *testing.T) {
+		description := "a description"
+		label := "a label"
+		res := ctx.Post(PostParams{
+			url: fmt.Sprintf("/api/datasources/uid/%s/correlations", writableDs),
+			body: fmt.Sprintf(`{
+					"targetUID": "%s",
+					"description": "%s",
+					"label": "%s",
+					"config": {
+						"field": 2
+					}
+				}`, writableDs, description, label),
+			user: adminUser,
+		})
+		require.Equal(t, http.StatusBadRequest, res.StatusCode)
+
+		responseBody, err := io.ReadAll(res.Body)
+		require.NoError(t, err)
+
+		var response errorResponseBody
+		err = json.Unmarshal(responseBody, &response)
+		require.NoError(t, err)
+
+		require.Contains(t, response.Message, "bad request data")
+
+		require.NoError(t, res.Body.Close())
+	})
 }

pkg/tests/api/correlations/correlations_read_test.go

@@ -75,6 +75,10 @@ func TestIntegrationReadCorrelation(t *testing.T) {
 		SourceUID: dsWithCorrelations.Uid,
 		TargetUID: dsWithCorrelations.Uid,
 		OrgId:     dsWithCorrelations.OrgId,
+		Config: correlations.CorrelationConfig{
+			Field:  "foo",
+			Target: struct{}{},
+		},
 	})
 
 	createDsCommand = &datasources.AddDataSourceCommand{