fixed bug

lib/legit-tests.js

@@ -37,36 +37,20 @@ var Test = (function () {
   }, {
     key: 'callMiddleware',
     value: function callMiddleware() {
-      var _iteratorNormalCompletion = true;
-      var _didIteratorError = false;
-      var _iteratorError = undefined;
-
-      try {
-        for (var _iterator = this.middleware[Symbol.iterator](), _step; !(_iteratorNormalCompletion = (_step = _iterator.next()).done); _iteratorNormalCompletion = true) {
-          var middleware = _step.value;
-
-          middleware.call();
-        }
-      } catch (err) {
-        _didIteratorError = true;
-        _iteratorError = err;
-      } finally {
-        try {
-          if (!_iteratorNormalCompletion && _iterator['return']) {
-            _iterator['return']();
-          }
-        } finally {
-          if (_didIteratorError) {
-            throw _iteratorError;
-          }
-        }
+      for (var i in this.middleware) {
+        this.middleware[i].call();
       }
+      this.middleware = [];
     }
   }, {
-    key: 'end',
-    value: function end(callback) {
+    key: 'test',
+    value: function test(callback) {
       this.callMiddleware();
-      callback.call(this, this.component, this.helpers);
+      callback.call({
+        component: this.component,
+        helpers: this.helpers
+      });
+      return this;
     }
   }]);
 

lib/middleware.js

@@ -0,0 +1,21 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', {
+  value: true
+});
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { 'default': obj }; }
+
+var _middlewareFind = require('./middleware/find');
+
+var _middlewareFind2 = _interopRequireDefault(_middlewareFind);
+
+var _middlewareSetState = require('./middleware/setState');
+
+var _middlewareSetState2 = _interopRequireDefault(_middlewareSetState);
+
+exports['default'] = {
+  Find: _middlewareFind2['default'],
+  SetState: _middlewareSetState2['default']
+};
+module.exports = exports['default'];

lib/middleware/find.js

@@ -15,15 +15,13 @@ var TestUtils = _react2['default'].addons.TestUtils;
 
 function find(selector) {
   var elements = undefined;
-  if (typeof selector === 'string') {
-    elements = TestUtils.scryRenderedDOMComponentsWithTag(this.component, selector);
-    this.helpers[selector] = elements;
-  } else {
-    if (selector['class']) {
-      elements = TestUtils.scryRenderedDOMComponentsWithClass(this.component, selector['class']);
-      this.helpers[selector['class']] = elements;
-    }
-  }
+  if (selector.match(/\./)) {
+    selector = selector.replace(/\./, '');
+    elements = TestUtils.scryRenderedDOMComponentsWithClass(this.component, selector);
+  } else elements = TestUtils.scryRenderedDOMComponentsWithTag(this.component, selector);
+
+  if (elements.length === 1) elements = elements[0];
+  this.helpers[selector] = elements;
 }
 
 module.exports = exports['default'];

package.json

@@ -1,6 +1,6 @@
 {
   "name": "legit-tests",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "a chainable testing library for React",
   "main": "lib/legit-tests.js",
   "scripts": {

src/legit-tests.js

@@ -25,7 +25,7 @@ class Test {
   test(callback) {
     this.callMiddleware()
     callback.call({
-      middleware: this.middleware,
+      component: this.component,
       helpers: this.helpers
     })
     return this

tests/setState.jsx

@@ -13,12 +13,12 @@ describe('setState middleware', () => {
     Test(<TestComponent/>)
     .use(SetState, {test: 'test'})
     .use(Find, 'div')
-    .test(function(component, helpers) {
+    .test(function() {
       expect(this.helpers.div.props.children).to.be.equal('test')
     })
     .use(SetState, {test: 'changed!'})
     .test(function() {
-      expect(this.helpers.div.props.children).to.be.equal('changed!')
+      expect(this.component.state.test).to.be.equal('changed!')
     })
   });