Parameterize counting visible lines

zenozeng · Aug 17, 2016 · 3d6d47e · 3d6d47e
1 parent 846dba6
commit 3d6d47e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/features/step-definitions/yafolding-steps.el b/features/step-definitions/yafolding-steps.el
@@ -22,10 +22,11 @@
 (And "^I call yafolding-hide-element$"
  (lambda () (yafolding-hide-element)))
 
-(Then "^I should see only one line$"
- (lambda ()
+(Then "^I should see \\(?:only \\)?\\([0-9]+\\) lines?$"
+ (lambda (num-lines)
  (save-excursion
- (next-line)
+ (beginning-of-buffer)
+ (next-line (string-to-number num-lines))
  (should (= (line-number-at-pos (point))
  (line-number-at-pos (point-max)))))))
 

diff --git a/features/yafolding.feature b/features/yafolding.feature
@@ -14,4 +14,4 @@ Feature: Fold all lines
  When I call yafolding-hide-all
  And I go to line 1
  And I call yafolding-hide-element
- Then I should see only one line
+ Then I should see only 1 line