Add drain reset function

nicklas-dohrn · Feb 19, 2025 · 82dc11c · 82dc11c
1 parent 4966266
commit 82dc11c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/pkg/egress/syslog/https_batch_test.go b/src/pkg/egress/syslog/https_batch_test.go
@@ -42,6 +42,7 @@ var _ = Describe("HTTPS_batch", func() {
 
 	BeforeEach(func() {
 		drain = newBatchMockDrain(200)
+		drain.Reset()
 		b = buildURLBinding(
 			drain.URL,
 			"test-app-id",
@@ -132,8 +133,6 @@ func newBatchMockDrain(status int) *SpyDrain {
 		Expect(err).ToNot(HaveOccurred())
 		defer r.Body.Close()
 
-		println(body)
-
 		message := &rfc5424.Message{}
 
 		messages := bytes.SplitAfter(body, []byte("\n"))

diff --git a/src/pkg/egress/syslog/https_test.go b/src/pkg/egress/syslog/https_test.go
@@ -251,6 +251,11 @@ type SpyDrain struct {
 	headers  []http.Header
 }
 
+func (d *SpyDrain) Reset() {
+	d.messages = nil
+	d.headers = nil
+}
+
 func (d *SpyDrain) appendMessage(message *rfc5424.Message) {
 	d.mu.Lock()
 	defer d.mu.Unlock()