From 2fdcd35cfbe76a6f13a9d41d75d99931b42a1aeb Mon Sep 17 00:00:00 2001
From: Prateek <prateek@perennial.finance>
Date: Thu, 19 Sep 2024 16:29:11 +0530
Subject: [PATCH] update tests

---
 .../perennial/test/integration/core/happyPath.test.ts  | 10 +++++++---
 .../test/unit/marketfactory/MarketFactory.test.ts      | 10 ++++++++--
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/packages/perennial/test/integration/core/happyPath.test.ts b/packages/perennial/test/integration/core/happyPath.test.ts
index d36a42ad1..133ba6ff1 100644
--- 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
index 61c22ba17..31666d936 100644
--- 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 () => {