Add tests

chiru-labs · Feb 2, 2024 · b679127 · b679127
1 parent 3d1bcdc
commit b679127
Showing 1 changed file with 28 additions and 1 deletion.
diff --git a/test/extensions/ERC721ASpot.test.js b/test/extensions/ERC721ASpot.test.js
@@ -137,13 +137,37 @@ describe('ERC721ASpot', function () {
       );
     });
 
-    it('sequential mints exceeding limit reverts', async function () {
+    it('reverts if sequential mint exceeds limit', async function () {
       await expect(this.erc721aSpot.safeMint(this.addr1.address, 11)).to.be.revertedWith(
         'SequentialMintExceedsLimit'
       );
       await this.erc721aSpot.safeMint(this.addr1.address, 10);
     });
 
+    it('reverts if _mintSpot tokenId is too small', async function () {
+      await expect(this.erc721aSpot.safeMintSpot(this.addr1.address, 19)).to.be.revertedWith(
+        'SpotMintTokenIdTooSmall'
+      );
+    });
+
+    context('with transfers', function () {
+      beforeEach(async function () {
+        await this.erc721aSpot.safeMint(this.addr1.address, 5);
+      });
+
+      it('reverts if token is not minted', async function () {
+        await expect(this.erc721aSpot
+          .connect(this.addr1)
+          .transferFrom(this.addr1.address, this.owner.address, 21)).to.be.revertedWith(
+            'OwnerQueryForNonexistentToken'
+          );
+        await this.erc721aSpot.safeMintSpot(this.addr1.address, 21);
+        await this.erc721aSpot
+          .connect(this.addr1)
+          .transferFrom(this.addr1.address, this.owner.address, 21);
+      });
+    });
+
     context('with burns', function () {
       beforeEach(async function () {
         await this.erc721aSpot.safeMint(this.addr1.address, 5);
@@ -210,19 +234,22 @@ describe('ERC721ASpot', function () {
       it('forwards extraData after burn and re-mint', async function () {
         await this.erc721aSpot.setExtraDataAt(20, 123);
         let explicitOwnership = await this.erc721aSpot.explicitOwnershipOf(20);
+        expect(await this.erc721aSpot.exists(20)).to.eq(true);
         expect(explicitOwnership.addr).to.eq(this.addr1.address);
         expect(explicitOwnership.burned).to.eq(false);
         expect(explicitOwnership.extraData).to.eq(123);
         expect(await this.erc721aSpot.exists(20)).to.eq(true);
 
         await this.erc721aSpot.connect(this.addr1).burn(20);
+        expect(await this.erc721aSpot.exists(20)).to.eq(false);
         explicitOwnership = await this.erc721aSpot.explicitOwnershipOf(20);
         expect(explicitOwnership.addr).to.eq(this.addr1.address);
         expect(explicitOwnership.burned).to.eq(true);
         expect(explicitOwnership.extraData).to.eq(123);
         expect(await this.erc721aSpot.exists(20)).to.eq(false);
 
         this.erc721aSpot.safeMintSpot(this.owner.address, 20);
+        expect(await this.erc721aSpot.exists(20)).to.eq(true);
         explicitOwnership = await this.erc721aSpot.explicitOwnershipOf(20);
         expect(explicitOwnership.addr).to.eq(this.owner.address);
         expect(explicitOwnership.burned).to.eq(false);