add context tests

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

server/context_test.go

@@ -0,0 +1,46 @@
+package server
+
+import (
+	"context"
+	"testing"
+)
+
+func TestFromContext(t *testing.T) {
+	ctx := context.WithValue(context.TODO(), serverKey{}, NewServer())
+
+	c, ok := FromContext(ctx)
+	if c == nil || !ok {
+		t.Fatal("FromContext not works")
+	}
+}
+
+func TestNewContext(t *testing.T) {
+	ctx := NewContext(context.TODO(), NewServer())
+
+	c, ok := FromContext(ctx)
+	if c == nil || !ok {
+		t.Fatal("NewContext 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")
+	}
+}
+
+func TestSetSubscriberOption(t *testing.T) {
+	type key struct{}
+	o := SetSubscriberOption(key{}, "test")
+	opts := &SubscriberOptions{}
+	o(opts)
+
+	if v, ok := opts.Context.Value(key{}).(string); !ok || v == "" {
+		t.Fatal("SetSubscriberOption not works")
+	}
+}

server/noop.go

@@ -281,7 +281,7 @@ func (n *noopServer) Deregister() error {
 			if sb.Options().Context != nil {
 				cx = sb.Options().Context
 			}
-			
+
 			ncx := cx
 			wg.Add(1)
 			go func(s broker.Subscriber) {

server/noop_test.go

@@ -10,6 +10,7 @@ import (
 	"go.unistack.org/micro/v3/codec"
 	"go.unistack.org/micro/v3/metadata"
 	"go.unistack.org/micro/v3/server"
+	"go.unistack.org/micro/v3/logger"
 )
 
 type TestHandler struct {
@@ -50,6 +51,7 @@ func TestNoopSub(t *testing.T) {
 		t.Fatal(err)
 	}
 
+	logger.DefaultLogger.Init(logger.WithLevel(logger.ErrorLevel))
 	s := server.NewServer(
 		server.Broker(b),
 		server.Codec("application/octet-stream", codec.NewCodec()),