feat(TWABDelegator): add decimals function

pooltogether · Mar 14, 2022 · 0df4f36 · 0df4f36
1 parent 2b65097
commit 0df4f36
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/contracts/TWABDelegator.sol b/contracts/TWABDelegator.sol
@@ -489,6 +489,15 @@ contract TWABDelegator is ERC20, LowLevelDelegator, PermitAndMulticall {
     return _computeAddress(_delegator, _slot);
   }
 
+  /**
+   * @notice Returns the ERC20 token decimals.
+   * @dev This value is equal to the decimals of the ticket being delegated.
+   * @return ERC20 token decimals
+   */
+  function decimals() public view virtual override returns (uint8) {
+    return ERC20(address(ticket)).decimals();
+  }
+
   /* ============ Internal Functions ============ */
 
   /**

diff --git a/test/TWABDelegator.test.ts b/test/TWABDelegator.test.ts
@@ -63,7 +63,7 @@ describe('Test Set Name', () => {
     ticket = await ticketContractFactory.deploy(
       'PoolTogether aUSDC Ticket',
       'PTaUSDC',
-      18,
+      6,
       controller.address,
     );
 
@@ -858,4 +858,10 @@ describe('Test Set Name', () => {
       expect(wasCreated).to.equal(true);
     });
   });
+
+  describe('decimals()', () => {
+    it('should return ticket decimals', async () => {
+      expect(await twabDelegator.decimals()).to.eq(6);
+    });
+  });
 });