diff --git a/template/test/e2e/custom-assertions/elementCount.js b/template/test/e2e/custom-assertions/elementCount.js index c0103f9431..972e9508ab 100644 --- a/template/test/e2e/custom-assertions/elementCount.js +++ b/template/test/e2e/custom-assertions/elementCount.js @@ -1,11 +1,12 @@ // A custom Nightwatch assertion. -// the name of the method is the filename. -// can be used in tests like this: +// The assertion name is the filename. +// Example usage: // // browser.assert.elementCount(selector, count) // -// for how to write custom assertions see +// For more information on custom assertions see: // http://nightwatchjs.org/guide#writing-custom-assertions + exports.assertion = function (selector, count) { this.message = 'Testing if element <' + selector + '> has count: ' + count{{#if_eq lintConfig "airbnb"}};{{/if_eq}} this.expected = count{{#if_eq lintConfig "airbnb"}};{{/if_eq}} @@ -17,8 +18,8 @@ exports.assertion = function (selector, count) { } this.command = function (cb) { var self = this{{#if_eq lintConfig "airbnb"}};{{/if_eq}} - return this.api.execute(function (selector) { - return document.querySelectorAll(selector).length{{#if_eq lintConfig "airbnb"}};{{/if_eq}} + return this.api.execute(function (selectorToCount) { + return document.querySelectorAll(selectorToCount).length{{#if_eq lintConfig "airbnb"}};{{/if_eq}} }, [selector], function (res) { cb.call(self, res){{#if_eq lintConfig "airbnb"}};{{/if_eq}} }){{#if_eq lintConfig "airbnb"}};{{/if_eq}}