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