move gotd fork into repo. (#111)
- 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
This commit is contained in:
@@ -0,0 +1,76 @@
|
||||
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())
|
||||
}
|
||||
Reference in New Issue
Block a user