64 lines
2.2 KiB
Go
64 lines
2.2 KiB
Go
package notify
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"git.casaderoll.de/michael/urbm/internal/model"
|
|
)
|
|
|
|
type testSecrets map[string]string
|
|
|
|
func (s testSecrets) Get(id string) (string, error) { return s[id], nil }
|
|
|
|
type roundTripFunc func(*http.Request) (*http.Response, error)
|
|
|
|
func (f roundTripFunc) RoundTrip(request *http.Request) (*http.Response, error) { return f(request) }
|
|
|
|
func TestSendGotify(t *testing.T) {
|
|
var body map[string]any
|
|
client := &http.Client{Transport: roundTripFunc(func(r *http.Request) (*http.Response, error) {
|
|
if r.URL.Path != "/message" || r.URL.Query().Get("token") != "secret-token" {
|
|
t.Errorf("unexpected request URL %s", r.URL.String())
|
|
}
|
|
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
|
|
return nil, err
|
|
}
|
|
return &http.Response{StatusCode: http.StatusOK, Status: "200 OK", Body: io.NopCloser(strings.NewReader("{}")), Header: make(http.Header)}, nil
|
|
})}
|
|
|
|
sender := &Sender{Secrets: testSecrets{"gotify-token": "secret-token"}, Client: client}
|
|
target := model.NotificationTarget{Enabled: true, Type: "gotify", URL: "https://gotify.example.test", TokenRef: "gotify-token"}
|
|
if err := sender.Send(context.Background(), target, "Backup failed", "repository unavailable", "failed"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if body["title"] != "Backup failed" || body["message"] != "repository unavailable" || body["priority"] != float64(8) {
|
|
t.Fatalf("unexpected Gotify payload %#v", body)
|
|
}
|
|
}
|
|
|
|
func TestSendUnraid(t *testing.T) {
|
|
var command string
|
|
var args []string
|
|
sender := &Sender{RunCommand: func(_ context.Context, name string, values ...string) error {
|
|
command, args = name, values
|
|
return nil
|
|
}}
|
|
target := model.NotificationTarget{Enabled: true, Type: "unraid"}
|
|
if err := sender.Send(context.Background(), target, "Backup warning", "completed with warnings", "warning"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if command != "/usr/local/emhttp/webGui/scripts/notify" {
|
|
t.Fatalf("unexpected command %q", command)
|
|
}
|
|
want := []string{"-e", "URBM", "-s", "Backup warning", "-d", "completed with warnings", "-i", "warning", "-l", "/Tools/URBM"}
|
|
if !reflect.DeepEqual(args, want) {
|
|
t.Fatalf("arguments = %#v, want %#v", args, want)
|
|
}
|
|
}
|