5733c71c50
We were ignoring quite a few error cases, and had one case where we weren't actually updating state where we wanted to. Unfortunately, if the linter doesn't pass, nobody has any reason to actually check lint failures in CI. There are three remaining XXXs marked in the code for lint.
91 lines
2.5 KiB
Go
91 lines
2.5 KiB
Go
package rpcenc
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/filecoin-project/go-jsonrpc"
|
|
sealing "github.com/filecoin-project/lotus/extern/storage-sealing"
|
|
)
|
|
|
|
type ReaderHandler struct {
|
|
}
|
|
|
|
func (h *ReaderHandler) ReadAll(ctx context.Context, r io.Reader) ([]byte, error) {
|
|
return ioutil.ReadAll(r)
|
|
}
|
|
|
|
func (h *ReaderHandler) ReadNullLen(ctx context.Context, r io.Reader) (int64, error) {
|
|
return r.(*sealing.NullReader).N, nil
|
|
}
|
|
|
|
func (h *ReaderHandler) ReadUrl(ctx context.Context, u string) (string, error) {
|
|
return u, nil
|
|
}
|
|
|
|
func TestReaderProxy(t *testing.T) {
|
|
var client struct {
|
|
ReadAll func(ctx context.Context, r io.Reader) ([]byte, error)
|
|
}
|
|
|
|
serverHandler := &ReaderHandler{}
|
|
|
|
readerHandler, readerServerOpt := ReaderParamDecoder()
|
|
rpcServer := jsonrpc.NewServer(readerServerOpt)
|
|
rpcServer.Register("ReaderHandler", serverHandler)
|
|
|
|
mux := mux.NewRouter()
|
|
mux.Handle("/rpc/v0", rpcServer)
|
|
mux.Handle("/rpc/streams/v0/push/{uuid}", readerHandler)
|
|
|
|
testServ := httptest.NewServer(mux)
|
|
defer testServ.Close()
|
|
|
|
re := ReaderParamEncoder("http://" + testServ.Listener.Addr().String() + "/rpc/streams/v0/push")
|
|
closer, err := jsonrpc.NewMergeClient("ws://"+testServ.Listener.Addr().String()+"/rpc/v0", "ReaderHandler", []interface{}{&client}, nil, re)
|
|
require.NoError(t, err)
|
|
|
|
defer closer()
|
|
|
|
read, err := client.ReadAll(context.TODO(), strings.NewReader("pooooootato"))
|
|
require.NoError(t, err)
|
|
require.Equal(t, "pooooootato", string(read), "potatoes weren't equal")
|
|
}
|
|
|
|
func TestNullReaderProxy(t *testing.T) {
|
|
var client struct {
|
|
ReadAll func(ctx context.Context, r io.Reader) ([]byte, error)
|
|
ReadNullLen func(ctx context.Context, r io.Reader) (int64, error)
|
|
}
|
|
|
|
serverHandler := &ReaderHandler{}
|
|
|
|
readerHandler, readerServerOpt := ReaderParamDecoder()
|
|
rpcServer := jsonrpc.NewServer(readerServerOpt)
|
|
rpcServer.Register("ReaderHandler", serverHandler)
|
|
|
|
mux := mux.NewRouter()
|
|
mux.Handle("/rpc/v0", rpcServer)
|
|
mux.Handle("/rpc/streams/v0/push/{uuid}", readerHandler)
|
|
|
|
testServ := httptest.NewServer(mux)
|
|
defer testServ.Close()
|
|
|
|
re := ReaderParamEncoder("http://" + testServ.Listener.Addr().String() + "/rpc/streams/v0/push")
|
|
closer, err := jsonrpc.NewMergeClient("ws://"+testServ.Listener.Addr().String()+"/rpc/v0", "ReaderHandler", []interface{}{&client}, nil, re)
|
|
require.NoError(t, err)
|
|
|
|
defer closer()
|
|
|
|
n, err := client.ReadNullLen(context.TODO(), sealing.NewNullReader(1016))
|
|
require.NoError(t, err)
|
|
require.Equal(t, int64(1016), n)
|
|
}
|