@kravc/dos 1.11.7 → 1.11.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kravc/dos",
-  "version": "1.11.7",
+  "version": "1.11.9",
   "description": "Convention-based, easy-to-use library for building API-driven serverless services.",
   "keywords": [
     "Service",

package/src/Document.js

@@ -154,6 +154,8 @@ class Document extends Component {
   }
 
   static async index(context, query = {}, options = {}) {
+    this._extendWithPartition(context, query)
+
     let { items, ...rest } = await this._index(query, options)
 
     const objects = items.map(item => new this(context, item))
@@ -174,6 +176,8 @@ class Document extends Component {
   }
 
   static async indexAll(context, query = {}, options = {}) {
+    this._extendWithPartition(context, query)
+
     let { items, ...rest } = await this._indexAll(query, options)
 
     const objects = items.map(item => new this(context, item))
@@ -203,11 +207,13 @@ class Document extends Component {
 
     /* NOTE: ensure that document to be updated exists and save it in the
              context so can be referenced in the after action helper */
-    if (!originalDocument) {
+    if (originalDocument) {
+      this._extendWithPartition(context, query)
+
+    } else {
       originalDocument = await this.read(context, query)
-    }
 
-    this._extendWithPartition(context, query)
+    }
 
     const updatedItem = await this._update(query, mutation)
 
@@ -243,8 +249,11 @@ class Document extends Component {
     /* NOTE: ensure that document to be removed exists and save it in the
              context so can be referenced in the after action helper */
     const originalDocument = await this.read(context, query)
+    const hasQueryPartition = !!query.partition
 
-    this._extendWithPartition(context, query)
+    if (!hasQueryPartition) {
+      this._extendWithPartition(context, query)
+    }
 
     await this._delete(query, context)