Add commonly-used SPL programs as a convenience to the user

program-test/src/lib.rs

@@ -1,29 +1,30 @@
 //! The solana-program-test provides a BanksClient-based test framework BPF programs
 
-use async_trait::async_trait;
-use chrono_humanize::{Accuracy, HumanTime, Tense};
-use log::*;
-use solana_banks_client::start_client;
-use solana_banks_server::banks_server::start_local_server;
-use solana_program::{
+use {
+    async_trait::async_trait,
+    chrono_humanize::{Accuracy, HumanTime, Tense},
+    log::*,
+    solana_banks_client::start_client,
+    solana_banks_server::banks_server::start_local_server,
+    solana_program::{
         account_info::AccountInfo, entrypoint::ProgramResult, fee_calculator::FeeCalculator,
         hash::Hash, instruction::Instruction, instruction::InstructionError, message::Message,
         native_token::sol_to_lamports, program_error::ProgramError, program_stubs, pubkey::Pubkey,
         rent::Rent,
-};
-use solana_runtime::{
+    },
+    solana_runtime::{
         bank::{Bank, Builtin},
         bank_forks::BankForks,
         genesis_utils::create_genesis_config_with_leader,
-};
-use solana_sdk::{
+    },
+    solana_sdk::{
         account::Account,
         keyed_account::KeyedAccount,
         process_instruction::BpfComputeBudget,
         process_instruction::{InvokeContext, MockInvokeContext, ProcessInstructionWithContext},
         signature::{Keypair, Signer},
-};
-use std::{
+    },
+    std::{
         cell::RefCell,
         collections::HashMap,
         convert::TryFrom,
@@ -33,10 +34,11 @@ use std::{
         rc::Rc,
         sync::{Arc, RwLock},
         time::{Duration, Instant},
+    },
 };
 
 // Export types so test clients can limit their solana crate dependencies
-pub use solana_banks_client::{BanksClient, BanksClientExt};
+pub use solana_banks_client::BanksClient;
 
 #[macro_use]
 extern crate solana_bpf_loader_program;
@@ -356,6 +358,24 @@ fn read_file<P: AsRef<Path>>(path: P) -> Vec<u8> {
     file_data
 }
 
+mod spl_token {
+    solana_sdk::declare_id!("TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA");
+}
+mod spl_memo {
+    solana_sdk::declare_id!("Memo1UhkJRfHyvLMcVucJwxXeuD728EqVDDwQDxFMNo");
+}
+mod spl_associated_token_account {
+    solana_sdk::declare_id!("ATokenGPvbdGVxr1b2hvZbsiqW5xWH25efTNsLJA8knL");
+}
+static SPL_PROGRAMS: &[(Pubkey, &[u8])] = &[
+    (spl_token::ID, include_bytes!("programs/spl_token-2.0.6.so")),
+    (spl_memo::ID, include_bytes!("programs/spl_memo-1.0.0.so")),
+    (
+        spl_associated_token_account::ID,
+        include_bytes!("programs/spl_associated-token-account-1.0.1.so"),
+    ),
+];
+
 pub struct ProgramTest {
     accounts: Vec<(Pubkey, Account)>,
     builtins: Vec<Builtin>,
@@ -593,9 +613,23 @@ impl ProgramTest {
             bank.add_builtin(&loader.0, loader.1, loader.2);
         }
 
+        // Add commonly-used SPL programs as a convenience to the user
+        for (program_id, elf) in SPL_PROGRAMS.iter() {
+            bank.store_account(
+                program_id,
+                &Account {
+                    lamports: Rent::default().minimum_balance(elf.len()).min(1),
+                    data: elf.to_vec(),
+                    owner: solana_program::bpf_loader::id(),
+                    executable: true,
+                    rent_epoch: 0,
+                },
+            )
+        }
+
         // User-supplied additional builtins
         for builtin in self.builtins {
-            bank.add_builtin(
+            bank.replace_builtin(
                 &builtin.name,
                 builtin.id,
                 builtin.process_instruction_with_context,
@@ -604,7 +638,7 @@ impl ProgramTest {
 
         for (address, account) in self.accounts {
             if bank.get_account(&address).is_some() {
-                panic!("An account at {} already exists", address);
+                info!("Overriding account at {}", address);
             }
             bank.store_account(&address, &account);
         }