From 8acc47ee1b687a0a572effa0af2fd9c96705a909 Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Sat, 31 Oct 2020 00:36:00 -0700
Subject: [PATCH] MockInvokeContext::get_programs() implementation

---
 sdk/src/process_instruction.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/sdk/src/process_instruction.rs b/sdk/src/process_instruction.rs
index 6b15f021f1..0785d10168 100644
--- a/sdk/src/process_instruction.rs
+++ b/sdk/src/process_instruction.rs
@@ -202,12 +202,12 @@ impl Logger for MockLogger {
     }
 }
 
-#[derive(Debug)]
 pub struct MockInvokeContext {
     pub key: Pubkey,
     pub logger: MockLogger,
     pub bpf_compute_budget: BpfComputeBudget,
     pub compute_meter: MockComputeMeter,
+    pub programs: Vec<(Pubkey, ProcessInstructionWithContext)>,
 }
 impl Default for MockInvokeContext {
     fn default() -> Self {
@@ -218,6 +218,7 @@ impl Default for MockInvokeContext {
             compute_meter: MockComputeMeter {
                 remaining: std::i64::MAX as u64,
             },
+            programs: vec![],
         }
     }
 }
@@ -238,7 +239,7 @@ impl InvokeContext for MockInvokeContext {
         Ok(&self.key)
     }
     fn get_programs(&self) -> &[(Pubkey, ProcessInstructionWithContext)] {
-        &[]
+        &self.programs
     }
     fn get_logger(&self) -> Rc<RefCell<dyn Logger>> {
         Rc::new(RefCell::new(self.logger.clone()))