@eresearchqut/ddb-repository 1.2.0 → 1.4.0

package/CHANGELOG.md

@@ -1,3 +1,17 @@
+# [1.4.0](https://github.com/eresearchqut/ddb-repository/compare/v1.3.0...v1.4.0) (2025-11-20)
+
+
+### Features
+
+* Add test for the tracking of the consumed capacity and rename test class ([d113a5a](https://github.com/eresearchqut/ddb-repository/commit/d113a5a948a7dffc80e9d3a65a11203c2249ed26))
+
+# [1.3.0](https://github.com/eresearchqut/ddb-repository/compare/v1.2.0...v1.3.0) (2025-11-20)
+
+
+### Features
+
+* Add test for the tracking of the consumed capacity ([4a1b23c](https://github.com/eresearchqut/ddb-repository/commit/4a1b23cadadbbb02d78507ca9d34715219fc952c))
+
 # [1.2.0](https://github.com/eresearchqut/ddb-repository/compare/v1.1.0...v1.2.0) (2025-11-19)
 
 

package/dist/DynamoDbRepository.js

@@ -27,7 +27,7 @@ var __asyncValues = (this && this.__asyncValues) || function (o) {
     function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DynamoDbRepository = exports.mapFilterExpressions = exports.mapFilterExpression = exports.FilterOperator = void 0;
+exports.DynamoDbRepository = exports.consumedCapacityMiddleware = exports.mapFilterExpressions = exports.mapFilterExpression = exports.FilterOperator = void 0;
 const client_dynamodb_1 = require("@aws-sdk/client-dynamodb");
 const util_dynamodb_1 = require("@aws-sdk/util-dynamodb");
 const lodash_1 = require("lodash");
@@ -79,6 +79,21 @@ const paginate = (array, pageSize) => {
         return acc;
     }, []);
 };
+const consumedCapacityMiddleware = (consumedCapacityMiddlewareConfig) => (next, context) => (args) => __awaiter(void 0, void 0, void 0, function* () {
+    try {
+        const { input } = args;
+        const { ReturnConsumedCapacity } = input;
+        const response = yield next(args);
+        const { output } = response;
+        const consumedCapacity = (0, lodash_1.get)(output, "ConsumedCapacity");
+        yield consumedCapacityMiddlewareConfig.onConsumedCapacity({ ReturnConsumedCapacity, ConsumedCapacity: consumedCapacity });
+        return response;
+    }
+    catch (error) {
+        throw error;
+    }
+});
+exports.consumedCapacityMiddleware = consumedCapacityMiddleware;
 class DynamoDbRepository {
     constructor(dynamoDBClient, tableName, hashKey, rangKey, returnConsumedCapacity = client_dynamodb_1.ReturnConsumedCapacity.TOTAL) {
         this.dynamoDBClient = dynamoDBClient;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eresearchqut/ddb-repository",
-  "version": "1.2.0",
+  "version": "1.4.0",
   "description": "",
   "type": "module",
   "main": "dist/index.js",
@@ -23,6 +23,7 @@
     "@semantic-release/git": "^10.0.1",
     "@semantic-release/github": "^12.0.2",
     "@semantic-release/npm": "^13.1.2",
+    "@smithy/types": "^4.9.0",
     "@testcontainers/localstack": "^10.15.0",
     "@types/jest": "^29.5.0",
     "@types/lodash": "^4.17.20",

package/src/DynamoDbRepository.ts

@@ -1,6 +1,7 @@
 import {
     BatchGetItemCommand,
     BatchGetItemCommandInput,
+    ConsumedCapacity,
     DeleteItemCommand,
     DynamoDBClient,
     GetItemCommand,
@@ -11,7 +12,14 @@ import {
     UpdateItemCommand,
 } from "@aws-sdk/client-dynamodb";
 import {marshall, unmarshall} from "@aws-sdk/util-dynamodb";
-import {replace, uniqWith, isEqual, pickBy} from "lodash";
+import {replace, uniqWith, isEqual, pickBy, get} from "lodash";
+import {
+    HandlerExecutionContext,
+    InitializeHandler,
+    InitializeHandlerArguments,
+    InitializeHandlerOutput,
+    MetadataBearer
+} from "@smithy/types";
 
 const expressionAttributeKey = (key: string) => replace(key, /-/g, "_");
 
@@ -119,6 +127,35 @@ const paginate = <T>(array: Array<T>, pageSize: number) => {
     }, [] as Array<Array<T>>);
 }
 
+export interface ConsumedCapacityDetail {
+    ReturnConsumedCapacity: ReturnConsumedCapacity | undefined
+    ConsumedCapacity: ConsumedCapacity | ConsumedCapacity[] | undefined
+}
+
+export interface ConsumedCapacityMiddlewareConfig {
+    onConsumedCapacity: (consumedCapacity: ConsumedCapacityDetail) => Promise<unknown>;
+}
+
+export const consumedCapacityMiddleware =
+    (consumedCapacityMiddlewareConfig: ConsumedCapacityMiddlewareConfig) =>
+        <Output extends MetadataBearer = MetadataBearer>(
+            next: InitializeHandler<any, Output>,
+            context: HandlerExecutionContext
+        ): InitializeHandler<any, Output> =>
+            async (args: InitializeHandlerArguments<any>): Promise<InitializeHandlerOutput<Output>> => {
+                try {
+                    const {input} = args;
+                    const {ReturnConsumedCapacity} = input;
+                    const response = await next(args);
+                    const {output} = response;
+                    const consumedCapacity = get(output, "ConsumedCapacity") as ConsumedCapacity | undefined;
+                    await consumedCapacityMiddlewareConfig.onConsumedCapacity({ReturnConsumedCapacity, ConsumedCapacity: consumedCapacity});
+                    return response;
+                } catch (error) {
+                    throw error;
+                }
+            };
+
 export class DynamoDbRepository<K, T> {
 
     constructor(

package/test/{repository.test.ts → DynamoDbRepository.test.ts}

@@ -1,7 +1,7 @@
 
 import { DynamoDBClient, CreateTableCommand, DescribeTableCommand } from "@aws-sdk/client-dynamodb";
 import { LocalstackContainer, StartedLocalStackContainer } from "@testcontainers/localstack";
-import {DynamoDbRepository, FilterOperator} from "../src";
+import {ConsumedCapacityDetail, consumedCapacityMiddleware, DynamoDbRepository, FilterOperator} from "../src";
 
 describe('DynamoDbRepository Integration Tests', () => {
     let container: StartedLocalStackContainer;
@@ -12,6 +12,19 @@ describe('DynamoDbRepository Integration Tests', () => {
     const tableName = 'test-table';
     const compositeTableName = 'test-composite-table';
     const gsiTableName = 'test-gsi-table';
+    const consumedCapacityRegister = new Array<ConsumedCapacityDetail>() ;
+
+    const getConsumedCapacity = (consumedCapacity: ConsumedCapacityDetail) => {
+        if (Array.isArray(consumedCapacity.ConsumedCapacity)) {
+            return consumedCapacity.ConsumedCapacity
+                .reduce((total, capacity) => total + (capacity?.CapacityUnits || 0), 0);
+        }
+        return consumedCapacity.ConsumedCapacity?.CapacityUnits || 0;
+    }
+
+    const sumConsumedCapacity = () =>
+        consumedCapacityRegister.reduce((total, value) =>
+            total + getConsumedCapacity(value), 0);
 
     beforeAll(async () => {
         // Start LocalStack container with DynamoDB
@@ -28,7 +41,11 @@ describe('DynamoDbRepository Integration Tests', () => {
             },
         });
 
-        // Create the test table with simple key
+        dynamoDBClient.middlewareStack
+            .add(consumedCapacityMiddleware({onConsumedCapacity: async (consumedCapacity) =>
+                    consumedCapacityRegister.push(consumedCapacity)}));
+
+        // Create the test table with a simple key
         await dynamoDBClient.send(
             new CreateTableCommand({
                 TableName: tableName,
@@ -42,7 +59,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             })
         );
 
-        // Create the test table with composite key (partition key + sort key)
+        // Create the test table with a composite key (partition key + sort key)
         await dynamoDBClient.send(
             new CreateTableCommand({
                 TableName: compositeTableName,
@@ -108,6 +125,10 @@ describe('DynamoDbRepository Integration Tests', () => {
         gsiRepository = new DynamoDbRepository(dynamoDBClient, gsiTableName, "userId", "itemId");
     });
 
+    beforeEach(async () => {
+        consumedCapacityRegister.splice(0, consumedCapacityRegister.length);
+    });
+
     afterAll(async () => {
         // Clean up
         if (dynamoDBClient) {
@@ -122,6 +143,7 @@ describe('DynamoDbRepository Integration Tests', () => {
         it('should return undefined when item does not exist', async () => {
             const result = await repository.getItem({ id: 'non-existent' });
             expect(result).toBeUndefined();
+            expect(sumConsumedCapacity()).toEqual(0.5);
         });
 
         it('should return the item when it exists', async () => {
@@ -132,6 +154,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             const result = await repository.getItem(key);
 
             expect(result).toEqual(record);
+            expect(sumConsumedCapacity()).toEqual(2);
         });
     });
 
@@ -143,6 +166,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             const result = await repository.putItem(key, record);
 
             expect(result).toEqual(record);
+            expect(sumConsumedCapacity()).toEqual(1.5);
         });
     });
 
@@ -155,6 +179,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             await repository.deleteItem(key);
             const afterDelete = await repository.getItem(key);
             expect(afterDelete).toBeUndefined();
+            expect(sumConsumedCapacity()).toEqual(2);
         });
     });
 
@@ -171,6 +196,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 name: 'Updated Name',
                 email: 'original@example.com'
             });
+            expect(sumConsumedCapacity()).toEqual(3);
         });
     });
 
@@ -183,6 +209,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                     { id: `batch-item-${i}`, name: `Item ${i}`, age: i * 10 }
                 );
             }
+            consumedCapacityRegister.splice(0, consumedCapacityRegister.length);
         });
 
         it('should retrieve multiple items by keys', async () => {
@@ -203,6 +230,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                     expect.objectContaining({ id: 'batch-item-3', name: 'Item 3', age: 30 }),
                 ])
             );
