From 6b8ca302d2a657ed45b8abc45cb91094179ec7e0 Mon Sep 17 00:00:00 2001
From: Sema Checherinda <Sema.Checherinda@clickhouse.com>
Date: Tue, 28 May 2024 17:58:32 +0200
Subject: [PATCH] test for TotalQpsLimitExceeded

---
 contrib/aws                                      |  2 +-
 tests/integration/helpers/s3_mocks/broken_s3.py  | 16 ++++++++++++++++
 .../test_checking_s3_blobs_paranoid/test.py      |  1 +
 3 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/contrib/aws b/contrib/aws
index eb96e740453..deeaa9e7c5f 160000
--- a/contrib/aws
+++ b/contrib/aws
@@ -1 +1 @@
-Subproject commit eb96e740453ae27afa1f367ba19f99bdcb38484d
+Subproject commit deeaa9e7c5fe690e3dacc4005d7ecfa7a66a32bb
diff --git a/tests/integration/helpers/s3_mocks/broken_s3.py b/tests/integration/helpers/s3_mocks/broken_s3.py
index 7d0127bc1c4..775424d66a7 100644
--- a/tests/integration/helpers/s3_mocks/broken_s3.py
+++ b/tests/integration/helpers/s3_mocks/broken_s3.py
@@ -195,6 +195,18 @@ class _ServerRuntime:
             )
             request_handler.write_error(429, data)
 
+    class TotalQpsLimitExceededAction:
+        def inject_error(self, request_handler):
+            data = (
+                '<?xml version="1.0" encoding="UTF-8"?>'
+                "<Error>"
+                "<Code>TotalQpsLimitExceeded</Code>"
+                "<Message>Please reduce your request rate.</Message>"
+                "<RequestId>txfbd566d03042474888193-00608d7537</RequestId>"
+                "</Error>"
+            )
+            request_handler.write_error(429, data)
+
     class RedirectAction:
         def __init__(self, host="localhost", port=1):
             self.dst_host = _and_then(host, str)
@@ -269,6 +281,10 @@ class _ServerRuntime:
                 self.error_handler = _ServerRuntime.QpsLimitExceededAction(
                     *self.action_args
                 )
+            elif self.action == "total_qps_limit_exceeded":
+                self.error_handler = _ServerRuntime.TotalQpsLimitExceededAction(
+                    *self.action_args
+                )
             else:
                 self.error_handler = _ServerRuntime.Expected500ErrorAction()
 
diff --git a/tests/integration/test_checking_s3_blobs_paranoid/test.py b/tests/integration/test_checking_s3_blobs_paranoid/test.py
index a7fe02b16de..476f7c61b28 100644
--- a/tests/integration/test_checking_s3_blobs_paranoid/test.py
+++ b/tests/integration/test_checking_s3_blobs_paranoid/test.py
@@ -205,6 +205,7 @@ def test_upload_s3_fail_upload_part_when_multi_part_upload(
     [
         ("slow_down", "DB::Exception: Slow Down."),
         ("qps_limit_exceeded", "DB::Exception: Please reduce your request rate."),
+        ("total_qps_limit_exceeded", "DB::Exception: Please reduce your request rate."),
         (
             "connection_refused",
             "Poco::Exception. Code: 1000, e.code() = 111, Connection refused",