Adam Van Ymeren
7a04f298d2
- update to latest telegram layer - remove some references to fields in tg.Entities that don't exist in the schema - originally added here: https://github.com/beeper/td/commit/820929062a2ba0104397bc01235ab58a9cff780e - referenced here - https://github.com/mautrix/telegramgo/commit/124f0967ed195b5a380c9bd02e170ada9710dde3 - https://github.com/mautrix/telegramgo/commit/4205047aab2e0639217148b5d125bfaab668bd8e
77 lines
1.5 KiB
Go
77 lines
1.5 KiB
Go
package transport_test
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/go-faster/errors"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/sync/errgroup"
|
|
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/bin"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/telegram/dcs"
|
|
"go.mau.fi/mautrix-telegram/pkg/gotd/transport"
|
|
)
|
|
|
|
func TestWebsocketListener(t *testing.T) {
|
|
a := require.New(t)
|
|
ctx := context.Background()
|
|
|
|
var handler http.Handler
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
handler.ServeHTTP(w, r)
|
|
}))
|
|
defer srv.Close()
|
|
|
|
listener, h := transport.WebsocketListener(srv.Listener.Addr())
|
|
handler = h
|
|
list := dcs.List{
|
|
Domains: map[int]string{
|
|
2: srv.URL,
|
|
},
|
|
}
|
|
|
|
server := transport.Listen(listener)
|
|
defer server.Close()
|
|
done := make(chan struct{})
|
|
|
|
grp, ctx := errgroup.WithContext(ctx)
|
|
grp.Go(func() error {
|
|
defer close(done)
|
|
|
|
conn, err := server.Accept()
|
|
if err != nil {
|
|
return errors.Wrap(err, "accept")
|
|
}
|
|
|
|
var b bin.Buffer
|
|
if err := conn.Recv(ctx, &b); err != nil {
|
|
return errors.Wrap(err, "recv")
|
|
}
|
|
|
|
if err := conn.Send(ctx, &b); err != nil {
|
|
return errors.Wrap(err, "send")
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
rs := dcs.Websocket(dcs.WebsocketOptions{})
|
|
conn, err := rs.Primary(ctx, 2, list)
|
|
a.NoError(err)
|
|
|
|
data, err := io.ReadAll(io.LimitReader(rand.Reader, 1024))
|
|
a.NoError(err)
|
|
a.NoError(conn.Send(ctx, &bin.Buffer{Buf: data}))
|
|
|
|
var b bin.Buffer
|
|
a.NoError(conn.Recv(ctx, &b))
|
|
a.Equal(data, b.Buf)
|
|
|
|
a.NoError(grp.Wait())
|
|
}
|