From a1fc6c9fde2b792b02cf568c1d271a6dbe54f15a Mon Sep 17 00:00:00 2001
From: Ian Jackson <ijackson@chiark.greenend.org.uk>
Date: Sun, 10 Jan 2021 20:25:22 +0000
Subject: [PATCH] wdt: Provide more otter() methods

Signed-off-by: Ian Jackson <ijackson@chiark.greenend.org.uk>
---
 wdriver.rs               | 11 +++++++++++
 wdriver/wdt-altergame.rs | 14 ++++----------
 2 files changed, 15 insertions(+), 10 deletions(-)

diff --git a/wdriver.rs b/wdriver.rs
index fc3619a1..e4e1b2ac 100644
--- a/wdriver.rs
+++ b/wdriver.rs
@@ -914,6 +914,17 @@ impl Setup {
     self.windows_squirreled.push(name.to_owned());
     window
   }
+
+  #[throws(AE)]
+  pub fn otter(&mut self, w: &Window, verb: &[&str], args: &[&str]) {
+    let args : Vec<String> =
+      ["--account", "server:"].iter().cloned().map(Into::into)
+      .chain(verb.iter().cloned().map(Into::into))
+      .chain(iter::once(w.table()))
+      .chain(args.iter().cloned().map(Into::into))
+      .collect();
+    self.ds.otter(&args)?;
+  }
 }
 
 impl Setup {
diff --git a/wdriver/wdt-altergame.rs b/wdriver/wdt-altergame.rs
index e19c0bd2..ae8294e7 100644
--- a/wdriver/wdt-altergame.rs
+++ b/wdriver/wdt-altergame.rs
@@ -39,20 +39,14 @@ impl Ctx {
   }
 
   #[throws(AE)]
-  fn otter_table(&mut self, verb: &[&str], args: &[&str]) {
-    let args : Vec<String> =
-      ["--account", "server:"].iter().cloned().map(Into::into)
-      .chain(verb.iter().cloned().map(Into::into))
-      .chain(iter::once(self.alice.table()))
-      .chain(args.iter().cloned().map(Into::into))
-      .collect();
-    self.su.ds.otter(&args)?;
+  fn otter(&mut self, verb: &[&str], args: &[&str]) {
+    self.su.otter(&self.alice, verb, args)?
   }
 
   #[throws(AE)]
   fn test_link(&mut self, kind: LinkKind, desc: &'static str, url: &str) {
     (||{
-      self.otter_table(&["set-link"], &[&kind.to_string(), url])?;
+      self.otter(&["set-link"], &[&kind.to_string(), url])?;
       self.check_link(desc, Some(url))?;
       Ok::<_,AE>(())
     })()
@@ -62,7 +56,7 @@ impl Ctx {
   #[throws(AE)]
   fn test_remove_link(&mut self, kind: LinkKind, desc: &'static str) {
     (||{
-      self.otter_table(&["set-link"], &[&kind.to_string(), ""])?;
+      self.otter(&["set-link"], &[&kind.to_string(), ""])?;
       self.check_link(desc, None)?;
       Ok::<_,AE>(())
     })()
-- 
2.30.2