From 32f8d86de73f43ec5ec15950794236fe350a6e90 Mon Sep 17 00:00:00 2001
From: ron <yrong1997@gmail.com>
Date: Tue, 9 Apr 2024 13:25:10 +0800
Subject: [PATCH] More test

---
 contracts/test/Gateway.t.sol | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/contracts/test/Gateway.t.sol b/contracts/test/Gateway.t.sol
index 9c063e6cb1..122de7584e 100644
--- a/contracts/test/Gateway.t.sol
+++ b/contracts/test/Gateway.t.sol
@@ -49,6 +49,7 @@ import {
 import {WETH9} from "canonical-weth/WETH9.sol";
 import "./mocks/GatewayUpgradeMock.sol";
 import {UD60x18, ud60x18, convert} from "prb/math/src/UD60x18.sol";
+import {ERC20} from "../src/ERC20.sol";
 
 contract GatewayTest is Test {
     // Emitted when token minted/burnt/transfered
@@ -938,15 +939,23 @@ contract GatewayTest is Test {
     function testAgentMintDot() public {
         testAgentRegisterDot();
 
+        uint256 amount = 1000;
+
         AgentExecuteParams memory params = AgentExecuteParams({
             agentID: assetHubAgentID,
-            payload: abi.encode(AgentExecuteCommand.MintToken, abi.encode(bytes32(uint256(1)), account1, 1000))
+            payload: abi.encode(AgentExecuteCommand.MintToken, abi.encode(bytes32(uint256(1)), account1, amount))
         });
 
         vm.expectEmit(true, true, false, false);
         emit Transfer(address(0), account1, 1000);
 
         GatewayMock(address(gateway)).agentExecutePublic(abi.encode(params));
+
+        address dotToken = GatewayMock(address(gateway)).tokenAddressOf(dotTokenID);
+
+        uint256 balance = ERC20(dotToken).balanceOf(account1);
+
+        assertEq(balance, amount);
     }
 
     function testSendRelayTokenToAssetHub() public {