From 63fdfb89cd9e3040ad48de8278ec8378e4ffe8a7 Mon Sep 17 00:00:00 2001 From: Amnon Heiman Date: Sun, 18 Aug 2024 12:19:43 +0300 Subject: [PATCH] Add tests for Alternator batch operation metrics This patch adds unit tests to verify the correctness of the newly introduced histogram metrics for get and write batch operation latencies. The test uses the existing latency test with the added metrics. Signed-off-by: Amnon Heiman --- test/alternator/test_metrics.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/test/alternator/test_metrics.py b/test/alternator/test_metrics.py index 39276fe455..bf1952e8ed 100644 --- a/test/alternator/test_metrics.py +++ b/test/alternator/test_metrics.py @@ -243,12 +243,16 @@ def check_sets_latency(metrics, operation_names): # Test latency metrics for PutItem, GetItem, DeleteItem, UpdateItem. # We can't check what exactly the latency is - just that it gets updated. def test_item_latency(test_table_s, metrics): - with check_sets_latency(metrics, ['DeleteItem', 'GetItem', 'PutItem', 'UpdateItem']): + with check_sets_latency(metrics, ['DeleteItem', 'GetItem', 'PutItem', 'UpdateItem', 'BatchWriteItem', 'BatchGetItem']): p = random_string() test_table_s.put_item(Item={'p': p}) test_table_s.get_item(Key={'p': p}) test_table_s.delete_item(Key={'p': p}) test_table_s.update_item(Key={'p': p}) + test_table_s.meta.client.batch_write_item(RequestItems = { + test_table_s.name: [{'PutRequest': {'Item': {'p': random_string(), 'a': 'hi'}}}]}) + test_table_s.meta.client.batch_get_item(RequestItems = { + test_table_s.name: {'Keys': [{'p': random_string()}], 'ConsistentRead': True}}) # Test latency metrics for GetRecords. Other Streams-related operations - # ListStreams, DescribeStream, and GetShardIterator, have an operation