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

}