Armani Ferrante 4 years ago
parent
commit
d0edec6003
2 changed files with 44 additions and 3 deletions
  1. 0 1
      tests/cfo/programs/cfo/src/lib.rs
  2. 44 2
      tests/cfo/tests/cfo.js

+ 0 - 1
tests/cfo/programs/cfo/src/lib.rs

@@ -637,7 +637,6 @@ pub struct DropStakeReward<'info> {
     mint: UncheckedAccount<'info>,
     srm: DropStakeRewardPool<'info>,
     msrm: DropStakeRewardPool<'info>,
-    msrm_registrar: Box<Account<'info, Registrar>>,
     token_program: Program<'info, Token>,
     registry_program: Program<'info, Registry>,
     lockup_program: Program<'info, Lockup>,

+ 44 - 2
tests/cfo/tests/cfo.js

@@ -4,7 +4,13 @@ const anchor = require("@project-serum/anchor");
 const serumCmn = require("@project-serum/common");
 const { Market } = require("@project-serum/serum");
 const utf8 = anchor.utils.bytes.utf8;
-const { PublicKey, SystemProgram, Keypair, SYSVAR_RENT_PUBKEY } = anchor.web3;
+const {
+  PublicKey,
+  SystemProgram,
+  Keypair,
+  SYSVAR_RENT_PUBKEY,
+  SYSVAR_CLOCK_PUBKEY,
+} = anchor.web3;
 const utils = require("./utils");
 const { setupStakePool } = require("./utils/stake");
 
@@ -49,7 +55,12 @@ describe("cfo", () => {
     // Serum DEX vault PDA for market B/USDC.
     marketBVaultSigner;
 
-  let registrar, msrmRegistrar;
+  let registrar, rewardEventQ, poolMint, vendor, vendorVault;
+  let msrmRegistrar,
+    msrmRewardEventQ,
+    msrmPoolMint,
+    msrmVendor,
+    msrmVendorVault;
 
   it("BOILERPLATE: Sets up a market with funded fees", async () => {
     ORDERBOOK_ENV = await utils.initMarket({
@@ -502,4 +513,35 @@ describe("cfo", () => {
         beforeAmount * (distribution.burn / 100.0)
     );
   });
+
+  it("Drops staking rewards", async () => {
+    //
+    await program.rpc.dropstakeReward({
+      accounts: {
+        officer,
+        stake,
+        mint: ORDERBOOK_ENV.mintA,
+        srm: {
+          registrar,
+          rewardEventQ,
+          poolMint,
+          vendor,
+          vendorVault,
+        },
+        msrm: {
+          registrar: msrmRegistrar,
+          rewardEventQ: msrmRewardEventQ,
+          poolMint: msrmPoolMint,
+          vendor: msrmVendor,
+          vendorVault: msrmVendorvault,
+        },
+        tokenProgram: TOKEN_PID,
+        registryProgram: REGISTRY_PID,
+        lockupProgram: LOCKUP_PID,
+        dexProgram: DEX_PID,
+        clock: SYSVAR_CLOCK_PUBKEY,
+        rent: SYSVAR_RENT_PUBKEY,
+      },
+    });
+  });
 });