test: add update_rewards toggle tokens flow test

[arad-starkware]

This change is 

---

Note

Adds a comprehensive flow test that toggles enabled BTC tokens across attestation and consensus epochs and validates rewards behavior with update_rewards.

• Tests:
  • Add update_rewards_token_enable_disable_flow_test in src/flow_test/test.cairo:
    • Adds two BTC tokens (8d/18d), toggles enable/disable across epochs, performs staking/delegation, attests, starts consensus, calls update_rewards (including twice in same epoch), and asserts staker/delegator rewards per token in attestation (v2) and consensus (v3) modes.
    • Uses NormalizedAmountTrait and v2/v3 reward calculators for STRK/BTC.
• Supporting changes:
  • Import updates: NormalizedAmountTrait; switch to/use calculate_staker_btc_pool_rewards_v3, calculate_staker_strk_rewards_v2, etc.
  • Tidy src/flow_test/flow_ideas.md by simplifying the k=1 -> k=2 token item.

^{Written by Cursor Bugbot for commit 2dc8dac. This will update automatically on new commits. Configure here.}

[arad-starkware]

• test: add accrue_rewards_across_versions flow #82 : 2 dependent PRs (#83 , #84 )
• test: add attest after upgrade test #71
• test: add token state after upgrade flow #59
• test: add update_rewards toggle tokens flow test #57 👈 (View in Graphite)
• test: add balances delay flow #65
• chore: update flow ideas #61
• main

This stack of pull requests is managed by Graphite. Learn more about stacking.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 95.81%. Comparing base (6294afd) to head (2dc8dac).
⚠️ Report is 1 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #57      +/-   ##
==========================================
+ Coverage   95.77%   95.81%   +0.03%     
==========================================
  Files          42       42              
  Lines       10102    10194      +92     
==========================================
+ Hits         9675     9767      +92     
  Misses        427      427              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[noa-starkware]

Reviewable status: 0 of 3 files reviewed, 2 unresolved discussions (waiting on @arad-starkware)

---

src/flow_test/test.cairo line 3039 at r3 (raw file):

/// Advance K epochs
/// Enable token A, disable token B
/// update_rewards

I think we need this test for both v2 rewards and consensus rewards, please update the test

---

src/flow_test/flows.cairo line 7070 at r3 (raw file):

/// Add tokens A and B
/// Enable token B
/// Advance epoch

rebase issues?

[arad-starkware]

Reviewable status: 0 of 3 files reviewed, 2 unresolved discussions (waiting on @noa-starkware)

---

src/flow_test/test.cairo line 3039 at r3 (raw file):

Previously, noa-starkware wrote…

I think we need this test for both v2 rewards and consensus rewards, please update the test

I'm not sure about the flow you're describing, do you mean specifically v2 rewards and consensus in the same test? we have MultipleTokensDelegationFlow for v2 rewards otherwise

---

src/flow_test/flows.cairo line 7070 at r3 (raw file):

Previously, noa-starkware wrote…

rebase issues?

yes

[arad-starkware]

Reviewable status: 0 of 3 files reviewed, 7 unresolved discussions (waiting on @noa-starkware)

---

src/flow_test/test.cairo line 3094 at r6 (raw file):

Previously, noa-starkware wrote…

test also staker rewards please

Done.

---

src/flow_test/test.cairo line 3111 at r6 (raw file):

Previously, noa-starkware wrote…

test also staker rewards please

Done.

---

src/flow_test/test.cairo line 3120 at r6 (raw file):

Previously, noa-starkware wrote…

test also staker rewards please

Done.

---

src/flow_test/test.cairo line 3128 at r6 (raw file):

Previously, noa-starkware wrote…

// Enable token B, disable token A

Done

---

src/flow_test/test.cairo line 3148 at r6 (raw file):

Previously, noa-starkware wrote…

test also staker rewards please

Done.

---

src/flow_test/test.cairo line 3156 at r6 (raw file):

Previously, noa-starkware wrote…

test also staker rewards please

Done.

[noa-starkware]

@noa-starkware reviewed 1 of 2 files at r11.
Reviewable status: 1 of 3 files reviewed, 3 unresolved discussions (waiting on @arad-starkware)

---

src/flow_test/test.cairo line 3244 at r11 (raw file):

    system.staking.disable_token(token_address: token_a.contract_address());

    // Calculate rewards for consensus rewards

• only for one token enabled

---

src/flow_test/test.cairo line 3269 at r11 (raw file):

    // update_rewards - test rewards only for token A
    system.update_rewards(:staker, disable_rewards: false);

update_rewards twice before advance epoch?

[arad-starkware]

Reviewable status: 1 of 3 files reviewed, all discussions resolved

---

src/flow_test/test.cairo line 3244 at r11 (raw file):

Previously, noa-starkware wrote…

• only for one token enabled

Done

---

src/flow_test/test.cairo line 3269 at r11 (raw file):

Previously, noa-starkware wrote…

update_rewards twice before advance epoch?

Done

[graphite-app]

Merge activity

• Dec 2, 9:44 AM UTC: This pull request can not be added to the Graphite merge queue. Please try rebasing and resubmitting to merge when ready.
• Dec 2, 9:44 AM UTC: Graphite disabled "merge when ready" on this PR due to: a merge conflict with the target branch; resolve the conflict and try again..
• Dec 4, 1:01 PM UTC: @arad-starkware merged this pull request with Graphite.