From: Ian Jackson <ijackson@chiark.greenend.org.uk>
Date: Sat, 13 Mar 2021 15:08:38 +0000 (+0000)
Subject: atpitests: Provide fakerng methods on MgmtChannel
X-Git-Tag: otter-0.4.0~122
X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=commitdiff_plain;h=529c0110ee0d4244d2ea3bd8515c0d4774790b80;p=otter.git

atpitests: Provide fakerng methods on MgmtChannel

Signed-off-by: Ian Jackson <ijackson@chiark.greenend.org.uk>
---

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()?;
   }
 }