Add unit test

Signed-off-by: Nghia Tran <[email protected]>
chainguard-dev · Jun 19, 2024 · f9da13c · f9da13c
1 parent b8d2b35
commit f9da13c
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/logger_test.go b/logger_test.go
@@ -113,3 +113,16 @@ func TestLoggerPC(t *testing.T) {
 		t.Errorf("want %v, got %v", want, got)
 	}
 }
+
+func TestWith(t *testing.T) {
+	ctx := context.WithValue(context.Background(), "test", "test")
+	log := NewLoggerWithContext(ctx, nil)
+	withed := log.With("a", "b")
+	if want := withed.ctx; want != ctx {
+		t.Errorf("want %v, got %v", want, ctx)
+	}
+	withed = log.WithGroup("a")
+	if want := withed.ctx; want != ctx {
+		t.Errorf("want %v, got %v", want, ctx)
+	}
+}