admin tests

tests/test_update_difficulty.rs

@@ -0,0 +1,49 @@
+use ore::{state::Treasury, utils::AccountDeserialize, TREASURY_ADDRESS};
+use solana_program::{hash::Hash, keccak::Hash as KeccakHash};
+use solana_program_test::{processor, BanksClient, ProgramTest};
+use solana_sdk::{
+    signature::{Keypair, Signer},
+    transaction::Transaction,
+};
+
+#[tokio::test]
+async fn test_update_difficulty() {
+    // Setup
+    let (mut banks, payer, blockhash) = setup_program_test_env().await;
+
+    // Submit tx
+    let ix = ore::instruction::initialize(payer.pubkey());
+    let tx = Transaction::new_signed_with_payer(&[ix], Some(&payer.pubkey()), &[&payer], blockhash);
+    let res = banks.process_transaction(tx).await;
+    assert!(res.is_ok());
+
+    // Get treasury account
+    let treasury_account = banks.get_account(TREASURY_ADDRESS).await.unwrap().unwrap();
+    let treasury = Treasury::try_from_bytes(&treasury_account.data).unwrap();
+
+    // Submit update admin ix
+    let new_difficulty = KeccakHash::new_unique();
+    let ix = ore::instruction::update_difficulty(payer.pubkey(), new_difficulty.into());
+    let tx = Transaction::new_signed_with_payer(&[ix], Some(&payer.pubkey()), &[&payer], blockhash);
+    let res = banks.process_transaction(tx).await;
+    assert!(res.is_ok());
+
+    // Assert treasury state
+    let treasury_account = banks.get_account(TREASURY_ADDRESS).await.unwrap().unwrap();
+    let treasury_ = Treasury::try_from_bytes(&treasury_account.data).unwrap();
+    assert_eq!(treasury_.bump, treasury.bump);
+    assert_eq!(treasury_.admin, treasury.admin);
+    assert_eq!(treasury_.difficulty, new_difficulty.into());
+    assert_eq!(treasury_.epoch_start_at, treasury.epoch_start_at);
+    assert_eq!(treasury_.reward_rate, treasury.reward_rate);
+    assert_eq!(
+        treasury_.total_claimed_rewards,
+        treasury.total_claimed_rewards
+    );
+}
+
+async fn setup_program_test_env() -> (BanksClient, Keypair, Hash) {
+    let mut program_test = ProgramTest::new("ore", ore::ID, processor!(ore::process_instruction));
+    program_test.prefer_bpf(true);
+    program_test.start().await
+}