From 1e8992a2043471e0d38f181963833158f530fa8f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andrei=20B=C4=83ncioiu?= <andrei.bancioiu@gmail.com>
Date: Thu, 30 May 2024 13:37:29 +0300
Subject: [PATCH] Adjust test.

---
 abi/serializer_test.go | 54 +++++++++++++++++++++++-------------------
 1 file changed, 30 insertions(+), 24 deletions(-)

diff --git a/abi/serializer_test.go b/abi/serializer_test.go
index 184a68d..f526ab1 100644
--- a/abi/serializer_test.go
+++ b/abi/serializer_test.go
@@ -551,30 +551,36 @@ func TestSerializer_InRealWorldScenarios(t *testing.T) {
 			},
 		}
 
-		action := EnumValue{
-			Fields: []Field{
-				{
-					Name:  "to",
-					Value: &actionTo,
-				},
-				{
-					Name:  "egld_amount",
-					Value: &actionEgldAmount,
-				},
-				{
-					Name: "opt_gas_limit",
-					Value: &OptionValue{
-						Value: &actionGasLimit,
-					},
-				},
-				{
-					Name:  "endpoint_name",
-					Value: &actionEndpointName,
-				},
-				{
-					Name:  "arguments",
-					Value: &actionArguments,
-				},
+		action := &EnumValue{
+			FieldsProvider: func(discriminant uint8) []Field {
+				if discriminant == 5 {
+					return []Field{
+						{
+							Name:  "to",
+							Value: actionTo,
+						},
+						{
+							Name:  "egld_amount",
+							Value: actionEgldAmount,
+						},
+						{
+							Name: "opt_gas_limit",
+							Value: &OptionValue{
+								Value: actionGasLimit,
+							},
+						},
+						{
+							Name:  "endpoint_name",
+							Value: actionEndpointName,
+						},
+						{
+							Name:  "arguments",
+							Value: actionArguments,
+						},
+					}
+				}
+
+				return nil
 			},
 		}