Add Entries() to fetch all log entries

abbaskiko · Oct 11, 2020 · 5319d9c · 5319d9c
1 parent 8d5b2f4
commit 5319d9c
Showing 1 changed file with 6 additions and 14 deletions.
diff --git a/testing/logger/capture.go b/testing/logger/capture.go
@@ -26,16 +26,10 @@ import (
 
 // LogCapture allows testing code to query log output.
 type LogCapture struct {
-	mu sync.Mutex
-	// entries []*LogEntry
+	mu      sync.Mutex
 	entries []string
 }
 
-//type LogEntry struct {
-//	Level zerolog.Level
-//	Msg   string
-//}
-
 // NewLogCapture captures logs for querying.
 func NewLogCapture() *LogCapture {
 	c := &LogCapture{
@@ -51,10 +45,6 @@ func (c *LogCapture) Run(e *zerolog.Event, level zerolog.Level, msg string) {
 	c.mu.Lock()
 	defer c.mu.Unlock()
 	c.entries = append(c.entries, msg)
-	//	c.entries = append(c.entries, &LogEntry{
-	//		Level: level,
-	//		Msg:   msg,
-	//	})
 	e.Discard()
 }
 
@@ -66,9 +56,11 @@ func (c *LogCapture) AssertHasEntry(t *testing.T, msg string) {
 		if c.entries[i] == msg {
 			return
 		}
-		//		if c.entries[i].Msg == msg {
-		//			return true
-		//		}
 	}
 	assert.Fail(t, fmt.Sprintf("Missing log message %q", msg), strings.Join(c.entries, "\n"))
 }
+
+// Entries returns all log entries.
+func (c *LogCapture) Entries() []string {
+	return c.entries
+}