Add source to messages to onParse callback

Co-authored-by: Julien Biezemans <jb@jbpros.com>
SmartBear · Jan 28, 2021 · 56d77d6 · 56d77d6
1 parent 55cd62d
commit 56d77d6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/lib/gherkin-linter/index.ts b/src/lib/gherkin-linter/index.ts
@@ -87,7 +87,7 @@ export default class GherkinLinter {
     const messages = generateMessages(
       this.getContentToLint(gherkin),
       '',
-      this.options
+      { includeSource: !!this.onParse, ...this.options }
     )
 
     this.lintingErrors = messages

diff --git a/test/lib/gherkin-linter/index.spec.js b/test/lib/gherkin-linter/index.spec.js
@@ -413,9 +413,11 @@ describe('GherkinLinter', () => {
     it('calls back with the cucumber message envelopes', () => {
       let messages
       const gherkinLinter = new GherkinLinter(msgs => (messages = msgs))
-      gherkinLinter.parse('Feature: our feature\n')
+      const source = 'Feature: our feature\n'
+      gherkinLinter.parse(source)
       expect(messages.length).toBeGreaterThan(0)
-      expect(messages[0].gherkinDocument.feature.name).toEqual('our feature')
+      expect(messages[0].source.data).toEqual(source)
+      expect(messages[1].gherkinDocument.feature.name).toEqual('our feature')
     })
   })
 })