fix rpc GenerateTx

ava-labs · Apr 25, 2024 · 4b07f8c · 4b07f8c
1 parent 93c5b3b
commit 4b07f8c
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/rpc/jsonrpc_client.go b/rpc/jsonrpc_client.go
@@ -132,11 +132,18 @@ func (cli *JSONRPCClient) GenerateTransaction(
 		return nil, nil, 0, err
 	}
 
-	maxUnits, err := chain.EstimateMaxUnits(parser.Rules(time.Now().UnixMilli()), actions, authFactory)
-	if err != nil {
-		return nil, nil, 0, err
+	var totalUnits fees.Dimensions
+	for _, action := range actions {
+		maxUnits, err := chain.EstimateMaxUnits(parser.Rules(time.Now().UnixMilli()), actions, authFactory)
+		if err != nil {
+			return nil, nil, 0, err
+		}
+		totalUnits, err = fees.Add(totalUnits, maxUnits)
+		if err != nil {
+			return nil, nil, 0, err
+		}
 	}
-	maxFee, err := fees.MulSum(unitPrices, maxUnits)
+	maxFee, err := fees.MulSum(unitPrices, totalUnits)
 	if err != nil {
 		return nil, nil, 0, err
 	}