+            expect(sumConsumedCapacity()).toEqual(1.5);
         });
 
         it('should handle empty keys array', async () => {
@@ -210,6 +238,7 @@ describe('DynamoDbRepository Integration Tests', () => {
 
             expect(results).toBeDefined();
             expect(results?.length).toBe(0);
+            expect(sumConsumedCapacity()).toEqual(0);
         });
 
         it('should handle non-existent keys gracefully', async () => {
@@ -230,6 +259,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             const ids = existingItems?.map(item => item.id);
             expect(ids).toContain('batch-item-1');
             expect(ids).toContain('batch-item-2');
+            expect(sumConsumedCapacity()).toEqual(1);
         });
 
         it('should handle batch size over 100 items (pagination)', async () => {
@@ -254,6 +284,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             expect(item50).toBeDefined();
             expect(item100).toBeDefined();
             expect(item150).toBeDefined();
+            expect(sumConsumedCapacity()).toEqual(300);
         }, 60000);
 
         it('should retrieve composite key items', async () => {
@@ -288,6 +319,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                     expect.objectContaining({ userId: 'user-batch-2', itemId: 'item-c', price: 300 }),
                 ])
             );
+            expect(sumConsumedCapacity()).toEqual(6);
         });
 
         it('should maintain order independence', async () => {
@@ -307,6 +339,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             expect(ids).toContain('batch-item-1');
             expect(ids).toContain('batch-item-3');
             expect(ids).toContain('batch-item-5');
+            expect(sumConsumedCapacity()).toEqual(1.5);
         });
 
         it('should handle duplicate keys in input', async () => {
@@ -328,6 +361,7 @@ describe('DynamoDbRepository Integration Tests', () => {
 
             expect(item1Count).toBeGreaterThanOrEqual(1);
             expect(item2Count).toBeGreaterThanOrEqual(1);
+            expect(sumConsumedCapacity()).toEqual(1);
         });
 
         it('should retrieve all attributes for batched items', async () => {
@@ -347,6 +381,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 age: 30,
                 status: 'active'
             });
+            expect(sumConsumedCapacity()).toEqual(2);
         });
     });
 
