Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: adds tests that verify IHRC904 rejectTokens targeted to an EOA are functional when called by a contract #1198

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
+359 −0
Open

feat: adds tests that verify IHRC904 rejectTokens targeted to an EOA are functional when called by a contract #1198

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
178c592
feat: added test coverage for TokenReject.sol
simzzz Jan 21, 2025
6c76a80
Merge branch 'main' into 1190-reject-tokens-coverage
simzzz Jan 27, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions test/constants.js
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -204,12 +204,15 @@ const Contract = {
CancunOpcodes: 'CancunOpcodes',
KZGPointEvaluation: 'KZGPointEvaluation',
StateRegistry: 'StateRegistry',
Airdrop: 'Airdrop',
TokenReject: 'TokenReject',
AliasAccountUtility: 'AliasAccountUtility',
};

const CALL_EXCEPTION = 'CALL_EXCEPTION';
const CONTRACT_REVERT_EXECUTED_CODE = 3;
const GAS_LIMIT_1_000_000 = { gasLimit: 1_000_000 };
const GAS_LIMIT_2_000_000 = { gasLimit: 2_000_000 };
const GAS_LIMIT_10_000_000 = { gasLimit: 10_000_000 };
const GAS_LIMIT_800000 = { gasLimit: 800000 };
const GAS_LIMIT_8000000 = { gasLimit: 8000000 };
Expand All @@ -223,6 +226,7 @@ const HOUR = 60 * MINUTE;
const DAY = 24 * HOUR;
const WEEK = 7 * DAY;
const GWEI = 1e9;
const HTS_SYSTEM_CONTRACT_ID = '0.0.359';

