154 строки
3,4 КиБ
Go
154 строки
3,4 КиБ
Go
package godog
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"math"
|
|
"strings"
|
|
"sync"
|
|
|
|
"github.com/DATA-DOG/godog/gherkin"
|
|
)
|
|
|
|
func init() {
|
|
Format("progress", "Prints a character per step.", progressFunc)
|
|
}
|
|
|
|
func progressFunc(suite string, out io.Writer) Formatter {
|
|
steps := 0
|
|
return &progress{
|
|
basefmt: basefmt{
|
|
started: timeNowFunc(),
|
|
indent: 2,
|
|
out: out,
|
|
},
|
|
stepsPerRow: 70,
|
|
lock: new(sync.Mutex),
|
|
steps: &steps,
|
|
}
|
|
}
|
|
|
|
type progress struct {
|
|
basefmt
|
|
lock *sync.Mutex
|
|
stepsPerRow int
|
|
steps *int
|
|
}
|
|
|
|
func (f *progress) Node(n interface{}) {
|
|
f.lock.Lock()
|
|
defer f.lock.Unlock()
|
|
f.basefmt.Node(n)
|
|
}
|
|
|
|
func (f *progress) Feature(ft *gherkin.Feature, p string, c []byte) {
|
|
f.lock.Lock()
|
|
defer f.lock.Unlock()
|
|
f.basefmt.Feature(ft, p, c)
|
|
}
|
|
|
|
func (f *progress) Summary() {
|
|
left := math.Mod(float64(*f.steps), float64(f.stepsPerRow))
|
|
if left != 0 {
|
|
if *f.steps > f.stepsPerRow {
|
|
fmt.Fprintf(f.out, s(f.stepsPerRow-int(left))+fmt.Sprintf(" %d\n", *f.steps))
|
|
} else {
|
|
fmt.Fprintf(f.out, " %d\n", *f.steps)
|
|
}
|
|
}
|
|
fmt.Fprintln(f.out, "")
|
|
|
|
if len(f.failed) > 0 {
|
|
fmt.Fprintln(f.out, "\n--- "+red("Failed steps:")+"\n")
|
|
for _, fail := range f.failed {
|
|
fmt.Fprintln(f.out, s(2)+red(fail.scenarioDesc())+blackb(" # "+fail.scenarioLine()))
|
|
fmt.Fprintln(f.out, s(4)+red(strings.TrimSpace(fail.step.Keyword)+" "+fail.step.Text)+blackb(" # "+fail.line()))
|
|
fmt.Fprintln(f.out, s(6)+red("Error: ")+redb(fmt.Sprintf("%+v", fail.err))+"\n")
|
|
}
|
|
}
|
|
f.basefmt.Summary()
|
|
}
|
|
|
|
func (f *progress) step(res *stepResult) {
|
|
switch res.typ {
|
|
case passed:
|
|
fmt.Fprint(f.out, green("."))
|
|
case skipped:
|
|
fmt.Fprint(f.out, cyan("-"))
|
|
case failed:
|
|
fmt.Fprint(f.out, red("F"))
|
|
case undefined:
|
|
fmt.Fprint(f.out, yellow("U"))
|
|
case pending:
|
|
fmt.Fprint(f.out, yellow("P"))
|
|
}
|
|
*f.steps++
|
|
if math.Mod(float64(*f.steps), float64(f.stepsPerRow)) == 0 {
|
|
fmt.Fprintf(f.out, " %d\n", *f.steps)
|
|
}
|
|
}
|
|
|
|
func (f *progress) Passed(step *gherkin.Step, match *StepDef) {
|
|
f.lock.Lock()
|
|
defer f.lock.Unlock()
|
|
f.basefmt.Passed(step, match)
|
|
f.step(f.passed[len(f.passed)-1])
|
|
}
|
|
|
|
func (f *progress) Skipped(step *gherkin.Step, match *StepDef) {
|
|
f.lock.Lock()
|
|
defer f.lock.Unlock()
|
|
f.basefmt.Skipped(step, match)
|
|
f.step(f.skipped[len(f.skipped)-1])
|
|
}
|
|
|
|
func (f *progress) Undefined(step *gherkin.Step, match *StepDef) {
|
|
f.lock.Lock()
|
|
defer f.lock.Unlock()
|
|
f.basefmt.Undefined(step, match)
|
|
f.step(f.undefined[len(f.undefined)-1])
|
|
}
|
|
|
|
func (f *progress) Failed(step *gherkin.Step, match *StepDef, err error) {
|
|
f.lock.Lock()
|
|
defer f.lock.Unlock()
|
|
f.basefmt.Failed(step, match, err)
|
|
f.step(f.failed[len(f.failed)-1])
|
|
}
|
|
|
|
func (f *progress) Pending(step *gherkin.Step, match *StepDef) {
|
|
f.lock.Lock()
|
|
defer f.lock.Unlock()
|
|
f.basefmt.Pending(step, match)
|
|
f.step(f.pending[len(f.pending)-1])
|
|
}
|
|
|
|
func (f *progress) Sync(cf ConcurrentFormatter) {
|
|
if source, ok := cf.(*progress); ok {
|
|
f.lock = source.lock
|
|
f.steps = source.steps
|
|
}
|
|
}
|
|
|
|
func (f *progress) Copy(cf ConcurrentFormatter) {
|
|
if source, ok := cf.(*progress); ok {
|
|
for _, v := range source.features {
|
|
f.features = append(f.features, v)
|
|
}
|
|
for _, v := range source.failed {
|
|
f.failed = append(f.failed, v)
|
|
}
|
|
for _, v := range source.passed {
|
|
f.passed = append(f.passed, v)
|
|
}
|
|
for _, v := range source.skipped {
|
|
f.skipped = append(f.skipped, v)
|
|
}
|
|
for _, v := range source.undefined {
|
|
f.undefined = append(f.undefined, v)
|
|
}
|
|
for _, v := range source.pending {
|
|
f.pending = append(f.pending, v)
|
|
}
|
|
}
|
|
}
|