81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
|
package validator
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/unistack-org/micro/v3/client"
|
||
|
|
"github.com/unistack-org/micro/v3/codec"
|
||
|
|
"github.com/unistack-org/micro/v3/server"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Handler struct {
|
||
|
|
t *testing.T
|
||
|
|
}
|
||
|
|
|
||
|
|
type Message struct {
|
||
|
|
Name string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Message) Validate() error {
|
||
|
|
return fmt.Errorf("SSS")
|
||
|
|
if len(m.Name) == 0 || m.Name != "test" {
|
||
|
|
return fmt.Errorf("name is empty")
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h *Handler) Sub(ctx context.Context, req *Message) error {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestValidator(t *testing.T) {
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
defer cancel()
|
||
|
|
|
||
|
|
// create server
|
||
|
|
srv := server.NewServer(
|
||
|
|
server.Name("helloworld"),
|
||
|
|
server.Codec("application/json", codec.NewCodec()),
|
||
|
|
server.WrapHandler(NewServerHandlerWrapper()),
|
||
|
|
server.WrapSubscriber(NewServerSubscriberWrapper()),
|
||
|
|
server.Context(ctx),
|
||
|
|
)
|
||
|
|
|
||
|
|
h := &Handler{t: t}
|
||
|
|
|
||
|
|
if err := srv.Subscribe(srv.NewSubscriber("test", h.Sub)); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := srv.Init(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// start server
|
||
|
|
if err := srv.Start(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
cli := client.NewClient(
|
||
|
|
client.ContentType("application/json"),
|
||
|
|
client.Codec("application/json", codec.NewCodec()),
|
||
|
|
client.Wrap(NewClientWrapper()),
|
||
|
|
)
|
||
|
|
|
||
|
|
if err := cli.Publish(ctx, cli.NewMessage("test", &Message{Name: "test1"}, client.WithMessageContentType("application/json"))); err == nil {
|
||
|
|
t.Fatalf("validator not works as message with bad contents")
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := cli.Publish(ctx, cli.NewMessage("test", &Message{Name: "test"}, client.WithMessageContentType("application/json"))); err == nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// stop server
|
||
|
|
if err := srv.Stop(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|