From fa7f5732491934ebcb5a708b08124960f8da4fd0 Mon Sep 17 00:00:00 2001
From: Vihas Makwana <vihas.makwana@elastic.co>
Date: Wed, 11 Dec 2024 15:32:24 +0530
Subject: [PATCH] chore: add options for retry/storage

---
 testbed/datasenders/stanza.go | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/testbed/datasenders/stanza.go b/testbed/datasenders/stanza.go
index 1dcfa60fe3bc..d056075e8c09 100644
--- a/testbed/datasenders/stanza.go
+++ b/testbed/datasenders/stanza.go
@@ -20,7 +20,9 @@ import (
 )
 
 type FileLogWriter struct {
-	file *os.File
+	file    *os.File
+	retry   string
+	storage string
 }
 
 // Ensure FileLogWriter implements LogDataSender.
@@ -41,6 +43,16 @@ func NewFileLogWriter() *FileLogWriter {
 	return f
 }
 
+func (f *FileLogWriter) WithRetry(retry string) *FileLogWriter {
+	f.retry = retry
+	return f
+}
+
+func (f *FileLogWriter) WithStorage(storage string) *FileLogWriter {
+	f.storage = storage
+	return f
+}
+
 func (f *FileLogWriter) Capabilities() consumer.Capabilities {
 	return consumer.Capabilities{MutatesData: false}
 }
@@ -120,7 +132,9 @@ func (f *FileLogWriter) GenConfigYAMLStr() string {
           layout: '%%Y-%%m-%%d'
         severity:
           parse_from: attributes.sev
-`, f.file.Name())
+    %s
+    %s
+`, f.file.Name(), f.retry, f.storage)
 }
 
 func (f *FileLogWriter) ProtocolName() string {