Change mock clock check (#269)

* Fix * Comment
pyth-network · Nov 14, 2023 · 121b94e · 121b94e · vercel · Nov 14, 2023
1 parent dcacb3c
commit 121b94e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/staking/app/StakeConnection.ts b/staking/app/StakeConnection.ts
@@ -278,10 +278,8 @@ export class StakeConnection {
 
   // Gets the current unix time, as would be perceived by the on-chain program
   public async getTime(): Promise<BN> {
-    // The Idl contains mockClockTime even when we build it with mock-clock feature disabled.
-    // Therefore if the field doesn't exist it gets parsed as 0.
-    // Thus, if mockClockTime is 0 we need to use real time.
-    if ("mockClockTime" in this.config && this.config.mockClockTime.gtn(0)) {
+    // This is a hack, we are using this deprecated flag to flag whether we are using the mock clock or not
+    if (this.config.freeze) {
       // On chain program using mock clock, so get that time
       const updatedConfig = await this.program.account.globalConfig.fetch(
         this.configAddress

diff --git a/staking/tests/clock_api_test.ts b/staking/tests/clock_api_test.ts
@@ -53,7 +53,7 @@ describe("clock_api", async () => {
 
   it("gets real time", async () => {
     // Delete the property, which will make the API think it's not using mock clock anymore
-    delete stakeConnection.config.mockClockTime;
+    stakeConnection.config.freeze = false;
     let sysTime = Date.now() / 1000;
     let solanaTime = (await stakeConnection.getTime()).toNumber();
     assert.ok(Math.abs(sysTime - solanaTime) < CLOCK_TOLERANCE_SECONDS);

diff --git a/staking/tests/utils/before.ts b/staking/tests/utils/before.ts
@@ -352,7 +352,7 @@ export function makeDefaultConfig(
     pythTokenMint: pythMint,
     unlockingDuration: 1,
     epochDuration: new BN(3600),
-    freeze: false,
+    freeze: true,
     mockClockTime: new BN(10),
     bump: 0,
     pythTokenListTime: null,