59 lines
1.6 KiB
Rust
59 lines
1.6 KiB
Rust
//! budget state
|
|
use crate::budget_expr::BudgetExpr;
|
|
use bincode::{self, deserialize, serialize_into};
|
|
use serde_derive::{Deserialize, Serialize};
|
|
use solana_sdk::instruction::InstructionError;
|
|
|
|
#[derive(Serialize, Deserialize, Debug, Clone, Default, PartialEq)]
|
|
pub struct BudgetState {
|
|
pub initialized: bool,
|
|
pub pending_budget: Option<BudgetExpr>,
|
|
}
|
|
|
|
impl BudgetState {
|
|
pub fn new(budget_expr: BudgetExpr) -> Self {
|
|
Self {
|
|
initialized: true,
|
|
pending_budget: Some(budget_expr),
|
|
}
|
|
}
|
|
|
|
pub fn is_pending(&self) -> bool {
|
|
self.pending_budget.is_some()
|
|
}
|
|
|
|
pub fn serialize(&self, output: &mut [u8]) -> Result<(), InstructionError> {
|
|
serialize_into(output, self).map_err(|_| InstructionError::AccountDataTooSmall)
|
|
}
|
|
|
|
pub fn deserialize(input: &[u8]) -> Result<Self, InstructionError> {
|
|
deserialize(input).map_err(|_| InstructionError::InvalidAccountData)
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use super::*;
|
|
use crate::id;
|
|
use solana_sdk::account::Account;
|
|
|
|
#[test]
|
|
fn test_serializer() {
|
|
let mut a = Account::new(0, 512, &id());
|
|
let b = BudgetState::default();
|
|
b.serialize(&mut a.data).unwrap();
|
|
let c = BudgetState::deserialize(&a.data).unwrap();
|
|
assert_eq!(b, c);
|
|
}
|
|
|
|
#[test]
|
|
fn test_serializer_data_too_small() {
|
|
let mut a = Account::new(0, 1, &id());
|
|
let b = BudgetState::default();
|
|
assert_eq!(
|
|
b.serialize(&mut a.data),
|
|
Err(InstructionError::AccountDataTooSmall)
|
|
);
|
|
}
|
|
}
|