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