Merge branch 'giannis/check-on-init-rwd-dist' of github.com:cardinal-…

…labs/cardinal-rewards-center into giannis/check-on-init-rwd-dist
solana-nft-programs · Jan 3, 2023 · 2bae855 · 2bae855
2 parents b5c45b2 + fc5083e
commit 2bae855
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/...inal-rewards-center/src/reward_distribution/reward_distributor/init_reward_distributor.rs b/...inal-rewards-center/src/reward_distribution/reward_distributor/init_reward_distributor.rs
@@ -32,7 +32,7 @@ pub struct InitRewardDistributorCtx<'info> {
  bump,
  )]
  reward_distributor: Box<Account<'info, RewardDistributor>>,
- #[account(constraint = authority.key() != stake_pool.authority.key() @ ErrorCode::InvalidAuthority)]
+ #[account(constraint = authority.key() == stake_pool.authority.key() @ ErrorCode::InvalidAuthority)]
  stake_pool: Box<Account<'info, StakePool>>,
  #[account(mut)]
  reward_mint: Box<Account<'info, Mint>>,

diff --git a/tests/reward-distribution/claim-rewards-sol.test.ts b/tests/reward-distribution/claim-rewards-sol.test.ts
@@ -6,12 +6,7 @@ import {
  getAssociatedTokenAddressSync,
 } from "@solana/spl-token";
 import type { PublicKey } from "@solana/web3.js";
-import {
- Keypair,
- LAMPORTS_PER_SOL,
- SystemProgram,
- Transaction,
-} from "@solana/web3.js";
+import { Keypair, SystemProgram, Transaction } from "@solana/web3.js";
 import { BN } from "bn.js";
 
 import {
@@ -43,8 +38,6 @@ const REWARD_AMOUNT = 2;
 let mintId: PublicKey;
 let rewardMintId: PublicKey;
 
-const PAYMENT_AMOUNT = 0.005 * LAMPORTS_PER_SOL;
-
 beforeAll(async () => {
  provider = await getProvider();
  const mintKeypair = Keypair.generate();
@@ -346,5 +339,12 @@ test("Claim rewards", async () => {
  const balanceAfter = await provider.connection.getBalance(
  provider.wallet.publicKey
  );
- expect(balanceBefore - PAYMENT_AMOUNT).toBeGreaterThan(balanceAfter);
+ const defaultPaymentInfo = await fetchIdlAccount(
+ provider.connection,
+ DEFAULT_PAYMENT_INFO,
+ "paymentInfo"
+ );
+ expect(balanceBefore - balanceAfter).toBeGreaterThan(
+ defaultPaymentInfo.parsed.paymentAmount.toNumber()
+ );
 });