Tony Chen
4 years ago
committed by
GitHub
parent
f8eeb9f388
commit
946e9ca814
@ -0,0 +1,212 @@ |
||||
package testproto |
||||
|
||||
import ( |
||||
"bytes" |
||||
context "context" |
||||
"fmt" |
||||
"net" |
||||
http "net/http" |
||||
"testing" |
||||
"time" |
||||
|
||||
"github.com/go-kratos/kratos/v2/encoding" |
||||
tr "github.com/go-kratos/kratos/v2/transport/http" |
||||
_struct "github.com/golang/protobuf/ptypes/struct" |
||||
) |
||||
|
||||
type echoService struct { |
||||
} |
||||
|
||||
func (s *echoService) Echo(ctx context.Context, m *SimpleMessage) (*SimpleMessage, error) { |
||||
return m, nil |
||||
} |
||||
|
||||
func (s *echoService) EchoBody(ctx context.Context, m *SimpleMessage) (*SimpleMessage, error) { |
||||
return m, nil |
||||
} |
||||
|
||||
func (s *echoService) EchoDelete(ctx context.Context, m *SimpleMessage) (*SimpleMessage, error) { |
||||
return m, nil |
||||
} |
||||
|
||||
func (s *echoService) EchoPatch(ctx context.Context, m *DynamicMessageUpdate) (*DynamicMessageUpdate, error) { |
||||
return m, nil |
||||
} |
||||
|
||||
func (s *echoService) EchoResponseBody(ctx context.Context, m *DynamicMessageUpdate) (*DynamicMessageUpdate, error) { |
||||
return m, nil |
||||
} |
||||
|
||||
type echoClient struct { |
||||
baseURL string |
||||
client *http.Client |
||||
} |
||||
|
||||
// post: /v1/example/echo/{id}
|
||||
func (c *echoClient) Echo(ctx context.Context, in *SimpleMessage) (out *SimpleMessage, err error) { |
||||
codec := encoding.GetCodec("json") |
||||
data, err := codec.Marshal(in) |
||||
if err != nil { |
||||
return |
||||
} |
||||
req, err := http.NewRequest("POST", fmt.Sprintf("%s/v1/example/echo/%s", c.baseURL, in.Id), bytes.NewReader(data)) |
||||
if err != nil { |
||||
return |
||||
} |
||||
req.Header.Set("content-type", "application/json") |
||||
out = new(SimpleMessage) |
||||
if err = tr.Do(c.client, req, out); err != nil { |
||||
return |
||||
} |
||||
return |
||||
} |
||||
|
||||
// post: /v1/example/echo_body
|
||||
func (c *echoClient) EchoBody(ctx context.Context, in *SimpleMessage) (out *SimpleMessage, err error) { |
||||
codec := encoding.GetCodec("json") |
||||
data, err := codec.Marshal(in) |
||||
if err != nil { |
||||
return |
||||
} |
||||
req, err := http.NewRequest("POST", fmt.Sprintf("%s/v1/example/echo_body", c.baseURL), bytes.NewReader(data)) |
||||
if err != nil { |
||||
return |
||||
} |
||||
req.Header.Set("content-type", "application/json") |
||||
out = new(SimpleMessage) |
||||
if err = tr.Do(c.client, req, out); err != nil { |
||||
return |
||||
} |
||||
return |
||||
} |
||||
|
||||
// delete: /v1/example/echo_delete/{id}/{num}
|
||||
func (c *echoClient) EchoDelete(ctx context.Context, in *SimpleMessage) (out *SimpleMessage, err error) { |
||||
req, err := http.NewRequest("DELETE", fmt.Sprintf("%s/v1/example/echo_delete/%s/%d", c.baseURL, in.Id, in.Num), nil) |
||||
if err != nil { |
||||
return |
||||
} |
||||
out = new(SimpleMessage) |
||||
if err = tr.Do(c.client, req, out); err != nil { |
||||
return |
||||
} |
||||
return |
||||
} |
||||
|
||||
// patch: /v1/example/echo_patch
|
||||
func (c *echoClient) EchoPatch(ctx context.Context, in *DynamicMessageUpdate) (out *DynamicMessageUpdate, err error) { |
||||
codec := encoding.GetCodec("json") |
||||
data, err := codec.Marshal(in.Body) |
||||
if err != nil { |
||||
return |
||||
} |
||||
req, err := http.NewRequest("PATCH", fmt.Sprintf("%s/v1/example/echo_patch", c.baseURL), bytes.NewReader(data)) |
||||
if err != nil { |
||||
return |
||||
} |
||||
req.Header.Set("content-type", "application/json") |
||||
out = new(DynamicMessageUpdate) |
||||
if err = tr.Do(c.client, req, out); err != nil { |
||||
return |
||||
} |
||||
return |
||||
} |
||||
|
||||
// post: /v1/example/echo_response_body
|
||||
func (c *echoClient) EchoResponseBody(ctx context.Context, in *DynamicMessageUpdate) (out *DynamicMessageUpdate, err error) { |
||||
codec := encoding.GetCodec("json") |
||||
data, err := codec.Marshal(in) |
||||
if err != nil { |
||||
return |
||||
} |
||||
req, err := http.NewRequest("POST", fmt.Sprintf("%s/v1/example/echo_response_body", c.baseURL), bytes.NewReader(data)) |
||||
if err != nil { |
||||
return |
||||
} |
||||
req.Header.Set("content-type", "application/json") |
||||
out = new(DynamicMessageUpdate) |
||||
if err = tr.Do(c.client, req, out.Body); err != nil { |
||||
return |
||||
} |
||||
return |
||||
} |
||||
|
||||
func TestEchoService(t *testing.T) { |
||||
s := &echoService{} |
||||
h := NewEchoServiceHandler(s) |
||||
srv := &http.Server{Addr: ":0", Handler: h} |
||||
lis, err := net.Listen("tcp", srv.Addr) |
||||
if err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
addr := lis.Addr().(*net.TCPAddr) |
||||
time.AfterFunc(time.Second, func() { |
||||
defer srv.Shutdown(context.Background()) |
||||
testEchoClient(t, fmt.Sprintf("http://127.0.0.1:%d", addr.Port)) |
||||
}) |
||||
if err := srv.Serve(lis); err != nil && err != http.ErrServerClosed { |
||||
t.Fatal(err) |
||||
} |
||||
} |
||||
|
||||
func testEchoClient(t *testing.T, baseURL string) { |
||||
var ( |
||||
err error |
||||
in = &SimpleMessage{Id: "test_id", Num: 100} |
||||
out = &SimpleMessage{} |
||||
) |
||||
check := func(in, out *SimpleMessage) { |
||||
if in.Id != out.Id || in.Num != out.Num { |
||||
t.Errorf("expected %s got %s", in, out) |
||||
} |
||||
} |
||||
|
||||
cli := &echoClient{baseURL: baseURL, client: http.DefaultClient} |
||||
|
||||
if out, err = cli.Echo(context.Background(), in); err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
check(in, out) |
||||
|
||||
if out, err = cli.EchoBody(context.Background(), in); err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
check(in, out) |
||||
|
||||
if out, err = cli.EchoDelete(context.Background(), in); err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
check(in, out) |
||||
|
||||
var ( |
||||
din = &DynamicMessageUpdate{Body: &DynamicMessage{ |
||||
ValueField: &_struct.Value{Kind: &_struct.Value_StringValue{StringValue: "test"}}, |
||||
}} |
||||
dout *DynamicMessageUpdate |
||||
) |
||||
if dout, err = cli.EchoPatch(context.Background(), din); err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
if din.Body.ValueField.GetStringValue() != dout.Body.ValueField.GetStringValue() { |
||||
t.Errorf("EchoPatch expected %s got %s", din, dout) |
||||
} |
||||
if dout, err = cli.EchoResponseBody(context.Background(), din); err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
if din.Body.ValueField.GetStringValue() != dout.Body.ValueField.GetStringValue() { |
||||
t.Errorf("EchoResponseBody expected %s got %s", din, dout) |
||||
} |
||||
} |
||||
|
||||
func TestJSON(t *testing.T) { |
||||
in := &SimpleMessage{Id: "test_id", Num: 100} |
||||
out := &SimpleMessage{} |
||||
codec := encoding.GetCodec("json") |
||||
data, err := codec.Marshal(in) |
||||
if err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
if err := codec.Unmarshal(data, out); err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
} |
||||
Loading…
Reference in new issue