Increase test coverage for bySig

1inch · Nov 30, 2024 · e2a7a26 · e2a7a26
1 parent 0fc6846
commit e2a7a26
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/bySig.ts b/src/bySig.ts
@@ -10,6 +10,7 @@ export enum NonceType {
     Account,    // Nonce for account
     Selector,   // Nonce for selector
     Unique,     // Nonce for unique
+    Invalid,    // Invalid Type
 }
 
 /**

diff --git a/test/contracts/BySig.test.ts b/test/contracts/BySig.test.ts
@@ -78,6 +78,15 @@ describe('BySig', function () {
     });
 
     describe('bySig', function () {
+        it('should return false for an invalid nonce type', async function () {
+            const { addrs: { alice }, token } = await loadFixture(deployAddressArrayMock);
+            const sig = {
+                traits: buildBySigTraits({ deadline: 0xffffffffff, nonceType: NonceType.Invalid, nonce: 0 }),
+                data: '0x',
+            };
+            await expect(token.bySig(alice, sig, '0x')).to.be.revertedWithCustomError(token, 'WrongNonceType');
+        });
+
         it('should revert after traits deadline', async function () {
             const { addrs: { alice }, token } = await loadFixture(deployAddressArrayMock);
             const sig = {