Drover's back, baby!

go/src/infra/libs/infra_util/test_utils.go

+package infra_util
+
+import (
+        "io/ioutil"
+        "os"
+)
+
+func WithTempDir(dir, name string, f func(tmpdir string)) {
+        var pval interface{}
+        d, err := ioutil.TempDir(dir, name)
+        if err != nil {
+                panic(err)
+        }
+        func() {
+                defer func() { pval = recover() }()
+                f(d)
+        }()
+        err = os.RemoveAll(d)
+        if err != nil {
+                panic(err)
+        }
+        if pval != nil {
+                panic(pval)
+        }
+}