Fix secrets flush in TestContext

Author: DogLooksGood

test/src/client_helpers.rs

@@ -203,6 +203,10 @@ impl TestClient {
     pub fn answer(&mut self, decision_id: DecisionId, answer: String) -> Result<Event> {
         self.client.answer_event(decision_id, answer)
     }
+
+    pub fn flush_secret_state(&mut self) {
+        self.client.flush_secret_states();
+    }
 }
 
 #[cfg(test)]

test/src/handler_helpers.rs

@@ -113,11 +113,13 @@ impl<H: GameHandler> TestHandler<H> {
                     context.cancel_dispatch();
                 }
             }
-            // Handle events (responses) from Clients/transactor(s) after they see updated ctx
-            for i in 0..clients.len() {
-                let c = clients.get_mut(i).unwrap();
+
+            for c in clients.iter_mut() {
                 let cli_evts = c.handle_updated_context(context)?;
                 evts.extend_from_slice(&cli_evts);
+                if event_effects.checkpoint.is_some() {
+                    c.flush_secret_state();
+                }
             }
 
             if let Some(dispatch) = context.get_dispatch() {