Skip to content

Update first.nr #15

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
SoulcoIIector wants to merge 1 commit into from
Closed

Update first.nr #15

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
73 changes: 36 additions & 37 deletions src/test/first.nr
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,59 +1,57 @@
use crate::test::utils;

use dep::aztec::test::{helpers::{cheatcodes, test_environment::TestEnvironment}};
use dep::aztec::protocol_types::storage::map::derive_storage_slot_in_map;
use dep::aztec::note::note_getter::{MAX_NOTES_PER_PAGE, view_notes};
use dep::aztec::note::note_viewer_options::NoteViewerOptions;
use dep::aztec::hash::compute_secret_hash;

use dep::aztec::{oracle::{execution::{get_block_number, get_contract_address}, storage::storage_read}};

use crate::EasyPrivateVoting;

fn read_storage_value<T: Copy>(contract_address: &str, slot: usize) -> T {
let block_number = get_block_number();
storage_read(contract_address, slot, block_number)
}

#[test]
unconstrained fn test_initializer() {
let (_, voting_contract_address, admin) = utils::setup();

let block_number = get_block_number();
let admin_slot = EasyPrivateVoting::storage_layout().admin.slot;
let admin_storage_value = storage_read(voting_contract_address, admin_slot, block_number);
assert(admin_storage_value == admin, "Vote ended should be false");
let admin_storage_value: Address = read_storage_value(voting_contract_address, admin_slot);

assert_eq!(admin_storage_value, admin, "Admin storage value should match");
}

#[test]
unconstrained fn test_check_vote_status() {
let (_, voting_contract_address, _) = utils::setup();

let vote_ended_expected: bool = false;

let block_number = get_block_number();
let expected_vote_status = false;
let status_slot = EasyPrivateVoting::storage_layout().vote_ended.slot;
let vote_ended_read: bool = storage_read(voting_contract_address, status_slot, block_number);
assert(vote_ended_expected == vote_ended_read, "Vote ended should be false");
let actual_vote_status: bool = read_storage_value(voting_contract_address, status_slot);

assert_eq!(actual_vote_status, expected_vote_status, "Vote ended should be false");
}

#[test]
unconstrained fn test_end_vote() {
let (env, voting_contract_address, admin) = utils::setup();

env.impersonate(admin);

let call_interface = EasyPrivateVoting::at(voting_contract_address).end_vote();
env.call_public(call_interface);

let vote_ended_expected = true;

let block_number = get_block_number();

let expected_vote_status = true;
let status_slot = EasyPrivateVoting::storage_layout().vote_ended.slot;
let vote_ended_read: bool = storage_read(voting_contract_address, status_slot, block_number);
assert(vote_ended_expected == vote_ended_read, "Vote ended should be true");
let actual_vote_status: bool = read_storage_value(voting_contract_address, status_slot);

assert_eq!(actual_vote_status, expected_vote_status, "Vote ended should be true");
}

#[test(should_fail)]
unconstrained fn test_fail_end_vote_by_non_admin() {
let (env, voting_contract_address, _) = utils::setup();
let alice = env.create_account();

env.impersonate(alice);

let call_interface = EasyPrivateVoting::at(voting_contract_address).end_vote();
env.call_public(call_interface);
}
Expand All @@ -63,62 +61,63 @@ unconstrained fn test_cast_vote() {
let (env, voting_contract_address, _) = utils::setup();
let alice = env.create_account();
env.impersonate(alice);

let candidate = 1;
env.advance_block_by(6);
let call_interface = EasyPrivateVoting::at(voting_contract_address).cast_vote(candidate);
env.call_private_void(call_interface);

// Read vote count from storage
let block_number = get_block_number();
let tally_slot = EasyPrivateVoting::storage_layout().tally.slot;
let candidate_tally_slot = derive_storage_slot_in_map(tally_slot, candidate);
let vote_count: u32 = storage_read(voting_contract_address, candidate_tally_slot, block_number);

assert(vote_count == 1, "vote tally should be incremented");
let vote_count: u32 = read_storage_value(voting_contract_address, candidate_tally_slot);
assert_eq!(vote_count, 1, "Vote tally should be incremented");
}

#[test]
unconstrained fn test_cast_vote_with_separate_accounts() {
let (env, voting_contract_address, _) = utils::setup();
let alice = env.create_account();
let bob = env.create_account();

let candidate = 101;

// Alice votes
env.impersonate(alice);
env.advance_block_by(1);
let call_interface = EasyPrivateVoting::at(voting_contract_address).cast_vote(candidate);
env.call_private_void(call_interface);


// Bob votes
env.impersonate(bob);
env.advance_block_by(1);
let call_interface = EasyPrivateVoting::at(voting_contract_address).cast_vote(candidate);
env.call_private_void(call_interface);

// Read vote count from storage
let block_number = get_block_number();
let tally_slot = EasyPrivateVoting::storage_layout().tally.slot;
let candidate_tally_slot = derive_storage_slot_in_map(tally_slot, candidate);
let vote_count: u32 = storage_read(voting_contract_address, candidate_tally_slot, block_number);

assert(vote_count == 2, "vote tally should be 2");
let vote_count: u32 = read_storage_value(voting_contract_address, candidate_tally_slot);
assert_eq!(vote_count, 2, "Vote tally should be 2");
}

#[test(should_fail)]
unconstrained fn test_fail_vote_twice() {
let (env, voting_contract_address, _) = utils::setup();
let alice = env.create_account();

let candidate = 101;


// Alice votes
env.impersonate(alice);
env.advance_block_by(1);
let call_interface = EasyPrivateVoting::at(voting_contract_address).cast_vote(candidate);
env.call_private_void(call_interface);

// Vote again as alice
// Vote again as Alice
env.advance_block_by(1);
let call_interface = EasyPrivateVoting::at(voting_contract_address).cast_vote(candidate);
env.call_private_void(call_interface);
}