forked from manifoldco/grafton
/
logging.go
137 lines (113 loc) · 2.31 KB
/
logging.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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package acceptance
import (
"fmt"
"strings"
"github.com/manifoldco/promptui"
)
var (
bold = promptui.Styler(promptui.FGBold)
faint = promptui.Styler(promptui.FGFaint)
)
type resultCode int
const (
pass resultCode = iota
fail
)
// LogLevel is the type for setting our informational log level, irrespective of
// test results.
type LogLevel string
// Log Levels the user can set.
const (
LogOff LogLevel = "off"
LogInfo LogLevel = "info"
LogVerbose LogLevel = "verbose"
)
func (l LogLevel) show(ol LogLevel) bool {
switch {
case l == "off":
return false
case l == "info":
return ol != LogVerbose
default:
return true
}
}
var lvl LogLevel
// SetLogLevel configures the log level for the duration of an acceptance test
// run.
func SetLogLevel(level LogLevel) {
lvl = level
}
// Infof prints the given format string and args at the appropriate indentation
// level, if the logger is at info level or above.
func Infof(format string, args ...interface{}) {
if lvl.show(LogInfo) {
printIndented(fmt.Sprintf(format, args...))
}
}
// Infoln prints the given args at the appropriate indentation level, if the
// logger is at info level or above.
func Infoln(args ...interface{}) {
if lvl.show(LogInfo) {
printIndented(fmt.Sprintln(args...))
}
}
var indent = 0
var entered = false
func enter(msg string) {
printIndented(msg)
entered = true
indent += 2
}
func exit() {
if entered {
fmt.Println()
entered = false
}
indent -= 2
}
func result(name string, code resultCode) {
if !entered {
indent -= 2
printIndented(faint(name))
indent += 2
}
var msg string
switch code {
case pass:
msg = promptui.IconGood
success++
case fail:
msg = promptui.IconBad
failures++
}
fmt.Println(" " + msg)
entered = false
}
func printIndented(msg string) {
if entered {
fmt.Println()
entered = false
}
prefix := strings.Repeat(" ", indent)
parts := strings.Split(msg, "\n")
for i, part := range parts {
if i == len(parts)-1 && part == "" {
continue
}
fmt.Print(prefix + part)
if i != len(parts)-1 {
fmt.Println()
}
}
}
func printSummary(fails, success int) {
styler := promptui.Styler(promptui.FGGreen)
if fails > 0 {
styler = promptui.Styler(promptui.FGRed)
}
msg := fmt.Sprintf("%d features, %d failures", fails+success, fails)
fmt.Println()
enter(styler(msg))
exit()
}