package webex import ( "encoding/json" "net/http" "strings" "testing" "github.com/TwiN/gatus/v5/alerting/alert" "github.com/TwiN/gatus/v5/client" "github.com/TwiN/gatus/v5/config/endpoint" "github.com/TwiN/gatus/v5/test" ) func TestAlertProvider_Validate(t *testing.T) { scenarios := []struct { name string provider AlertProvider expected error }{ { name: "valid", provider: AlertProvider{DefaultConfig: Config{WebhookURL: "https://webexapis.com/v1/webhooks/incoming/123"}}, expected: nil, }, { name: "invalid-webhook-url", provider: AlertProvider{DefaultConfig: Config{}}, expected: ErrWebhookURLNotSet, }, } for _, scenario := range scenarios { t.Run(scenario.name, func(t *testing.T) { err := scenario.provider.Validate() if err != scenario.expected { t.Errorf("expected %v, got %v", scenario.expected, err) } }) } } func TestAlertProvider_Send(t *testing.T) { defer client.InjectHTTPClient(nil) firstDescription := "description-1" secondDescription := "description-2" scenarios := []struct { name string provider AlertProvider alert alert.Alert resolved bool mockRoundTripper test.MockRoundTripper expectedError bool }{ { name: "triggered", provider: AlertProvider{DefaultConfig: Config{WebhookURL: "https://webexapis.com/v1/webhooks/incoming/123"}}, alert: alert.Alert{Description: &firstDescription, SuccessThreshold: 5, FailureThreshold: 3}, resolved: false, mockRoundTripper: test.MockRoundTripper(func(r *http.Request) *http.Response { body := make(map[string]interface{}) json.NewDecoder(r.Body).Decode(&body) if body["markdown"] == nil { t.Error("expected 'markdown' field in request body") } markdown := body["markdown"].(string) if !strings.Contains(markdown, "ALERT") { t.Errorf("expected markdown to contain 'ALERT', got %s", markdown) } if !strings.Contains(markdown, "failed 3 time(s)") { t.Errorf("expected markdown to contain failure count, got %s", markdown) } return &http.Response{StatusCode: http.StatusOK, Body: http.NoBody} }), expectedError: false, }, { name: "resolved", provider: AlertProvider{DefaultConfig: Config{WebhookURL: "https://webexapis.com/v1/webhooks/incoming/123"}}, alert: alert.Alert{Description: &secondDescription, SuccessThreshold: 5, FailureThreshold: 3}, resolved: true, mockRoundTripper: test.MockRoundTripper(func(r *http.Request) *http.Response { body := make(map[string]interface{}) json.NewDecoder(r.Body).Decode(&body) markdown := body["markdown"].(string) if !strings.Contains(markdown, "RESOLVED") { t.Errorf("expected markdown to contain 'RESOLVED', got %s", markdown) } return &http.Response{StatusCode: http.StatusOK, Body: http.NoBody} }), expectedError: false, }, { name: "error-response", provider: AlertProvider{DefaultConfig: Config{WebhookURL: "https://webexapis.com/v1/webhooks/incoming/123"}}, alert: alert.Alert{Description: &firstDescription, SuccessThreshold: 5, FailureThreshold: 3}, resolved: false, mockRoundTripper: test.MockRoundTripper(func(r *http.Request) *http.Response { return &http.Response{StatusCode: http.StatusUnauthorized, Body: http.NoBody} }), expectedError: true, }, } for _, scenario := range scenarios { t.Run(scenario.name, func(t *testing.T) { client.InjectHTTPClient(&http.Client{Transport: scenario.mockRoundTripper}) err := scenario.provider.Send( &endpoint.Endpoint{Name: "endpoint-name"}, &scenario.alert, &endpoint.Result{ ConditionResults: []*endpoint.ConditionResult{ {Condition: "[CONNECTED] == true", Success: scenario.resolved}, {Condition: "[STATUS] == 200", Success: scenario.resolved}, }, }, scenario.resolved, ) if scenario.expectedError && err == nil { t.Error("expected error, got none") } if !scenario.expectedError && err != nil { t.Error("expected no error, got", err.Error()) } }) } } func TestAlertProvider_GetDefaultAlert(t *testing.T) { if (&AlertProvider{DefaultAlert: &alert.Alert{}}).GetDefaultAlert() == nil { t.Error("expected default alert to be not nil") } if (&AlertProvider{DefaultAlert: nil}).GetDefaultAlert() != nil { t.Error("expected default alert to be nil") } }