Subversion Repositories openrisc

// 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
}