diff --git a/programs/cardinal-rewards-center/src/reward_distribution/reward_distributor/init_reward_distributor.rs b/programs/cardinal-rewards-center/src/reward_distribution/reward_distributor/init_reward_distributor.rs index 079383ab..623242a6 100644 --- a/programs/cardinal-rewards-center/src/reward_distribution/reward_distributor/init_reward_distributor.rs +++ b/programs/cardinal-rewards-center/src/reward_distribution/reward_distributor/init_reward_distributor.rs @@ -32,7 +32,7 @@ pub struct InitRewardDistributorCtx<'info> { bump, )] reward_distributor: Box>, - #[account(constraint = authority.key() != stake_pool.authority.key() @ ErrorCode::InvalidAuthority)] + #[account(constraint = authority.key() == stake_pool.authority.key() @ ErrorCode::InvalidAuthority)] stake_pool: Box>, #[account(mut)] reward_mint: Box>, diff --git a/tests/reward-distribution/claim-rewards-sol.test.ts b/tests/reward-distribution/claim-rewards-sol.test.ts index 63dd15b3..6bbe92cf 100644 --- 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() + ); });