feat(message_gateway): integrate nikoksr/notify engine and support multi-channel push

2026-10-10 17:26:38 +08:00 · 2026-09-02 22:30:59 +08:00
parent 90f3efdd50
commit 7df31befb5
13 changed files with 252 additions and 788 deletions
@@ -5,112 +5,24 @@ package push

 import (
 	"context"
-	"encoding/json"
-	"net/http"
-	"net/http/httptest"
 	"testing"
-
-	"github.com/stretchr/testify/assert"
-	"github.com/stretchr/testify/require"
 )

-func TestTelegramPusher_Send(t *testing.T) {
-	t.Run("successful send with HTML parse mode", func(t *testing.T) {
-		var receivedReq telegramMessageRequest
-		server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-			assert.Equal(t, "/botmy-token/sendMessage", r.URL.Path)
-			assert.Equal(t, http.MethodPost, r.Method)
-			assert.Equal(t, "application/json", r.Header.Get("Content-Type"))
+func TestTelegramPusherValidation(t *testing.T) {
+	pusher := &TelegramPusher{}

-			err := json.NewDecoder(r.Body).Decode(&receivedReq)
-			require.NoError(t, err)
+	err := pusher.ValidateConfig(Config{Secret: "123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11"})
+	if err != nil {
+		t.Errorf("ValidateConfig failed: %v", err)
+	}

-			w.WriteHeader(http.StatusOK)
-			_, _ = w.Write([]byte(`{"ok": true}`))
-		}))
-		defer server.Close()
+	err = pusher.ValidateConfig(Config{})
+	if err == nil {
+		t.Errorf("expected error for empty config, got nil")
+	}

-		pusher := &TelegramPusher{}
-		cfg := Config{
-			Channel: "telegram",
-			URL:     server.URL,
-			Secret:  "my-token",
-		}
-		body := map[string]any{
-			"title":   "Alert",
-			"content": "Host down",
-			"level":   "CRITICAL",
-		}
-		_, err := pusher.Send(context.Background(), cfg, "123456", body, "", nil)
-		require.NoError(t, err)
-
-		assert.Equal(t, "123456", receivedReq.ChatID)
-		assert.Contains(t, receivedReq.Text, "[CRITICAL] Alert")
-		assert.Contains(t, receivedReq.Text, "Host down")
-		assert.Equal(t, "HTML", receivedReq.ParseMode)
-	})
-
-	t.Run("fallback to plain text on HTML error", func(t *testing.T) {
-		var requests []*telegramMessageRequest
-		server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-			var req telegramMessageRequest
-			err := json.NewDecoder(r.Body).Decode(&req)
-			require.NoError(t, err)
-			requests = append(requests, &req)
-
-			if len(requests) == 1 {
-				w.WriteHeader(http.StatusBadRequest)
-				_, _ = w.Write([]byte(`{"ok": false, "error_code": 400, "description": "Bad Request: can't parse entities"}`))
-			} else {
-				w.WriteHeader(http.StatusOK)
-				_, _ = w.Write([]byte(`{"ok": true}`))
-			}
-		}))
-		defer server.Close()
-
-		pusher := &TelegramPusher{}
-		cfg := Config{
-			Channel: "telegram",
-			URL:     server.URL,
-			Secret:  "my-token",
-		}
-		body := map[string]any{
-			"title":   "Alert & Info",
-			"content": "A < B comparison",
-			"level":   "INFO",
-		}
-		_, err := pusher.Send(context.Background(), cfg, "123456", body, "", nil)
-		require.NoError(t, err)
-
-		require.Len(t, requests, 2)
-		assert.Equal(t, "HTML", requests[0].ParseMode)
-		assert.Equal(t, "", requests[1].ParseMode)
-		assert.Contains(t, requests[1].Text, "[INFO] Alert & Info")
-		assert.Contains(t, requests[1].Text, "A < B comparison")
-	})
-
-	t.Run("validation error", func(t *testing.T) {
-		pusher := &TelegramPusher{}
-		cfg := Config{
-			Channel: "telegram",
-			URL:     "https://api.telegram.org",
-		}
-		err := pusher.ValidateConfig(cfg)
-		assert.Error(t, err)
-
-		cfg = Config{
-			Channel: "telegram",
-			URL:     "ftp://api.telegram.org",
-			Secret:  "token",
-		}
-		err = pusher.ValidateConfig(cfg)
-		assert.Error(t, err)
-
-		cfg = Config{
-			Channel: "telegram",
-			Secret:  "token",
-		}
-		err = pusher.ValidateConfig(cfg)
-		assert.NoError(t, err)
-	})
+	_, err = pusher.Send(context.Background(), Config{Secret: "123:token"}, "not-a-number", map[string]any{"title": "test"}, "", nil)
+	if err == nil {
+		t.Errorf("expected error for invalid chat_id, got nil")
+	}
 }