From 5566c15759f5c510995826400ac4f08aea2950bb Mon Sep 17 00:00:00 2001
From: radtriste <tradisso@redhat.com>
Date: Tue, 11 Feb 2020 10:25:39 +0100
Subject: [PATCH] added GetFeatures method to godog.Suite

---
 run.go | 50 +++++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 41 insertions(+), 9 deletions(-)

diff --git a/run.go b/run.go
index c230e50e..398bb532 100644
--- a/run.go
+++ b/run.go
@@ -285,18 +285,50 @@ type TestSuite struct {
 // If there are flag related errors they will be directed to os.Stderr
 func (ts TestSuite) Run() int {
 	if ts.Options == nil {
-		ts.Options = &Options{}
-		ts.Options.Output = colors.Colored(os.Stdout)
-
-		flagSet := flagSet(ts.Options)
-		if err := flagSet.Parse(os.Args[1:]); err != nil {
-			fmt.Fprintln(os.Stderr, err)
+		var err error
+		ts.Options, err = getDefaultOptions()
+		if err != nil {
 			return exitOptionError
 		}
-
-		ts.Options.Paths = flagSet.Args()
 	}
-
 	r := runner{testSuiteInitializer: ts.TestSuiteInitializer, scenarioInitializer: ts.ScenarioInitializer}
 	return runWithOptions(ts.Name, r, *ts.Options)
 }
+
+// RetrieveFeatures will parse and return the features based on test suite option
+// Any modification on the parsed features will not have any impact on the next Run of the Test Suite
+func (ts TestSuite) RetrieveFeatures() ([]*models.Feature, error) {
+	opt := ts.Options
+
+	if opt == nil {
+		var err error
+		opt, err = getDefaultOptions()
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	if len(opt.Paths) == 0 {
+		inf, err := os.Stat("features")
+		if err == nil && inf.IsDir() {
+			opt.Paths = []string{"features"}
+		}
+	}
+
+	return parser.ParseFeatures(opt.Tags, opt.Paths)
+}
+
+func getDefaultOptions() (*Options, error) {
+	opt := &Options{}
+	opt.Output = colors.Colored(os.Stdout)
+
+	flagSet := flagSet(opt)
+	if err := flagSet.Parse(os.Args[1:]); err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		return nil, err
+	}
+
+	opt.Paths = flagSet.Args()
+
+	return opt, nil
+}