@@ -365,7 +400,9 @@ describe('DynamoDbRepository Integration Tests', () => {
                     name: 'Test Item',
                     age: 25
                 });
+                expect(sumConsumedCapacity()).toEqual(2);
             });
+
         });
 
         describe('with composite key (partition + sort)', () => {
@@ -383,6 +420,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                     { userId, itemId: 'item-3' },
                     { userId, itemId: 'item-3', name: 'Item Three', category: 'electronics', price: 150 }
                 );
+                consumedCapacityRegister.splice(0, consumedCapacityRegister.length);
             });
 
             it('should retrieve all items for a partition key', async () => {
@@ -402,7 +440,9 @@ describe('DynamoDbRepository Integration Tests', () => {
                 expect(results).toBeDefined();
                 expect(results?.length).toBe(2);
                 expect(results?.every(item => item.category === 'electronics')).toBe(true);
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
+
         });
 
         describe('with GSI (Global Secondary Index)', () => {
@@ -468,6 +508,7 @@ describe('DynamoDbRepository Integration Tests', () => {
 
                 // Wait for GSI to be consistent
                 await new Promise(resolve => setTimeout(resolve, 2000));
+                consumedCapacityRegister.splice(0, consumedCapacityRegister.length);
             });
 
             it('should query items using GSI and fetch full items via batchGetItems', async () => {
@@ -478,6 +519,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 expect(results).toBeDefined();
                 // When using index, it queries GSI then uses batchGetItems to fetch full items
                 expect(Array.isArray(results)).toBe(true);
+                expect(sumConsumedCapacity()).toEqual(2);
             });
 
             it('should query all items with specific status using GSI', async () => {
@@ -493,6 +535,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                         expect(item.status).toBe('active');
                     });
                 }
+                expect(sumConsumedCapacity()).toEqual(2);
             });
 
             it('should combine GSI query with filter expressions', async () => {
@@ -511,6 +554,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                         expect(item.category).toBe('electronics');
                     });
                 }
