Files
mautrix-telegram/pkg/gotd/telegram/handle_updates_test.go
T
2025-06-27 20:03:37 -07:00

41 lines
692 B
Go

package telegram
import (
"context"
"testing"
"github.com/stretchr/testify/require"
"go.mau.fi/mautrix-telegram/pkg/gotd/tg"
)
type mockHandler struct {
LastUpdate tg.UpdatesClass
}
func (m *mockHandler) Handle(ctx context.Context, u tg.UpdatesClass) error {
m.LastUpdate = u
return nil
}
func TestClient_processUpdates(t *testing.T) {
msg := &tg.Message{
ID: 1,
}
upd := &tg.Updates{
Updates: []tg.UpdateClass{&tg.UpdateNewMessage{
Message: msg,
}},
}
t.Run("Handle", func(t *testing.T) {
mock := &mockHandler{}
c := new(Client)
c.updateHandler = mock
err := c.processUpdates(upd)
require.NoError(t, err)
require.Equal(t, upd, mock.LastUpdate)
})
}