From 1003403ca288437265fe4387d29b26035df07685 Mon Sep 17 00:00:00 2001
From: Mario Rodriguez Molins <mario.rodriguez@elastic.co>
Date: Mon, 12 Aug 2024 11:07:37 +0200
Subject: [PATCH] Allow to read pipeline trace value as string or slice of
 strings

---
 internal/testrunner/runners/system/tester.go | 33 ++++++++++++++++----
 1 file changed, 27 insertions(+), 6 deletions(-)

diff --git a/internal/testrunner/runners/system/tester.go b/internal/testrunner/runners/system/tester.go
index dc16c99b8..2c6713d18 100644
--- a/internal/testrunner/runners/system/tester.go
+++ b/internal/testrunner/runners/system/tester.go
@@ -842,14 +842,35 @@ type scenarioTest struct {
 	startTestTime      time.Time
 }
 
+type pipelineTrace []string
+
+func (p *pipelineTrace) UnmarshalJSON(d []byte) error {
+	var alias interface{}
+	if err := json.Unmarshal(d, &alias); err != nil {
+		return err
+	}
+	switch v := alias.(type) {
+	case string:
+		*p = append(*p, v)
+	case []any:
+		// asume it is going to be an array of strings
+		for _, value := range v {
+			*p = append(*p, fmt.Sprint(value))
+		}
+	default:
+		return fmt.Errorf("unexpected type found for pipeline_trace: %T", v)
+	}
+	return nil
+}
+
 type failureStoreDocument struct {
 	Error struct {
-		Type          string   `json:"type"`
-		Message       string   `json:"message"`
-		StackTrace    string   `json:"stack_trace"`
-		PipelineTrace []string `json:"pipeline_trace"`
-		Pipeline      string   `json:"pipeline"`
-		ProcessorType string   `json:"processor_type"`
+		Type          string        `json:"type"`
+		Message       string        `json:"message"`
+		StackTrace    string        `json:"stack_trace"`
+		PipelineTrace pipelineTrace `json:"pipeline_trace"`
+		Pipeline      string        `json:"pipeline"`
+		ProcessorType string        `json:"processor_type"`
 	} `json:"error"`
 }