Vasiliy Tolstov
ffa01de78f
* remove subscribe from server * remove publish from client * broker package refactoring Co-authored-by: vtolstov <vtolstov@users.noreply.github.com> Reviewed-on: #396 Co-authored-by: Vasiliy Tolstov <v.tolstov@unistack.org> Co-committed-by: Vasiliy Tolstov <v.tolstov@unistack.org>
62 lines
1.2 KiB
Go
62 lines
1.2 KiB
Go
package broker
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestFromContext(t *testing.T) {
|
|
ctx := context.WithValue(context.TODO(), brokerKey{}, NewBroker())
|
|
c, ok := FromContext(ctx)
|
|
if c == nil || !ok {
|
|
t.Fatal("FromContext not works")
|
|
}
|
|
}
|
|
|
|
func TestFromNilContext(t *testing.T) {
|
|
// nolint: staticcheck
|
|
c, ok := FromContext(nil)
|
|
if ok || c != nil {
|
|
t.Fatal("FromContext not works")
|
|
}
|
|
}
|
|
|
|
func TestNewContext(t *testing.T) {
|
|
ctx := NewContext(context.TODO(), NewBroker())
|
|
c, ok := FromContext(ctx)
|
|
if c == nil || !ok {
|
|
t.Fatal("NewContext not works")
|
|
}
|
|
}
|
|
|
|
func TestNewNilContext(t *testing.T) {
|
|
// nolint: staticcheck
|
|
ctx := NewContext(nil, NewBroker())
|
|
c, ok := FromContext(ctx)
|
|
if c == nil || !ok {
|
|
t.Fatal("NewContext not works")
|
|
}
|
|
}
|
|
|
|
func TestSetSubscribeOption(t *testing.T) {
|
|
type key struct{}
|
|
o := SetSubscribeOption(key{}, "test")
|
|
opts := &SubscribeOptions{}
|
|
o(opts)
|
|
|
|
if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" {
|
|
t.Fatal("SetSubscribeOption not works")
|
|
}
|
|
}
|
|
|
|
func TestSetOption(t *testing.T) {
|
|
type key struct{}
|
|
o := SetOption(key{}, "test")
|
|
opts := &Options{}
|
|
o(opts)
|
|
|
|
if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" {
|
|
t.Fatal("SetOption not works")
|
|
}
|
|
}
|