昔念
83ecb90baf
- 更新 README.md 中的项目结构说明 - 添加 pprof 性能分析工具的使用说明 - 更新 build.bat 文件，增加 proto 文件编译命令 - 升级 go-logr/logr 依赖至 v1.3.0 - 降级 go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc 依赖至 v1.16.0 - 降级 go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp 依赖至 v1.16.0 - 升级 go.opentelemetry.io/otel/trace 依赖至 v1.20.0 - 移除 logic/main.go 中的冗余代码 - 重构 logic/server.go 中的 Start 函数 - 更新 login/main.go 文件
55 lines
1.4 KiB
Go
55 lines
1.4 KiB
Go
package httpio
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/gorilla/mux"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/filecoin-project/go-jsonrpc"
|
|
)
|
|
|
|
type ReaderHandler struct {
|
|
}
|
|
|
|
func (h *ReaderHandler) ReadAll(ctx context.Context, r io.Reader) ([]byte, error) {
|
|
return io.ReadAll(r)
|
|
}
|
|
|
|
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(context.Background(), "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), "potatos weren't equal")
|
|
}
|