store msg in c.Message instead of c.data[MessagKey]

aserto-dev · Jul 30, 2024 · e65200c · e65200c
1 parent 975c0f7
commit e65200c
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/errors.go b/errors.go
@@ -136,10 +136,10 @@ func (e *AsertoError) Msg(message string) *AsertoError {
 	c := e.Copy()
 
 	if message != "" {
-		if existingMsg, ok := c.data[MessageKey]; ok {
-			c.data[MessageKey] = existingMsg + colon + message
+		if c.Message != "" {
+			c.Message = c.Message + colon + message
 		} else {
-			c.data[MessageKey] = message
+			c.Message = message
 		}
 	}
 
@@ -151,11 +151,14 @@ func (e *AsertoError) Msgf(message string, args ...interface{}) *AsertoError {
 
 	message = fmt.Sprintf(message, args...)
 
-	if existingMsg, ok := c.data[MessageKey]; ok {
-		c.data[MessageKey] = existingMsg + colon + message
-	} else {
-		c.data[MessageKey] = message
+	if message != "" {
+		if c.Message != "" {
+			c.Message = c.Message + colon + message
+		} else {
+			c.Message = message
+		}
 	}
+
 	return c
 }