From 529c0110ee0d4244d2ea3bd8515c0d4774790b80 Mon Sep 17 00:00:00 2001 From: Ian Jackson Date: Sat, 13 Mar 2021 15:08:38 +0000 Subject: [PATCH] atpitests: Provide fakerng methods on MgmtChannel Signed-off-by: Ian Jackson --- apitest.rs | 10 ++++++++++ apitest/at-otter.rs | 5 ++--- 2 files changed, 12 insertions(+), 3 deletions(-) diff --git a/apitest.rs b/apitest.rs index cb474c80..0138efc2 100644 --- a/apitest.rs +++ b/apitest.rs @@ -766,6 +766,16 @@ impl MgmtChannel { trace!("gen={} ...", gen); gen } + + fn fakerng_load(&mut self, values: &[&dyn ToString]) -> Result<(),AE> { + let values = values.iter().map(|v| v.to_string()).collect(); + self.cmd(&MC::LoadFakeRng(values))?; + Ok(()) + } + fn fakerng_unfake(&mut self) -> Result<(),AE> { + self.cmd(&MC::LoadFakeRng(vec![]))?; + Ok(()) + } } // ==================== core entrypoint, for wdriver too ==================== diff --git a/apitest/at-otter.rs b/apitest/at-otter.rs index f1003d82..f7ae25d3 100644 --- a/apitest/at-otter.rs +++ b/apitest/at-otter.rs @@ -384,8 +384,7 @@ impl Ctx { fn hidden_hand(&mut self) { prepare_game(&self.su().ds, TABLE)?; let mut alice = self.connect_player(&self.alice)?; - self.su_mut() - .mgmt_conn.cmd(&MC::LoadFakeRng(vec![ "1".to_owned() ]))?; + self.su_mut().mgmt_conn.fakerng_load(&[&"1"])?; let pieces = alice.pieces()?; @@ -419,7 +418,7 @@ impl Ctx { // to repro a bug, have Bob move the RHS pawn out again - self.su_mut().mgmt_conn.cmd(&MC::LoadFakeRng(vec![]))?; + self.su_mut().mgmt_conn.fakerng_unfake()?; } } -- 2.30.2