From d8fe2ff8b4baba36d8d9447e1dbe83ecaf19aaed Mon Sep 17 00:00:00 2001 From: Vasiliy Tolstov Date: Thu, 20 Jan 2022 15:29:21 +0300 Subject: [PATCH] add logger context test Signed-off-by: Vasiliy Tolstov --- logger/logger_test.go | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/logger/logger_test.go b/logger/logger_test.go index 82cadb3a..dd61487e 100644 --- a/logger/logger_test.go +++ b/logger/logger_test.go @@ -7,6 +7,24 @@ import ( "testing" ) +func TestContext(t *testing.T) { + ctx := context.TODO() + buf := bytes.NewBuffer(nil) + l := NewLogger(WithLevel(TraceLevel), WithOutput(buf)) + if err := l.Init(); err != nil { + t.Fatal(err) + } + + nl, ok := FromContext(NewContext(ctx, l.Fields("key", "val"))) + if !ok { + t.Fatal("context without logger") + } + nl.Info(ctx, "message") + if !bytes.Contains(buf.Bytes(), []byte(`"key":"val"`)) { + t.Fatalf("logger fields not works, buf contains: %s", buf.Bytes()) + } +} + func TestFields(t *testing.T) { ctx := context.TODO() buf := bytes.NewBuffer(nil)