PTT test cases added to the token transfer scenario (#15486)

smartcontractkit · Dec 3, 2024 · 8971282 · 8971282
1 parent 86d5f99
commit 8971282
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 2 deletions.
diff --git a/deployment/ccip/changeset/test_helpers.go b/deployment/ccip/changeset/test_helpers.go
@@ -1107,6 +1107,7 @@ func TransferAndWaitForSuccess(
 	receiver common.Address,
 	data []byte,
 	expectedStatus int,
+	extraArgs []byte,
 ) {
 	identifier := SourceDestPair{
 		SourceChainSelector: sourceChain,
@@ -1127,7 +1128,7 @@ func TransferAndWaitForSuccess(
 		Data:         data,
 		TokenAmounts: tokens,
 		FeeToken:     common.HexToAddress("0x0"),
-		ExtraArgs:    nil,
+		ExtraArgs:    extraArgs,
 	})
 	expectedSeqNum[identifier] = msgSentEvent.SequenceNumber
 	expectedSeqNumExec[identifier] = []uint64{msgSentEvent.SequenceNumber}

diff --git a/integration-tests/smoke/ccip/ccip_token_transfer_test.go b/integration-tests/smoke/ccip/ccip_token_transfer_test.go
@@ -103,6 +103,7 @@ func TestTokenTransfer(t *testing.T) {
 		tokenAmounts           []router.ClientEVMTokenAmount
 		receiver               common.Address
 		data                   []byte
+		extraData              []byte
 		expectedTokenBalances  map[common.Address]*big.Int
 		expectedExecutionState int
 	}{
@@ -156,13 +157,59 @@ func TestTokenTransfer(t *testing.T) {
 					Amount: oneE18,
 				},
 			},
-			receiver: state.Chains[sourceChain].Receiver.Address(),
+			receiver:  state.Chains[sourceChain].Receiver.Address(),
+			extraData: changeset.MakeEVMExtraArgsV2(300_000, false),
 			expectedTokenBalances: map[common.Address]*big.Int{
 				selfServeSrcToken.Address(): new(big.Int).Add(oneE18, oneE18),
 				srcToken.Address():          oneE18,
 			},
 			expectedExecutionState: changeset.EXECUTION_STATE_SUCCESS,
 		},
+		{
+			name:     "Sending token transfer with custom gasLimits to the EOA is successful",
+			srcChain: destChain,
+			dstChain: sourceChain,
+			tokenAmounts: []router.ClientEVMTokenAmount{
+				{
+					Token:  selfServeDestToken.Address(),
+					Amount: oneE18,
+				},
+				{
+					Token:  destToken.Address(),
+					Amount: new(big.Int).Add(oneE18, oneE18),
+				},
+			},
+			receiver:  utils.RandomAddress(),
+			extraData: changeset.MakeEVMExtraArgsV2(1, false),
+			expectedTokenBalances: map[common.Address]*big.Int{
+				selfServeSrcToken.Address(): oneE18,
+				srcToken.Address():          new(big.Int).Add(oneE18, oneE18),
+			},
+			expectedExecutionState: changeset.EXECUTION_STATE_SUCCESS,
+		},
+		{
+			name:     "Sending PTT with too low gas limit leads to the revert when receiver is a contract",
+			srcChain: destChain,
+			dstChain: sourceChain,
+			tokenAmounts: []router.ClientEVMTokenAmount{
+				{
+					Token:  selfServeDestToken.Address(),
+					Amount: oneE18,
+				},
+				{
+					Token:  destToken.Address(),
+					Amount: oneE18,
+				},
+			},
+			receiver:  state.Chains[sourceChain].Receiver.Address(),
+			data:      []byte("this should be reverted because gasLimit is too low, no tokens are transferred as well"),
+			extraData: changeset.MakeEVMExtraArgsV2(1, false),
+			expectedTokenBalances: map[common.Address]*big.Int{
+				selfServeSrcToken.Address(): big.NewInt(0),
+				srcToken.Address():          big.NewInt(0),
+			},
+			expectedExecutionState: changeset.EXECUTION_STATE_FAILURE,
+		},
 	}
 
 	for _, tt := range tcs {
@@ -184,6 +231,7 @@ func TestTokenTransfer(t *testing.T) {
 				tt.receiver,
 				tt.data,
 				tt.expectedExecutionState,
+				tt.extraData,
 			)
 
 			for token, balance := range tt.expectedTokenBalances {

diff --git a/integration-tests/smoke/ccip/ccip_usdc_test.go b/integration-tests/smoke/ccip/ccip_usdc_test.go
@@ -201,6 +201,7 @@ func TestUSDCTokenTransfer(t *testing.T) {
 				tt.receiver,
 				tt.data,
 				tt.expectedExecutionState,
+				nil,
 			)
 
 			for token, balance := range tt.expectedTokenBalances {