update tests

equilibria-xyz · Sep 19, 2024 · 2fdcd35 · 2fdcd35
1 parent 3dac8e7
commit 2fdcd35
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/packages/perennial/test/integration/core/happyPath.test.ts b/packages/perennial/test/integration/core/happyPath.test.ts
@@ -1997,7 +1997,7 @@ describe('Happy Path', () => {
     const POSITION = parse6decimal('10')
     const POSITION_B = parse6decimal('1')
     const COLLATERAL = parse6decimal('1000')
-    const { user, userB, dsu, marketFactory } = instanceVars
+    const { owner, user, userB, dsu, marketFactory } = instanceVars
 
     const market = await createMarket(instanceVars)
 
@@ -2014,9 +2014,13 @@ describe('Happy Path', () => {
       .connect(user)
       ['update(address,uint256,uint256,uint256,int256,bool)'](user.address, POSITION, 0, 0, COLLATERAL, false)
 
-    // set user as extension for userB with signature
-    await marketFactory.connect(userB).updateExtension(user.address, true)
+    // try to update extension using incorrect owner
+    await expect(marketFactory.connect(userB).updateExtension(user.address, true))
+      .to.be.revertedWithCustomError(marketFactory, 'OwnableNotOwnerError')
+      .withArgs(userB.address)
 
+    // update extension with owner
+    await marketFactory.connect(owner).updateExtension(user.address, true)
     // user opens long position for userB
     await expect(
       market

diff --git a/packages/perennial/test/unit/marketfactory/MarketFactory.test.ts b/packages/perennial/test/unit/marketfactory/MarketFactory.test.ts
@@ -399,18 +399,24 @@ describe('MarketFactory', () => {
 
   describe('#updateExtension', async () => {
     it('updates the operator status', async () => {
-      await expect(factory.connect(user).updateExtension(owner.address, true))
+      await expect(factory.connect(owner).updateExtension(owner.address, true))
         .to.emit(factory, 'ExtensionUpdated')
         .withArgs(owner.address, true)
 
       expect(await factory.extensions(owner.address)).to.equal(true)
 
-      await expect(factory.connect(user).updateExtension(owner.address, false))
+      await expect(factory.connect(owner).updateExtension(owner.address, false))
         .to.emit(factory, 'ExtensionUpdated')
         .withArgs(owner.address, false)
 
       expect(await factory.extensions(owner.address)).to.equal(false)
     })
+
+    it('reverts if not owner', async () => {
+      await expect(factory.connect(user).updateExtension(user.address, true))
+        .to.be.revertedWithCustomError(factory, 'OwnableNotOwnerError')
+        .withArgs(user.address)
+    })
   })
 
   describe('#updateOperator', async () => {