157 lines
4.4 KiB
Go
157 lines
4.4 KiB
Go
package chat
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"testing"
|
||
)
|
||
|
||
const (
|
||
uidA UserID = "u-a"
|
||
uidB UserID = "u-b"
|
||
tg Address = "tg://123"
|
||
tui Address = "tui://local"
|
||
)
|
||
|
||
// fakeOnMsg — тест-колбэк, копящий входящие.
|
||
type fakeOnMsg struct{ got []Incoming }
|
||
|
||
func (f *fakeOnMsg) h(inc Incoming) { f.got = append(f.got, inc) }
|
||
|
||
func TestRouter_AttachAndIncoming(t *testing.T) {
|
||
cb := &fakeOnMsg{}
|
||
r := NewRouter(cb.h)
|
||
|
||
tgCh := newFakeChannel(tg)
|
||
if err := r.Attach(tgCh); err != nil {
|
||
t.Fatalf("Attach: %v", err)
|
||
}
|
||
|
||
tgCh.emit(uidA, tg, "привет")
|
||
if len(cb.got) != 1 {
|
||
t.Fatalf("handler got %d, want 1", len(cb.got))
|
||
}
|
||
got := cb.got[0]
|
||
if got.UserID != uidA || got.Address != tg || got.Msg.Text != "привет" {
|
||
t.Errorf("incoming = %+v", got)
|
||
}
|
||
}
|
||
|
||
func TestRouter_AttachNil(t *testing.T) {
|
||
r := NewRouter(nil)
|
||
if err := r.Attach(nil); err == nil {
|
||
t.Fatal("Attach(nil) должен вернуть ошибку")
|
||
}
|
||
}
|
||
|
||
func TestRouter_Send_NoRoute(t *testing.T) {
|
||
r := NewRouter(nil)
|
||
// M1: нет маршрута → Send no-op, nil
|
||
if err := r.Send(context.Background(), uidA, Message{Text: "x"}); err != nil {
|
||
t.Fatalf("Send без маршрута: %v", err)
|
||
}
|
||
}
|
||
|
||
func TestRouter_Send_UsesCurrentRoute(t *testing.T) {
|
||
cb := &fakeOnMsg{}
|
||
r := NewRouter(cb.h)
|
||
tgCh := newFakeChannel(tg)
|
||
|
||
_ = r.Attach(tgCh)
|
||
tgCh.emit(uidA, tg, "hi") // устанавливает маршрут
|
||
|
||
if err := r.Send(context.Background(), uidA, Message{Text: "отв"}); err != nil {
|
||
t.Fatalf("Send: %v", err)
|
||
}
|
||
if tgCh.sentCount() != 1 {
|
||
t.Fatalf("sent = %d, want 1", tgCh.sentCount())
|
||
}
|
||
if tgCh.sent[0].Msg.Text != "отв" {
|
||
t.Errorf("msg = %q", tgCh.sent[0].Msg.Text)
|
||
}
|
||
}
|
||
|
||
func TestRouter_SwitchChannel_Continues(t *testing.T) {
|
||
cb := &fakeOnMsg{}
|
||
r := NewRouter(cb.h)
|
||
tgCh := newFakeChannel(tg)
|
||
tuiCh := newFakeChannel(tui)
|
||
_ = r.Attach(tgCh)
|
||
_ = r.Attach(tuiCh)
|
||
|
||
// начал в TG
|
||
tgCh.emit(uidA, tg, "hi")
|
||
// продолжил в GUI
|
||
tuiCh.emit(uidA, tui, "продолжаю тут")
|
||
if tgCh.sentCount() != 0 || tuiCh.sentCount() != 0 {
|
||
t.Fatal("до Send ничего не шлём")
|
||
}
|
||
|
||
// ответ должен уйти в последний канал (GUI)
|
||
_ = r.Send(context.Background(), uidA, Message{Text: "отв"})
|
||
if tuiCh.sentCount() != 1 {
|
||
t.Errorf("tui sent = %d, want 1 (последний маршрут)", tuiCh.sentCount())
|
||
}
|
||
if tgCh.sentCount() != 0 {
|
||
t.Errorf("tg sent = %d, want 0", tgCh.sentCount())
|
||
}
|
||
}
|
||
|
||
func TestRouter_Ask_PendingThenAnswer(t *testing.T) {
|
||
cb := &fakeOnMsg{}
|
||
r := NewRouter(cb.h)
|
||
tgCh := newFakeChannel(tg)
|
||
_ = r.Attach(tgCh)
|
||
tgCh.emit(uidA, tg, "hi")
|
||
|
||
prompt := Message{Text: "Как зовут?", Options: []Option{{ID: "a", Label: "Анна"}}}
|
||
if err := r.Ask(context.Background(), uidA, prompt); err != nil {
|
||
t.Fatalf("Ask: %v", err)
|
||
}
|
||
if _, ok := r.Pending(uidA); !ok {
|
||
t.Fatal("pending не открыт")
|
||
}
|
||
// повторный Ask → M3
|
||
if err := r.Ask(context.Background(), uidA, prompt); !errors.Is(err, ErrWaitingAnswer) {
|
||
t.Fatalf("второй Ask err = %v, want ErrWaitingAnswer", err)
|
||
}
|
||
|
||
// ответ с того же адреса потребляет pending
|
||
tgCh.emit(uidA, tg, "Анна")
|
||
if _, ok := r.Pending(uidA); ok {
|
||
t.Fatal("pending должен быть закрыт после ответа")
|
||
}
|
||
if len(cb.got) != 2 {
|
||
t.Fatalf("handler got %d, want 2 (hi + ответ)", len(cb.got))
|
||
}
|
||
if cb.got[1].Msg.QuestionID == "" {
|
||
t.Error("ответ должен нести QuestionID вопроса")
|
||
}
|
||
}
|
||
|
||
func TestRouter_Ask_NoRoute(t *testing.T) {
|
||
r := NewRouter(nil)
|
||
if err := r.Ask(context.Background(), uidB, Message{Text: "q"}); !errors.Is(err, ErrRouteNotFound) {
|
||
t.Fatalf("Ask без маршрута err = %v, want ErrRouteNotFound", err)
|
||
}
|
||
}
|
||
|
||
func TestRouter_Ask_PendingNotConsumedFromOtherAddr(t *testing.T) {
|
||
cb := &fakeOnMsg{}
|
||
r := NewRouter(cb.h)
|
||
tgCh := newFakeChannel(tg)
|
||
tuiCh := newFakeChannel(tui)
|
||
_ = r.Attach(tgCh)
|
||
_ = r.Attach(tuiCh)
|
||
tgCh.emit(uidA, tg, "hi")
|
||
|
||
if err := r.Ask(context.Background(), uidA, Message{Text: "q"}); err != nil {
|
||
t.Fatalf("Ask: %v", err)
|
||
}
|
||
// ответ из ДРУГОГО канала → это новый message, pending НЕ потребляется
|
||
tuiCh.emit(uidA, tui, "ответ из gui")
|
||
if _, ok := r.Pending(uidA); !ok {
|
||
t.Fatal("pending должен остаться (ответ из другого адреса)")
|
||
}
|
||
}
|