crowetic/protocol
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Modify batch allowances to take an array of spenders

Browse Source
This commit is contained in:
Amir Bandeali
2019-05-26 17:19:01 -07:00
parent 4d9f2586d9
commit 543011c3de
2 changed files with 20 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
contracts/asset-proxy/contracts/src/libs/LibAssetData.sol
View File

@@ -162,7 +162,7 @@ contract LibAssetData is
/// @dev Calls getAllowance() for each element of assetData.
/// @param owner Owner of the tokens specified by assetData.
/// @param spender Address whose authority to spend is in question.
/// @param spenders Array of addresses whose authority to spend is in question for each corresponding assetData.
/// @param assetData Description of tokens, per the AssetProxy contract
/// specification.
/// @return An array of token allowances from getAllowance(), with each
@@ -170,7 +170,7 @@ contract LibAssetData is
/// input.
function getBatchAllowances(
address owner,
address spender,
address[] memory spenders,
bytes[] memory assetData
)
public
@@ -180,7 +180,7 @@ contract LibAssetData is
uint256 length = assetData.length;
allowances = new uint256[](length);
for (uint256 i = 0; i != length; i++) {
allowances[i] = getAllowance(owner, spender, assetData[i]);
allowances[i] = getAllowance(owner, spenders[i], assetData[i]);
}
return allowances;
}
@@ -210,7 +210,7 @@ contract LibAssetData is
/// @dev Calls getBatchBalances() and getBatchAllowances() for each element
/// of assetData.
/// @param owner Owner of the tokens specified by assetData.
/// @param spender Address whose authority to spend is in question.
/// @param spenders Array of addresses whose authority to spend is in question for each corresponding assetData.
/// @param assetData Description of tokens, per the AssetProxy contract
/// specification.
/// @return An array of token balances from getBalance(), and an array of
@@ -218,7 +218,7 @@ contract LibAssetData is
/// corresponding to the same-indexed element in the assetData input.
function getBatchBalancesAndAllowances(
address owner,
address spender,
address[] memory spenders,
bytes[] memory assetData
)
public
@@ -229,7 +229,7 @@ contract LibAssetData is
)
{
balances = getBatchBalances(owner, assetData);
allowances = getBatchAllowances(owner, spender, assetData);
allowances = getBatchAllowances(owner, spenders, assetData);
return (balances, allowances);
}
@@ -470,7 +470,7 @@ contract LibAssetData is
(bool success, bytes memory returnData) = tokenAddress.staticcall(ownerOfCalldata);
owner = success ? returnData.readAddress(12) : address(0);
owner = (success && returnData.length == 32) ? returnData.readAddress(12) : address(0);
return owner;
}
}

20
contracts/asset-proxy/test/lib_asset_data.ts
View File

@@ -383,10 +383,14 @@ describe('LibAssetData', () => {
await setERC20AllowanceAsync();
await setERC721AllowanceAsync();
expect(
await libAssetData.getBatchAllowances.callAsync(tokenOwnerAddress, approvedSpenderAddress, [
await libAssetData.encodeERC20AssetData.callAsync(erc20TokenAddress),
await libAssetData.encodeERC721AssetData.callAsync(erc721TokenAddress, firstERC721TokenId),
]),
await libAssetData.getBatchAllowances.callAsync(
tokenOwnerAddress,
[approvedSpenderAddress, approvedSpenderAddress],
[
await libAssetData.encodeERC20AssetData.callAsync(erc20TokenAddress),
await libAssetData.encodeERC721AssetData.callAsync(erc721TokenAddress, firstERC721TokenId),
],
),
).to.deep.equal([new BigNumber(1), new BigNumber(1)]);
});
@@ -418,9 +422,11 @@ describe('LibAssetData', () => {
it('should query balances and allowances together, from an asset data array', async () => {
await setERC20AllowanceAsync();
expect(
await libAssetData.getBatchBalancesAndAllowances.callAsync(tokenOwnerAddress, approvedSpenderAddress, [
await libAssetData.encodeERC20AssetData.callAsync(erc20TokenAddress),
]),
await libAssetData.getBatchBalancesAndAllowances.callAsync(
tokenOwnerAddress,
[approvedSpenderAddress, approvedSpenderAddress],
[await libAssetData.encodeERC20AssetData.callAsync(erc20TokenAddress)],
),
).to.deep.equal([[new BigNumber(erc20TokenTotalSupply)], [new BigNumber(1)]]);
});
});