feat(alerts): integration tests for polling frequency field

Author: gffuentes

pkg/alerts/nrql_conditions_integration_test.go

@@ -30,6 +30,8 @@ var (
 	nrqlConditionTitleTemplate          = "Title {{ createdAt }}"                     // needed for setting pointer
 	nrqlConditionPredictBy              = 7200                                        // needed for setting pointer
 	nrqlConditionSignalSeasonality      = NrqlSignalSeasonalities.Weekly              // needed for setting pointer
+	nrqlConditionBasePollingFrequency   = 3600                                        // needed for setting pointer
+	nrqlConditionBasePollingAggWindow   = 3600                                        // needed for setting pointer
 	nrqlConditionCreateBase             = NrqlConditionCreateBase{
 		Description: "test description",
 		Enabled:     true,
@@ -1229,3 +1231,108 @@ func TestIntegrationNrqlConditions_DisableHealthStatusReporting(t *testing.T) {
 		}
 	}()
 }
+
+func TestIntegrationNrqlConditions_SignalPollingFrequency(t *testing.T) {
+	t.Parallel()
+
+	testAccountID, err := mock.GetTestAccountID()
+	if err != nil {
+		t.Skipf("%s", err)
+	}
+
+	var (
+		updatedPollingFrequency = 7200
+	)
+
+	var conditionCreateInput = NrqlConditionCreateInput{
+		NrqlConditionCreateBase: NrqlConditionCreateBase{
+			Enabled: true,
+			Name:    fmt.Sprintf("test-nrql-condition-%s", testNrqlConditionRandomString),
+			Nrql: NrqlConditionCreateQuery{
+				Query:         "select count(*) from CloudCost",
+				DataAccountId: &testAccountID,
+			},
+			Terms: []NrqlConditionTerm{
+				{
+					Threshold:            &nrqlConditionBaseThreshold,
+					ThresholdOccurrences: ThresholdOccurrences.AtLeastOnce,
+					ThresholdDuration:    3600,
+					Operator:             AlertsNRQLConditionTermsOperatorTypes.ABOVE,
+					Priority:             NrqlConditionPriorities.Critical,
+				},
+			},
+			ViolationTimeLimitSeconds: 3600,
+			Signal: &AlertsNrqlConditionCreateSignal{
+				AggregationWindow: &nrqlConditionBasePollingAggWindow,
+				EvaluationDelay:   &nrqlConditionEvaluationDelay,
+				AggregationDelay:  &nrqlConditionBaseAggDelay,
+				PollingFrequency:  &nrqlConditionBasePollingFrequency,
+			},
+		},
+	}
+	var conditionUpdateInput = NrqlConditionUpdateInput{
+		NrqlConditionUpdateBase: NrqlConditionUpdateBase{
+			Enabled: true,
+			Name:    fmt.Sprintf("test-nrql-condition-%s", testNrqlConditionRandomString),
+			Nrql: NrqlConditionUpdateQuery{
+				Query:         "select count(*) from CloudCost",
+				DataAccountId: &testAccountID,
+			},
+			Terms: []NrqlConditionTerm{
+				{
+					Threshold:            &nrqlConditionBaseThreshold,
+					ThresholdOccurrences: ThresholdOccurrences.AtLeastOnce,
+					ThresholdDuration:    7200,
+					Operator:             AlertsNRQLConditionTermsOperatorTypes.ABOVE,
+					Priority:             NrqlConditionPriorities.Critical,
+				},
+			},
+			ViolationTimeLimitSeconds: 3600,
+			Signal: &AlertsNrqlConditionUpdateSignal{
+				AggregationWindow: &nrqlConditionBasePollingAggWindow,
+				EvaluationDelay:   &nrqlConditionEvaluationDelay,
+				AggregationDelay:  &nrqlConditionBaseAggDelay,
+				PollingFrequency:  &updatedPollingFrequency,
+			},
+		},
+	}
+
+	var randStr = mock.RandSeq(5)
+
+	// Setup
+	client := newIntegrationTestClient(t)
+	testPolicy := AlertsPolicyInput{
+		IncidentPreference: AlertsIncidentPreferenceTypes.PER_POLICY,
+		Name:               fmt.Sprintf("test-alert-policy-%s", randStr),
+	}
+	policy, err := client.CreatePolicyMutation(testAccountID, testPolicy)
+	require.NoError(t, err)
+
+	// Test: Create (condition with Signal.PollingFrequency field)
+
+	createdCondition, err := client.CreateNrqlConditionStaticMutation(testAccountID, policy.ID, conditionCreateInput)
+	require.NoError(t, err)
+	require.NotNil(t, createdCondition)
+	require.NotNil(t, createdCondition.ID)
+	require.NotNil(t, createdCondition.PolicyID)
+	require.Equal(t, &nrqlConditionBasePollingFrequency, createdCondition.Signal.PollingFrequency)
+
+	// Test: Get (condition with Signal.PollingFrequency field)
+	readResult, err := client.GetNrqlConditionQuery(testAccountID, createdCondition.ID)
+	require.NoError(t, err)
+	require.NotNil(t, readResult)
+	require.Equal(t, &nrqlConditionBasePollingFrequency, readResult.Signal.PollingFrequency)
+
+	// Test: Update (condition with non-default Signal.PollingFrequency field)
+	updatedCondition, err := client.UpdateNrqlConditionStaticMutation(testAccountID, createdCondition.ID, conditionUpdateInput)
+	require.NoError(t, err)
+	require.Equal(t, &updatedPollingFrequency, updatedCondition.Signal.PollingFrequency)
+
+	// Deferred teardown
+	defer func() {
+		_, err := client.DeletePolicyMutation(testAccountID, policy.ID)
+		if err != nil {
+			t.Logf("error cleaning up alert policy %s (%s): %s", policy.ID, policy.Name, err)
+		}
+	}()
+}