mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-06 15:46:37 +08:00
ryan
416603b616
- Replace db.DB(ctx) with database.DB(ctx) from plugins/infra/database
- Replace db.Redis/db.PrefixedKey/db.GetJSON/db.SetJSON with cachepkg.* from plugins/infra/cache
- Replace pkg/persistence/idgen with pkg/idgen (already exists)
- Replace pkg/persistence/batchwriter with pkg/batchwriter (already exists)
- Replace pkg/persistence/migrator with pkg/migrator (already exists)
- Replace pkg/persistence/logstore with plugins/domain/risk_control/logstore
- Delete defunct pkg/{persistence,cap,message_gateway,push,shared,task}
- Fix vet issues: db alias in domain_test.go, driver_asynq_worker.TaskHandler reference
- Update Makefile architecture guard
- Update docs and skill references
- Update go.mod: gorilla/sessions promotion to direct dependency
57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package telegram
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/Rain-kl/Wavelet/plugins/domain/message_gateway"
|
|
tele "gopkg.in/telebot.v4"
|
|
)
|
|
|
|
func TestHandleUpdate_DropsGroups(t *testing.T) {
|
|
var got int
|
|
a := &Adapter{onInbound: func(ctx context.Context, msg message_gateway.InboundMessage) error {
|
|
got++
|
|
return nil
|
|
}}
|
|
a.handleTeleMessage(context.Background(), &tele.Message{
|
|
ID: 1,
|
|
Text: "hi",
|
|
Chat: &tele.Chat{ID: -100, Type: tele.ChatGroup},
|
|
Sender: &tele.User{ID: 1},
|
|
})
|
|
if got != 0 {
|
|
t.Fatalf("group must be ignored")
|
|
}
|
|
}
|
|
|
|
func TestHandleUpdate_PrivateText(t *testing.T) {
|
|
var got message_gateway.InboundMessage
|
|
a := &Adapter{
|
|
cfg: message_gateway.ChannelConfig{ID: 7, Type: "telegram"},
|
|
onInbound: func(ctx context.Context, msg message_gateway.InboundMessage) error {
|
|
got = msg
|
|
return nil
|
|
},
|
|
}
|
|
a.handleTeleMessage(context.Background(), &tele.Message{
|
|
ID: 9,
|
|
Text: "hi",
|
|
Chat: &tele.Chat{ID: 42, Type: tele.ChatPrivate},
|
|
Sender: &tele.User{ID: 42},
|
|
})
|
|
if got.Text != "hi" || got.PlatformUserID != "42" || got.ChannelID != 7 {
|
|
t.Fatalf("%+v", got)
|
|
}
|
|
}
|
|
|
|
func TestNew_RequiresToken(t *testing.T) {
|
|
_, err := New(message_gateway.ChannelConfig{}, nil)
|
|
if err == nil {
|
|
t.Fatal("expected error")
|
|
}
|
|
}
|