add tests

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>

validator_test.go

@@ -0,0 +1,80 @@
+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)
+	}
+
+}