From ecc1d6907b6c8c317416a4350f683a8b3407a23f Mon Sep 17 00:00:00 2001
From: Brad Fitzpatrick <bradfitz@tailscale.com>
Date: Wed, 30 Aug 2023 19:37:35 -0700
Subject: [PATCH] types/logger: add TestLogger

We have this in another repo and I wanted it here too.

Updates #cleanup

Change-Id: If93dc73f11eaaada5024acf2a885a153b88db5a0
Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
---
 types/logger/logger.go | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/types/logger/logger.go b/types/logger/logger.go
index 70f956ded..232679ba7 100644
--- a/types/logger/logger.go
+++ b/types/logger/logger.go
@@ -373,3 +373,19 @@ func (a asJSONResult) Format(s fmt.State, verb rune) {
 	}
 	s.Write(v)
 }
+
+// TBLogger is the testing.TB subset needed by TestLogger.
+type TBLogger interface {
+	Helper()
+	Logf(format string, args ...any)
+}
+
+// TestLogger returns a logger that logs to tb.Logf
+// with a prefix to make it easier to distinguish spam
+// from explicit test failures.
+func TestLogger(tb TBLogger) Logf {
+	return func(format string, args ...any) {
+		tb.Helper()
+		tb.Logf("    ... "+format, args...)
+	}
+}