Add test for allocations

hashicorp · Mar 5, 2024 · fb7680a · fb7680a
1 parent 64b4fbc
commit fb7680a
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/memory_test.go b/memory_test.go
@@ -25,3 +25,28 @@ func BenchmarkLoggerMemory(b *testing.B) {
 		}
 	}
 }
+
+func TestLoggerMemory(t *testing.T) {
+	var buf bytes.Buffer
+
+	logger := New(&LoggerOptions{
+		Name:   "test",
+		Output: &buf,
+		Level:  Info,
+	})
+
+	avg := testing.AllocsPerRun(100, func() {
+		logger.Trace("this is some message",
+			"name", "foo",
+			"what", "benchmarking yourself",
+		)
+
+		if buf.Len() != 0 {
+			panic("oops")
+		}
+	})
+
+	if avg != 0 {
+		t.Fatalf("ignored logs allocated")
+	}
+}