mirror of
https://github.com/hanzoai/status.git
synced 2026-10-30 18:21:48 +00:00
zeekayandHanzo Dev
78ab4f2f83
zip's registration is a verb method on the group, so a route that is not a typed operation says which method it answers with. The security package's own two-method router goes with it: it existed so both the app and a sub-router could satisfy one name, and restating a routing surface is what let the two disagree — the methods it named no longer exist. The app spells itself as a group, which registers at exactly the paths it did. The toolchain and the pin move together, because below zip v1.36.50 the JSON encoder was selected on a build tag: a repo whose go directive rises first writes a wire that is neither the old release's nor the new one's until the pin catches up. metrics_test.go still does not compile, before and after: it compares a metric registry against a gatherer from a different metrics library, so the two Gather signatures disagree. That is its own change. Co-authored-by: Hanzo Dev <[email protected]>
142 lines
4.6 KiB
Go
142 lines
4.6 KiB
Go
package squadcast
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"hanzo.ai/status/alerting/alert"
|
|
"hanzo.ai/status/client"
|
|
"hanzo.ai/status/config/endpoint"
|
|
"hanzo.ai/status/test"
|
|
)
|
|
|
|
func TestAlertProvider_Validate(t *testing.T) {
|
|
scenarios := []struct {
|
|
name string
|
|
provider AlertProvider
|
|
expected error
|
|
}{
|
|
{
|
|
name: "valid",
|
|
provider: AlertProvider{DefaultConfig: Config{WebhookURL: "https://api.squadcast.com/v3/incidents/api/abcd1234"}},
|
|
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://api.squadcast.com/v3/incidents/api/abcd1234"}},
|
|
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["status"] != "trigger" {
|
|
t.Errorf("expected status to be 'trigger', got %v", body["status"])
|
|
}
|
|
if body["event_id"] == nil {
|
|
t.Error("expected 'event_id' field in request body")
|
|
}
|
|
message := body["message"].(string)
|
|
if !strings.Contains(message, "ALERT") {
|
|
t.Errorf("expected message to contain 'ALERT', got %s", message)
|
|
}
|
|
description := body["description"].(string)
|
|
if !strings.Contains(description, "failed 3 time(s)") {
|
|
t.Errorf("expected description to contain failure count, got %s", description)
|
|
}
|
|
return &http.Response{StatusCode: http.StatusOK, Body: http.NoBody}
|
|
}),
|
|
expectedError: false,
|
|
},
|
|
{
|
|
name: "resolved",
|
|
provider: AlertProvider{DefaultConfig: Config{WebhookURL: "https://api.squadcast.com/v3/incidents/api/abcd1234"}},
|
|
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)
|
|
if body["status"] != "resolve" {
|
|
t.Errorf("expected status to be 'resolve', got %v", body["status"])
|
|
}
|
|
message := body["message"].(string)
|
|
if !strings.Contains(message, "RESOLVED") {
|
|
t.Errorf("expected message to contain 'RESOLVED', got %s", message)
|
|
}
|
|
return &http.Response{StatusCode: http.StatusOK, Body: http.NoBody}
|
|
}),
|
|
expectedError: false,
|
|
},
|
|
{
|
|
name: "error-response",
|
|
provider: AlertProvider{DefaultConfig: Config{WebhookURL: "https://api.squadcast.com/v3/incidents/api/abcd1234"}},
|
|
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")
|
|
}
|
|
}
|