...
Run Format

Source file src/go/doc/testdata/example.go

Documentation: go/doc/testdata

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

View as plain text