1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
// Copyright 2019 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.
//go:build js || wasip1
package testing
import (
"fmt"
"io"
"os"
"strings"
"time"
)
// TODO(@musiol, @odeke-em): unify this code back into
// example.go when js/wasm gets an os.Pipe implementation.
func runExample(eg InternalExample) (ok bool) {
if chatty.on {
fmt.Printf("%s=== RUN %s\n", chatty.prefix(), eg.Name)
}
// Capture stdout to temporary file. We're not using
// os.Pipe because it is not supported on js/wasm.
stdout := os.Stdout
f := createTempFile(eg.Name)
os.Stdout = f
finished := false
start := time.Now()
// Clean up in a deferred call so we can recover if the example panics.
defer func() {
timeSpent := time.Since(start)
// Restore stdout, get output and remove temporary file.
os.Stdout = stdout
var buf strings.Builder
_, seekErr := f.Seek(0, io.SeekStart)
_, readErr := io.Copy(&buf, f)
out := buf.String()
f.Close()
os.Remove(f.Name())
if seekErr != nil {
fmt.Fprintf(os.Stderr, "testing: seek temp file: %v\n", seekErr)
os.Exit(1)
}
if readErr != nil {
fmt.Fprintf(os.Stderr, "testing: read temp file: %v\n", readErr)
os.Exit(1)
}
err := recover()
ok = eg.processRunResult(out, timeSpent, finished, err)
}()
// Run example.
eg.F()
finished = true
return
}
func createTempFile(exampleName string) *os.File {
for i := 0; ; i++ {
name := fmt.Sprintf("%s/go-example-stdout-%s-%d.txt", os.TempDir(), exampleName, i)
f, err := os.OpenFile(name, os.O_RDWR|os.O_CREATE|os.O_EXCL, 0600)
if err != nil {
if os.IsExist(err) {
continue
}
fmt.Fprintf(os.Stderr, "testing: open temp file: %v\n", err)
os.Exit(1)
}
return f
}
}
|
