38 lines
696 B
Go
38 lines
696 B
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"go.unistack.org/micro/v3/codec"
|
|
)
|
|
|
|
type testHook struct {
|
|
f bool
|
|
}
|
|
|
|
func (t *testHook) Publish(fn FuncPublish) FuncPublish {
|
|
return func(ctx context.Context, msg Message, opts ...PublishOption) error {
|
|
t.f = true
|
|
return fn(ctx, msg, opts...)
|
|
}
|
|
}
|
|
|
|
func TestNoopHook(t *testing.T) {
|
|
h := &testHook{}
|
|
|
|
c := NewClient(Codec("application/octet-stream", codec.NewCodec()), Hooks(HookPublish(h.Publish)))
|
|
|
|
if err := c.Init(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := c.Publish(context.TODO(), c.NewMessage("", nil, MessageContentType("application/octet-stream"))); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !h.f {
|
|
t.Fatal("hook not works")
|
|
}
|
|
}
|