diff --git a/router/context_test.go b/router/context_test.go index 2f330352..f16a172d 100644 --- a/router/context_test.go +++ b/router/context_test.go @@ -5,6 +5,24 @@ import ( "testing" ) +func TestFromNilContext(t *testing.T) { + // nolint: staticcheck + c, ok := FromContext(nil) + if ok || c != nil { + t.Fatal("FromContext not works") + } +} + +func TestNewNilContext(t *testing.T) { + // nolint: staticcheck + ctx := NewContext(nil, NewRouter()) + + c, ok := FromContext(ctx) + if c == nil || !ok { + t.Fatal("NewContext not works") + } +} + func TestFromContext(t *testing.T) { ctx := context.WithValue(context.TODO(), routerKey{}, NewRouter())