forked from gotestyourself/gotestsum
-
Notifications
You must be signed in to change notification settings - Fork 0
/
handler.go
87 lines (76 loc) · 2 KB
/
handler.go
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
77
78
79
80
81
82
83
84
85
86
87
package main
import (
"io"
"os"
"github.com/pkg/errors"
log "github.com/sirupsen/logrus"
"gotest.tools/gotestsum/internal/junitxml"
"gotest.tools/gotestsum/testjson"
)
type eventHandler struct {
formatter testjson.EventFormatter
out io.Writer
err io.Writer
jsonFile io.WriteCloser
}
func (h *eventHandler) Err(text string) error {
_, err := h.err.Write([]byte(text + "\n"))
return err
}
func (h *eventHandler) Event(event testjson.TestEvent, execution *testjson.Execution) error {
if h.jsonFile != nil {
_, err := h.jsonFile.Write(append(event.Bytes(), '\n'))
if err != nil {
return errors.Wrap(err, "failed to write JSON file")
}
}
line, err := h.formatter(event, execution)
if err != nil {
return errors.Wrap(err, "failed to format event")
}
_, err = h.out.Write([]byte(line))
return errors.Wrap(err, "failed to write event")
}
func (h *eventHandler) Close() error {
if h.jsonFile != nil {
if err := h.jsonFile.Close(); err != nil {
log.WithError(err).Error("failed to close JSON file")
}
}
return nil
}
var _ testjson.EventHandler = &eventHandler{}
func newEventHandler(opts *options, wout io.Writer, werr io.Writer) (*eventHandler, error) {
formatter := testjson.NewEventFormatter(opts.format)
if formatter == nil {
return nil, errors.Errorf("unknown format %s", opts.format)
}
handler := &eventHandler{
formatter: formatter,
out: wout,
err: werr,
}
var err error
if opts.jsonFile != "" {
handler.jsonFile, err = os.Create(opts.jsonFile)
if err != nil {
return handler, errors.Wrap(err, "failed to open JSON file")
}
}
return handler, nil
}
func writeJUnitFile(filename string, execution *testjson.Execution) error {
if filename == "" {
return nil
}
junitFile, err := os.Create(filename)
if err != nil {
return errors.Wrap(err, "failed to open JUnit file")
}
defer func() {
if err := junitFile.Close(); err != nil {
log.WithError(err).Error("failed to close JUnit file")
}
}()
return junitxml.Write(junitFile, execution)
}