From e216f06d4d25d37a67239e93a8e2e400552be396 Mon Sep 17 00:00:00 2001
From: Max Wittig <max.wittig@siemens.com>
Date: Wed, 18 Apr 2018 10:34:07 +0200
Subject: chore(tests): add rate limit tests

---
 tools/python_test_v4.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'tools/python_test_v4.py')

diff --git a/tools/python_test_v4.py b/tools/python_test_v4.py
index 695722f..83dd967 100644
--- a/tools/python_test_v4.py
+++ b/tools/python_test_v4.py
@@ -646,3 +646,28 @@ gl.user_activities.list()
 
 # events
 gl.events.list()
+
+# rate limit
+settings = gl.settings.get()
+settings.throttle_authenticated_api_enabled = True
+settings.throttle_authenticated_api_requests_per_period = 1
+settings.throttle_authenticated_api_period_in_seconds = 3
+settings.save()
+projects = list()
+for i in range(0, 20):
+    projects.append(gl.projects.create(
+        {'name': str(i) + "ok"}))
+
+error_message = None
+for i in range(20, 40):
+    try:
+        projects.append(
+            gl.projects.create(
+                {'name': str(i) + 'shouldfail'}, obey_rate_limit=False))
+    except gitlab.GitlabCreateError as e:
+        error_message = e.error_message
+        break
+assert 'Retry later' in error_message
+[current_project.delete() for current_project in projects]
+settings.throttle_authenticated_api_enabled = False
+settings.save()
-- 
cgit v1.2.1