Add more tests

Author: febo

Diff for: Cargo.lock

@@ -22,6 +22,7 @@ spl-token-interface = { version = "^0", path = "../interface" }
 
 [dev-dependencies]
 assert_matches = "1.5.0"
+num-traits = "0.2"
 solana-program-test = "2.1"
 solana-sdk = "2.1"
 spl-token = { version="^4", features=["no-entrypoint"] }

Diff for: p-token/Cargo.toml

@@ -6,3 +6,46 @@ pub mod account;
 pub mod mint;
 
 pub const TOKEN_PROGRAM_ID: Pubkey = Pubkey::new_from_array(spl_token_interface::program::ID);
+
+#[macro_export]
+macro_rules! assert_instruction_error {
+    ( $error:expr, $matcher:pat ) => {
+        match $error {
+            solana_program_test::BanksClientError::TransactionError(
+                solana_sdk::transaction::TransactionError::InstructionError(_, $matcher),
+            ) => {
+                assert!(true)
+            }
+            err => assert!(false, "Expected instruction error but got '{:#?}'", err),
+        };
+    };
+}
+
+#[macro_export]
+macro_rules! assert_custom_error {
+    ( $error:expr, $matcher:pat ) => {
+        match $error {
+            solana_program_test::BanksClientError::TransactionError(
+                solana_sdk::transaction::TransactionError::InstructionError(
+                    _,
+                    solana_sdk::instruction::InstructionError::Custom(x),
+                ),
+            ) => match num_traits::FromPrimitive::from_i32(x as i32) {
+                Some($matcher) => assert!(true),
+                Some(other) => {
+                    assert!(
+                        false,
+                        "Expected another custom instruction error than '{:#?}'",
+                        other
+                    )
+                }
+                None => assert!(false, "Expected custom instruction error"),
+            },
+            err => assert!(
+                false,
+                "Expected custom instruction error but got '{:#?}'",
+                err
+            ),
+        };
+    };
+}