added 'prefix' option to Table.prototype.star()

tmont · tmont · commit 26346fbbb243 · 2013-10-04T18:38:08.000-07:00
diff --git a/lib/table.js b/lib/table.js
@@ -121,7 +121,14 @@ Table.prototype.getName = function() {
   return this._name;
 };
 
-Table.prototype.star = function() {
+Table.prototype.star = function(options) {
+  options = options || {};
+  if (options.prefix) {
+    return this.columns.map(function(column) {
+      return this[column.name].as(options.prefix + column.name);
+    }.bind(this));
+  }
+
   return new Column({table: this, star: true});
 };
 
diff --git a/test/dialects/table-tests.js b/test/dialects/table-tests.js
@@ -54,6 +54,23 @@ Harness.test({
   params: []
 });
 
+Harness.test({
+  query: user.select(user.star({ prefix: 'foo_' })).from(user),
+  pg: {
+    text: 'SELECT "user"."id" AS "foo_id", "user"."name" AS "foo_name" FROM "user"',
+    string: 'SELECT "user"."id" AS "foo_id", "user"."name" AS "foo_name" FROM "user"'
+  },
+  sqlite: {
+    text: 'SELECT "user"."id" AS "foo_id", "user"."name" AS "foo_name" FROM "user"',
+    string: 'SELECT "user"."id" AS "foo_id", "user"."name" AS "foo_name" FROM "user"'
+  },
+  mysql: {
+    text: 'SELECT `user`.`id` AS `foo_id`, `user`.`name` AS `foo_name` FROM `user`',
+    string: 'SELECT `user`.`id` AS `foo_id`, `user`.`name` AS `foo_name` FROM `user`'
+  },
+  params: []
+});
+
 Harness.test({
   query: user.select(user.id).from(user).where(user.name.equals('foo')),
   pg: {
