NoBey/openzeppelin-contracts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add ERC1155.totalSupply that returns overall supply count (#3962)

Browse Source
This commit is contained in:
JulissaDantes
2023-01-25 15:39:02 -05:00
committed by GitHub
parent 54c31ad98b
commit e919d96ff2
3 changed files with 55 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
test/token/ERC1155/extensions/ERC1155Supply.test.js
View File

@ -1,7 +1,9 @@
const { BN } = require('@openzeppelin/test-helpers');
const { BN, constants } = require('@openzeppelin/test-helpers');
const { expect } = require('chai');
const { ZERO_ADDRESS } = constants;
const ERC1155Supply = artifacts.require('$ERC1155Supply');
contract('ERC1155Supply', function (accounts) {
@ -25,7 +27,8 @@ contract('ERC1155Supply', function (accounts) {
});
it('totalSupply', async function () {
expect(await this.token.totalSupply(firstTokenId)).to.be.bignumber.equal('0');
expect(await this.token.methods['totalSupply(uint256)'](firstTokenId)).to.be.bignumber.equal('0');
expect(await this.token.methods['totalSupply()']()).to.be.bignumber.equal('0');
});
});
@ -40,7 +43,8 @@ contract('ERC1155Supply', function (accounts) {
});
it('totalSupply', async function () {
expect(await this.token.totalSupply(firstTokenId)).to.be.bignumber.equal(firstTokenAmount);
expect(await this.token.methods['totalSupply(uint256)'](firstTokenId)).to.be.bignumber.equal(firstTokenAmount);
expect(await this.token.methods['totalSupply()']()).to.be.bignumber.equal(firstTokenAmount);
});
});
@ -60,8 +64,13 @@ contract('ERC1155Supply', function (accounts) {
});
it('totalSupply', async function () {
expect(await this.token.totalSupply(firstTokenId)).to.be.bignumber.equal(firstTokenAmount);
expect(await this.token.totalSupply(secondTokenId)).to.be.bignumber.equal(secondTokenAmount);
expect(await this.token.methods['totalSupply(uint256)'](firstTokenId)).to.be.bignumber.equal(firstTokenAmount);
expect(await this.token.methods['totalSupply(uint256)'](secondTokenId)).to.be.bignumber.equal(
secondTokenAmount,
);
expect(await this.token.methods['totalSupply()']()).to.be.bignumber.equal(
firstTokenAmount.add(secondTokenAmount),
);
});
});
});
@ -78,7 +87,8 @@ contract('ERC1155Supply', function (accounts) {
});
it('totalSupply', async function () {
expect(await this.token.totalSupply(firstTokenId)).to.be.bignumber.equal('0');
expect(await this.token.methods['totalSupply(uint256)'](firstTokenId)).to.be.bignumber.equal('0');
expect(await this.token.methods['totalSupply()']()).to.be.bignumber.equal('0');
});
});
@ -99,9 +109,20 @@ contract('ERC1155Supply', function (accounts) {
});
it('totalSupply', async function () {
expect(await this.token.totalSupply(firstTokenId)).to.be.bignumber.equal('0');
expect(await this.token.totalSupply(secondTokenId)).to.be.bignumber.equal('0');
expect(await this.token.methods['totalSupply(uint256)'](firstTokenId)).to.be.bignumber.equal('0');
expect(await this.token.methods['totalSupply(uint256)'](secondTokenId)).to.be.bignumber.equal('0');
expect(await this.token.methods['totalSupply()']()).to.be.bignumber.equal('0');
});
});
});
context('other', function () {
it('supply unaffected by no-op', async function () {
this.token.safeTransferFrom(ZERO_ADDRESS, ZERO_ADDRESS, firstTokenId, firstTokenAmount, '0x', {
from: ZERO_ADDRESS,
});
expect(await this.token.methods['totalSupply(uint256)'](firstTokenId)).to.be.bignumber.equal('0');
expect(await this.token.methods['totalSupply()']()).to.be.bignumber.equal('0');
});
});
});