Get access to runtime errors in Budget unit-tests

programs/budget_api/src/budget_state.rs

@@ -2,19 +2,11 @@
 use crate::budget_expr::BudgetExpr;
 use bincode::{self, deserialize, serialize_into};
 use serde_derive::{Deserialize, Serialize};
+use solana_sdk::native_program::ProgramError;
 
 #[derive(Serialize, Deserialize, Debug, Clone, PartialEq)]
 pub enum BudgetError {
-    InsufficientFunds,
-    ContractAlreadyExists,
-    ContractNotPending,
-    SourceIsPendingContract,
-    UninitializedContract,
     DestinationMissing,
-    FailedWitness,
-    AccountDataTooSmall,
-    AccountDataDeserializeFailure,
-    UnsignedKey,
 }
 
 #[derive(Serialize, Deserialize, Debug, Clone, Default, PartialEq)]
@@ -35,14 +27,12 @@ impl BudgetState {
         self.pending_budget.is_some()
     }
 
-    pub fn serialize(&self, output: &mut [u8]) -> Result<(), BudgetError> {
-        serialize_into(output, self).map_err(|err| match *err {
-            _ => BudgetError::AccountDataTooSmall,
-        })
+    pub fn serialize(&self, output: &mut [u8]) -> Result<(), ProgramError> {
+        serialize_into(output, self).map_err(|_| ProgramError::AccountDataTooSmall)
     }
 
-    pub fn deserialize(input: &[u8]) -> bincode::Result<Self> {
-        deserialize(input)
+    pub fn deserialize(input: &[u8]) -> Result<Self, ProgramError> {
+        deserialize(input).map_err(|_| ProgramError::InvalidAccountData)
     }
 }
 
@@ -67,7 +57,7 @@ mod test {
         let b = BudgetState::default();
         assert_eq!(
             b.serialize(&mut a.data),
-            Err(BudgetError::AccountDataTooSmall)
+            Err(ProgramError::AccountDataTooSmall)
         );
     }
 }