From f9ecb9b0568f19049b1aee707c7c9185e983693c Mon Sep 17 00:00:00 2001 From: Vasiliy Tolstov Date: Tue, 3 May 2022 16:05:18 +0300 Subject: [PATCH] register: improve coverage Signed-off-by: Vasiliy Tolstov --- register/context_test.go | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/register/context_test.go b/register/context_test.go index e6160dad..614a0b08 100644 --- a/register/context_test.go +++ b/register/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, NewRegister()) + + c, ok := FromContext(ctx) + if c == nil || !ok { + t.Fatal("NewContext not works") + } +} + func TestFromContext(t *testing.T) { ctx := context.WithValue(context.TODO(), registerKey{}, NewRegister())