godog/suite_test.go

66 строки
1,4 КиБ
Go

package godog
import (
"fmt"
"regexp"
"github.com/DATA-DOG/godog/gherkin"
)
type suiteFeature struct {
suite
}
func (s *suiteFeature) BeforeScenario(scenario *gherkin.Scenario) {
// reset feature paths
cfg.paths = []string{}
}
func (s *suiteFeature) featurePath(args ...*Arg) error {
cfg.paths = append(cfg.paths, args[0].String())
return nil
}
func (s *suiteFeature) parseFeatures(args ...*Arg) (err error) {
s.features, err = cfg.features()
return
}
func (s *suiteFeature) numParsed(args ...*Arg) (err error) {
if len(s.features) != args[0].Int() {
err = fmt.Errorf("expected %d features to be parsed, but have %d", args[0].Int(), len(s.features))
}
return
}
func (s *suiteFeature) numScenariosRegistered(args ...*Arg) (err error) {
var num int
for _, ft := range s.features {
num += len(ft.Scenarios)
}
if num != args[0].Int() {
err = fmt.Errorf("expected %d scenarios to be registered, but got %d", args[0].Int(), num)
}
return
}
func SuiteContext(g Suite) {
s := &suiteFeature{
suite: suite{},
}
g.BeforeScenario(s)
g.Step(
regexp.MustCompile(`^a feature path "([^"]*)"$`),
StepHandlerFunc(s.featurePath))
g.Step(
regexp.MustCompile(`^I parse features$`),
StepHandlerFunc(s.parseFeatures))
g.Step(
regexp.MustCompile(`^I should have ([\d]+) features? files?$`),
StepHandlerFunc(s.numParsed))
g.Step(
regexp.MustCompile(`^I should have ([\d]+) scenarios? registered$`),
StepHandlerFunc(s.numScenariosRegistered))
}