module.exports = {
Events,
Expand All @@ -231,6 +235,7 @@ module.exports = {
CALL_EXCEPTION,
CONTRACT_REVERT_EXECUTED_CODE,
GAS_LIMIT_1_000_000,
GAS_LIMIT_2_000_000,
GAS_LIMIT_10_000_000,
GAS_LIMIT_800000,
GAS_LIMIT_8000000,
Expand All @@ -245,4 +250,5 @@ module.exports = {
WEEK,
WEI,
GWEI,
HTS_SYSTEM_CONTRACT_ID,
};
295 changes: 295 additions & 0 deletions test/system-contracts/hedera-token-service/hrc-904/TokenRejectContract.js
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,295 @@
const { expect } = require('chai');
const { ethers } = require('hardhat');
const utils = require('../utils');
const Constants = require('../../../constants');

describe('HIP904 TokenRejectContract Test Suite', function () {
let tokenRejectContract;
let tokenCreateContract;
let airdropContract;
let signers;
let owner;
let receiver;
let walletIHRC904AccountFacade;

async function setupToken() {
const tokenAddress =
await utils.createFungibleTokenWithSECP256K1AdminKeyWithoutKYC(
tokenCreateContract,
owner,
utils.getSignerCompressedPublicKey()
);

await utils.updateTokenKeysViaHapi(tokenAddress, [
await airdropContract.getAddress(),
await tokenCreateContract.getAddress(),
]);

await utils.associateToken(
tokenCreateContract,
tokenAddress,
Constants.Contract.TokenCreateContract
);

return tokenAddress;
}

async function setupNft() {
const nftTokenAddress =
await utils.createNonFungibleTokenWithSECP256K1AdminKeyWithoutKYC(
tokenCreateContract,
owner,
utils.getSignerCompressedPublicKey()
);

await utils.updateTokenKeysViaHapi(
nftTokenAddress,
[
await airdropContract.getAddress(),
await tokenCreateContract.getAddress(),
],
true,
true,
false,
true,
true,
true,
false
);

await utils.associateToken(
tokenCreateContract,
nftTokenAddress,
Constants.Contract.TokenCreateContract
);

return nftTokenAddress;
}

before(async function () {
signers = await ethers.getSigners();
tokenRejectContract = await utils.deployContract(
Constants.Contract.TokenReject
);
tokenCreateContract = await utils.deployContract(
Constants.Contract.TokenCreateContract
);
airdropContract = await utils.deployContract(Constants.Contract.Airdrop);
owner = signers[0].address;

const receiverPrivateKey = ethers.hexlify(ethers.randomBytes(32));
receiver = new ethers.Wallet(receiverPrivateKey).connect(ethers.provider);

await signers[0].sendTransaction({
to: receiver.address,
value: ethers.parseEther('100'),
});

await utils.updateAccountKeysViaHapi([
await tokenRejectContract.getAddress(),
await tokenCreateContract.getAddress(),
await airdropContract.getAddress(),
]);

await utils.updateAccountKeysViaHapi(
[
await tokenRejectContract.getAddress(),
await tokenCreateContract.getAddress(),
await airdropContract.getAddress(),
],
[receiverPrivateKey]
);

const IHRC904AccountFacade = new ethers.Interface(
(await hre.artifacts.readArtifact('IHRC904AccountFacade')).abi
);

walletIHRC904AccountFacade = new ethers.Contract(
receiver.address,
IHRC904AccountFacade,
receiver
);
});

it('should reject tokens for a single account', async function () {
const tokenAddress = await setupToken();
const receiver = signers[1];

const ftAmount = BigInt(1);
const airdropTx = await airdropContract.tokenAirdrop(
tokenAddress,
owner,
receiver.address,
ftAmount,
{
value: BigInt('850000000000000000'),
gasLimit: 2_000_000,
}
);
await airdropTx.wait();

await walletIHRC904AccountFacade.setUnlimitedAutomaticAssociations(true, {
gasLimit: 2_000_000,
});

const tx = await tokenRejectContract.rejectTokens(
receiver.address,
[tokenAddress],
[],
Constants.GAS_LIMIT_2_000_000
);
const responseCode = await utils.getHTSResponseCode(tx.hash);
expect(responseCode).to.eq('22'); // SUCCESS code
});

it('should reject NFTs for a single account', async function () {
const nftTokenAddress = await setupNft();
const receiver = signers[1];

const serial = utils.mintNFTToAddress(tokenCreateContract, nftTokenAddress);

const airdropTx = await airdropContract.nftAirdrop(
nftTokenAddress,
owner,
receiver.address,
serial,
{
value: BigInt('850000000000000000'),
gasLimit: 2_000_000,
}
);
await airdropTx.wait();

await walletIHRC904AccountFacade.setUnlimitedAutomaticAssociations(true, {
gasLimit: 2_000_000,
});

const tx = await tokenRejectContract.rejectTokens(
receiver.address,
[],
[nftTokenAddress],
Constants.GAS_LIMIT_2_000_000
);
const responseCode = await utils.getHTSResponseCode(tx.hash);
expect(responseCode).to.eq('22'); // SUCCESS code
});

it('should reject tokens for multiple accounts', async function () {
const tokenAddress = await setupToken();
const receivers = signers.slice(1, 3);

for (const receiver of receivers) {
const airdropTx = await airdropContract.tokenAirdrop(
tokenAddress,
owner,
receiver.address,
BigInt(1),
{
value: BigInt('850000000000000000'),
gasLimit: 2_000_000,
}
);
await airdropTx.wait();

const tx = await tokenRejectContract.rejectTokens(
receiver.address,
[tokenAddress],
[],
Constants.GAS_LIMIT_2_000_000
);
const responseCode = await utils.getHTSResponseCode(tx.hash);
expect(responseCode).to.eq('22'); // SUCCESS code
}
});

it('should fail when sender does not have any associated tokens', async function () {
const tokenAddress = await setupToken();

await walletIHRC904AccountFacade.setUnlimitedAutomaticAssociations(false, {
gasLimit: 2_000_000,
});

const airdropTx = await airdropContract.tokenAirdrop(
tokenAddress,
owner,
receiver.address,
BigInt(1),
{
value: BigInt('850000000000000000'),
gasLimit: 2_000_000,
}
);
await airdropTx.wait();

const tx = await tokenRejectContract.rejectTokens(
receiver.address,
[tokenAddress],
[],
Constants.GAS_LIMIT_2_000_000
);
const responseCode = await utils.getHTSResponseCode(tx.hash);
expect(responseCode).to.eq('184'); // TOKEN_NOT_ASSOCIATED_TO_ACCOUNT code
});

it('should fail when sender does not have a pending airdrop', async function () {
const tokenAddress = await setupToken();
const receiver = signers[1];

const tx = await tokenRejectContract.rejectTokens(
receiver.address,
[tokenAddress],
[],
Constants.GAS_LIMIT_2_000_000
);
const responseCode = await utils.getHTSResponseCode(tx.hash);
expect(responseCode).to.eq('178'); // INSUFFICIENT_TOKEN_BALANCE code
});

it('should fail when provided fungible token is invalid', async function () {
const invalidToken = ethers.Wallet.createRandom().address;
const nftTokenAddress = await setupNft();

const tx = await tokenRejectContract.rejectTokens(
receiver.address,
[invalidToken],
[nftTokenAddress],
Constants.GAS_LIMIT_2_000_000
);
const responseCode = await utils.getHTSResponseCode(tx.hash);
expect(responseCode).to.eq('167'); // INVALID_TOKEN_ID code
});

it('should fail when provided NFT is invalid', async function () {
const invalidNft = ethers.Wallet.createRandom().address;

const nftTokenAddress = await setupNft();
const receiver = signers[1];

const serial = utils.mintNFTToAddress(tokenCreateContract, nftTokenAddress);

const airdropTx = await airdropContract.nftAirdrop(
nftTokenAddress,
owner,
receiver.address,
serial,
{
value: BigInt('850000000000000000'),
gasLimit: 2_000_000,
}
);
await airdropTx.wait();

await walletIHRC904AccountFacade.setUnlimitedAutomaticAssociations(true, {
gasLimit: 2_000_000,
});

const tx = await tokenRejectContract.rejectTokens(
receiver.address,
[],
[invalidNft],
Constants.GAS_LIMIT_2_000_000
);
const responseCode = await utils.getHTSResponseCode(tx.hash);
expect(responseCode).to.eq('226'); // INVALID_NFT_ID code
});
});
Loading