Add limit function to tables.

braddunbar · braddunbar · commit c9500b505ec0 · 2015-08-12T22:01:04.000-04:00
diff --git a/lib/table.js b/lib/table.js
@@ -297,4 +297,10 @@ Table.prototype.indexes = function() {
   return new Query(this).indexes();
 };
 
+Table.prototype.limit = function () {
+  var query = new Query(this);
+  query.limit.apply(query, arguments);
+  return query;
+};
+
 module.exports = Table;
diff --git a/test/table-tests.js b/test/table-tests.js
@@ -235,3 +235,14 @@ test('dialects', function () {
   actual = foo.join(bar).on(bar.id.equals(1)).toString();
   assert.equal(actual, '"foo" INNER JOIN "bar" ON ("bar"."id" = 1)');
 });
+
+test('limit', function () {
+  var user = Table.define({
+    name: 'user',
+    columns: ['id', 'name']
+  });
+  var query = user.limit(3);
+  assert.equal(query.nodes.length, 1);
+  assert.equal(query.nodes[0].type, 'LIMIT');
+  assert.equal(query.nodes[0].count, 3);
+});
