Skip to content

Commit

Permalink
✅ update tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@lukacan
lukacan committed Feb 15, 2024
1 parent 84cf04e commit c6ad0ca
Show file tree
Hide file tree
Showing 2 changed files with 178 additions and 192 deletions.
128 changes: 82 additions & 46 deletions crates/client/tests/test_data/expected_source_codes/expected_accounts_snapshots.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -26,59 +26,75 @@ impl<'info> InitVestingSnapshot<'info> {
accounts: &'info mut [Option<trdelnik_client::solana_sdk::account::Account>],
) -> core::result::Result<Self, FuzzingError> {
let accounts = get_account_infos_option(accounts, metas)
.map_err(|_| FuzzingError::CannotGetAccounts)?;
.map_err(|_| FuzzingError::NotAbleToObtainAccountInfos)?;
let mut accounts_iter = accounts.into_iter();
let sender: Signer<'_> = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts("sender".to_string()))?
.map(|acc| anchor_lang::accounts::signer::Signer::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound("sender".to_string()))?
.map_err(|_| FuzzingError::CannotDeserializeAccount("sender".to_string()))?;
let sender_token_account: anchor_lang::accounts::account::Account<TokenAccount> =
accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts(
"sender_token_account".to_string(),
))?
.map(|acc| anchor_lang::accounts::account::Account::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound(
"sender_token_account".to_string(),
))?
.map_err(|_| {
FuzzingError::CannotDeserializeAccount("sender_token_account".to_string())
})?;
let escrow: Option<anchor_lang::accounts::account::Account<Escrow>> = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts("escrow".to_string()))?
.map(|acc| {
if acc.key() != PROGRAM_ID {
anchor_lang::accounts::account::Account::try_from(&acc)
.map_err(|e| e.to_string())
.map_err(|_| FuzzingError::CannotDeserializeAccount("escrow".to_string()))
} else {
Err("Optional account not provided".to_string())
Err(FuzzingError::OptionalAccountNotProvided(
"escrow".to_string(),
))
}
})
.transpose()
.unwrap_or(None);
let escrow_token_account: anchor_lang::accounts::account::Account<TokenAccount> =
accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts(
"escrow_token_account".to_string(),
))?
.map(|acc| anchor_lang::accounts::account::Account::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound(
"escrow_token_account".to_string(),
))?
.map_err(|_| {
FuzzingError::CannotDeserializeAccount("escrow_token_account".to_string())
})?;
let mint: anchor_lang::accounts::account::Account<Mint> = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts("mint".to_string()))?
.map(|acc| anchor_lang::accounts::account::Account::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound("mint".to_string()))?
.map_err(|_| FuzzingError::CannotDeserializeAccount("mint".to_string()))?;
let token_program: anchor_lang::accounts::program::Program<Token> = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts("token_program".to_string()))?
.map(|acc| anchor_lang::accounts::program::Program::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound("token_program".to_string()))?
.map_err(|_| FuzzingError::CannotDeserializeAccount("token_program".to_string()))?;
let system_program: anchor_lang::accounts::program::Program<System> = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts(
"system_program".to_string(),
))?
.map(|acc| anchor_lang::accounts::program::Program::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound("system_program".to_string()))?
.map_err(|_| FuzzingError::CannotDeserializeAccount("system_program".to_string()))?;
Ok(Self {
sender,
sender_token_account,
Expand All @@ -96,63 +112,83 @@ impl<'info> WithdrawUnlockedSnapshot<'info> {
accounts: &'info mut [Option<trdelnik_client::solana_sdk::account::Account>],
) -> core::result::Result<Self, FuzzingError> {
let accounts = get_account_infos_option(accounts, metas)
.map_err(|_| FuzzingError::CannotGetAccounts)?;
.map_err(|_| FuzzingError::NotAbleToObtainAccountInfos)?;
let mut accounts_iter = accounts.into_iter();
let recipient: Signer<'_> = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts("recipient".to_string()))?
.map(|acc| anchor_lang::accounts::signer::Signer::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound("recipient".to_string()))?
.map_err(|_| FuzzingError::CannotDeserializeAccount("recipient".to_string()))?;
let recipient_token_account: anchor_lang::accounts::account::Account<TokenAccount> =
accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts(
"recipient_token_account".to_string(),
))?
.map(|acc| anchor_lang::accounts::account::Account::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound(
"recipient_token_account".to_string(),
))?
.map_err(|_| {
FuzzingError::CannotDeserializeAccount("recipient_token_account".to_string())
})?;
let escrow: Option<anchor_lang::accounts::account::Account<Escrow>> = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts("escrow".to_string()))?
.map(|acc| {
if acc.key() != PROGRAM_ID {
anchor_lang::accounts::account::Account::try_from(&acc)
.map_err(|e| e.to_string())
.map_err(|_| FuzzingError::CannotDeserializeAccount("escrow".to_string()))
} else {
Err("Optional account not provided".to_string())
Err(FuzzingError::OptionalAccountNotProvided(
"escrow".to_string(),
))
}
})
.transpose()
.unwrap_or(None);
let escrow_token_account: anchor_lang::accounts::account::Account<TokenAccount> =
accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts(
"escrow_token_account".to_string(),
))?
.map(|acc| anchor_lang::accounts::account::Account::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound(
"escrow_token_account".to_string(),
))?
.map_err(|_| {
FuzzingError::CannotDeserializeAccount("escrow_token_account".to_string())
})?;
let escrow_pda_authority = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::AccountNotFound)?;
.ok_or(FuzzingError::NotEnoughAccounts(
"escrow_pda_authority".to_string(),
))?
.ok_or(FuzzingError::AccountNotFound(
"escrow_pda_authority".to_string(),
))?;
let mint: anchor_lang::accounts::account::Account<Mint> = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts("mint".to_string()))?
.map(|acc| anchor_lang::accounts::account::Account::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound("mint".to_string()))?
.map_err(|_| FuzzingError::CannotDeserializeAccount("mint".to_string()))?;
let token_program: anchor_lang::accounts::program::Program<Token> = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts("token_program".to_string()))?
.map(|acc| anchor_lang::accounts::program::Program::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound("token_program".to_string()))?
.map_err(|_| FuzzingError::CannotDeserializeAccount("token_program".to_string()))?;
let system_program: anchor_lang::accounts::program::Program<System> = accounts_iter
.next()
.ok_or(FuzzingError::NotEnoughAccounts)?
.ok_or(FuzzingError::NotEnoughAccounts(
"system_program".to_string(),
))?
.map(|acc| anchor_lang::accounts::program::Program::try_from(&acc))
.ok_or(FuzzingError::AccountNotFound)?
.map_err(|_| FuzzingError::CannotDeserializeAccount)?;
.ok_or(FuzzingError::AccountNotFound("system_program".to_string()))?
.map_err(|_| FuzzingError::CannotDeserializeAccount("system_program".to_string()))?;
Ok(Self {
recipient,
recipient_token_account,
Expand Down
Loading

0 comments on commit c6ad0ca

Please sign in to comment.