// Copyright 2009 The Go Authors. All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file. package testing import ( "bytes" "fmt" "io" "os" "strings" "time" ) type InternalExample struct { Name string F func() Output string } func RunExamples(examples []InternalExample) (ok bool) { ok = true var eg InternalExample stdout, stderr := os.Stdout, os.Stderr defer func() { os.Stdout, os.Stderr = stdout, stderr if e := recover(); e != nil { fmt.Printf("--- FAIL: %s\npanic: %v\n", eg.Name, e) os.Exit(1) } }() for _, eg = range examples { if *chatty { fmt.Printf("=== RUN: %s\n", eg.Name) } // capture stdout and stderr r, w, err := os.Pipe() if err != nil { fmt.Fprintln(os.Stderr, err) os.Exit(1) } os.Stdout, os.Stderr = w, w outC := make(chan string) go func() { buf := new(bytes.Buffer) _, err := io.Copy(buf, r) if err != nil { fmt.Fprintf(stderr, "testing: copying pipe: %v\n", err) os.Exit(1) } outC <- buf.String() }() // run example t0 := time.Now() eg.F() dt := time.Now().Sub(t0) // close pipe, restore stdout/stderr, get output w.Close() os.Stdout, os.Stderr = stdout, stderr out := <-outC // report any errors tstr := fmt.Sprintf("(%.2f seconds)", dt.Seconds()) if g, e := strings.TrimSpace(out), strings.TrimSpace(eg.Output); g != e { fmt.Printf("--- FAIL: %s %s\ngot:\n%s\nwant:\n%s\n", eg.Name, tstr, g, e) ok = false } else if *chatty { fmt.Printf("--- PASS: %s %s\n", eg.Name, tstr) } } return }