+                expect(sumConsumedCapacity()).toEqual(1.5);
             });
 
             it('should return full item attributes when querying via GSI', async () => {
@@ -532,6 +576,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                         expect(item).toHaveProperty('createdAt');
                     }
                 }
+                expect(sumConsumedCapacity()).toEqual(2);
             });
 
             it('should handle GSI query with multiple items', async () => {
@@ -550,6 +595,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                         expect(item).toHaveProperty('itemId');
                     });
                 }
+                expect(sumConsumedCapacity()).toEqual(2);
             });
 
             it('should respect projection when querying GSI', async () => {
@@ -568,6 +614,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                         expect(item).toHaveProperty('status');
                     });
                 }
+                expect(sumConsumedCapacity()).toEqual(2);
             });
 
             it('should handle empty results from GSI query', async () => {
@@ -578,6 +625,7 @@ describe('DynamoDbRepository Integration Tests', () => {
 
                 expect(results).toBeDefined();
                 expect(results?.length).toBe(0);
+                expect(sumConsumedCapacity()).toEqual(0);
             });
         });
 
@@ -601,6 +649,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 }
 
                 await new Promise(resolve => setTimeout(resolve, 2000));
+                consumedCapacityRegister.splice(0, consumedCapacityRegister.length);
             });
 
             it('should retrieve all items across multiple pages via GSI', async () => {
@@ -611,8 +660,10 @@ describe('DynamoDbRepository Integration Tests', () => {
 
                 expect(results).toBeDefined();
                 expect(Array.isArray(results)).toBe(true);
+                expect(sumConsumedCapacity()).toEqual(62);
                 // Should handle pagination internally via batchGetItems
             }, 60000);
+
         });
     });
 
