Add helper method to SimpleLogrusHook to get logged lines

grafana · Jan 17, 2023 · f3cecec · f3cecec
1 parent 2378549
commit f3cecec
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/lib/testutils/logrus_hook.go b/lib/testutils/logrus_hook.go
@@ -37,6 +37,16 @@ func (smh *SimpleLogrusHook) Drain() []logrus.Entry {
 	return res
 }
 
+// Lines returns the logged lines.
+func (smh *SimpleLogrusHook) Lines() []string {
+	entries := smh.Drain()
+	lines := make([]string, len(entries))
+	for i, entry := range entries {
+		lines[i] = entry.Message
+	}
+	return lines
+}
+
 var _ logrus.Hook = &SimpleLogrusHook{}
 
 // LogContains is a helper function that checks the provided list of log entries