return update query single element

afontainec · Sep 27, 2019 · 30f0169 · 30f0169
1 parent a054a80
commit 30f0169
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/models/table.js b/models/table.js
@@ -137,9 +137,12 @@ class Table {
   // UDATE
   // ################################################
 
-  update(input, newValues, options) {
+  async update(input, newValues, options) {
     const whereQuery = typeof input === 'object' ? input : { id: input };
-    return this.updateWhere(whereQuery, newValues, options);
+    const query = this.updateWhere(whereQuery, newValues, options);
+    if (Table.returnAsQuery(options)) return query;
+    const result = await Table.fetchQuery(query);
+    return result[0];
   }
 
   updateQuery(whereQuery, values, options) {

diff --git a/test/integrationTests/tablegateway/update/update.js b/test/integrationTests/tablegateway/update/update.js
@@ -37,6 +37,13 @@ describe('TABLE GATEWAY: update', () => { // eslint-disable-line
     assert.notEqual(entry.updated_at.toString(), updated.updated_at.toString());
   });
 
+  it('update entry', async () => { // eslint-disable-line
+    const entry = await Coffee.findById(1);
+    entry.name = 'updated';
+    const updated = await Coffee.update(1, Utils.cloneJSON(entry));
+    assert.equal(updated.name, 'updated');
+  });
+
 
   it('update entry, without givin the id', async () => { // eslint-disable-line
     const entry = {};