@@ -649,6 +700,7 @@ describe('DynamoDbRepository Integration Tests', () => {
             for (const data of testData) {
                 await repository.putItem({ id: data.id }, data);
             }
+            consumedCapacityRegister.splice(0, consumedCapacityRegister.length);
         });
 
         describe('EQUALS operator', () => {
@@ -662,6 +714,7 @@ describe('DynamoDbRepository Integration Tests', () => {
 
                 expect(results).toBeDefined();
                 expect(results?.every(item => item.status === 'active')).toBe(true);
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should filter items with exact number match', async () => {
@@ -676,6 +729,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results[0].age).toBe(30);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should return empty array when no match found', async () => {
@@ -688,6 +742,7 @@ describe('DynamoDbRepository Integration Tests', () => {
 
                 expect(results).toBeDefined();
                 expect(results?.length).toBe(0);
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
         });
 
@@ -704,6 +759,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results[0].status).not.toBe('inactive');
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should filter items not matching number value', async () => {
@@ -718,6 +774,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results[0].age).not.toBe(25);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
         });
 
@@ -734,6 +791,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => item.age && item.age > 30)).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should filter items greater than decimal value', async () => {
@@ -748,6 +806,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => item.score && item.score > 89.0)).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should return empty array when no items are greater', async () => {
@@ -760,6 +819,7 @@ describe('DynamoDbRepository Integration Tests', () => {
 
                 expect(results).toBeDefined();
                 expect(results?.length).toBe(0);
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
         });
 
@@ -776,6 +836,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => item.age && item.age >= 30)).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should include items with exact value', async () => {
@@ -791,6 +852,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                     const exactMatch = results.find(item => item.score === 95.5);
                     expect(exactMatch).toBeDefined();
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
         });
 
@@ -807,6 +869,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => item.age && item.age < 30)).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should filter items less than decimal value', async () => {
@@ -821,6 +884,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => item.score && item.score < 75.0)).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
         });
 
@@ -837,6 +901,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => item.age && item.age <= 25)).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should include items with exact value', async () => {
@@ -852,6 +917,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                     const exactMatch = results.find(item => item.score === 90.0);
                     expect(exactMatch).toBeDefined();
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
         });
 
@@ -868,6 +934,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => ['active', 'pending'].includes(item.status || ''))).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should filter items with value in array of numbers', async () => {
@@ -882,6 +949,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => [25, 30, 35].includes(item.age || 0))).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should return empty array when value not in list', async () => {
@@ -894,6 +962,7 @@ describe('DynamoDbRepository Integration Tests', () => {
 
                 expect(results).toBeDefined();
                 expect(results?.length).toBe(0);
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should handle single value in array', async () => {
@@ -908,6 +977,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results[0].status).toBe('active');
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
         });
 
@@ -924,6 +994,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => item.age && item.age >= 25 && item.age <= 35)).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should filter items with decimal value between range', async () => {
@@ -938,6 +1009,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => item.score && item.score >= 80.0 && item.score <= 90.0)).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should include boundary values', async () => {
@@ -954,6 +1026,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                     const hasUpperBound = results.some(item => item.age === 40);
                     expect(hasLowerBound || hasUpperBound).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should return empty array when no values in range', async () => {
@@ -966,6 +1039,7 @@ describe('DynamoDbRepository Integration Tests', () => {
 
                 expect(results).toBeDefined();
                 expect(results?.length).toBe(0);
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should filter items with string value between range (lexicographical)', async () => {
@@ -982,6 +1056,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                         item.name && item.name >= 'Alice' && item.name <= 'Diana'
                     )).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
         });
 
@@ -1021,6 +1096,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                         item.score && item.score >= 80.0
                     )).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
         });
 
@@ -1037,6 +1113,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                 if (results && results.length > 0) {
                     expect(results.every(item => item.status !== 'inactive')).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should negate IN operator', async () => {
@@ -1053,6 +1130,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                         !['active', 'pending'].includes(item.status || '')
                     )).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
 
             it('should negate BETWEEN operator', async () => {
@@ -1069,6 +1147,7 @@ describe('DynamoDbRepository Integration Tests', () => {
                         item.age && (item.age < 25 || item.age > 35)
                     )).toBe(true);
                 }
+                expect(sumConsumedCapacity()).toEqual(0.5);
             });
         });