1 files changed, 54 insertions, 0 deletions
diff --git a/moon_manager/tests/performance_tests_policies/actions.py b/moon_manager/tests/performance_tests_policies/actions.py
new file mode 100644
index 00000000..0f31d645
--- /dev/null
+++ b/moon_manager/tests/performance_tests_policies/actions.py
@@ -0,0 +1,54 @@
+# Software Name: MOON
+
+# Version: 5.4
+
+# SPDX-FileCopyrightText: Copyright (c) 2018-2020 Orange and its contributors
+# SPDX-License-Identifier: Apache-2.0
+
+# This software is distributed under the 'Apache License 2.0',
+# the text of which is available at 'http://www.apache.org/licenses/LICENSE-2.0.txt'
+# or see the "LICENSE" file for more details.
+
+
+import json
+from locust import TaskSet, task
+from uuid import uuid4
+import utils
+
+
+class ActionRequests(TaskSet):
+    token = ""
+
+    def on_start(self):
+        """ on_start is called when a Locust start before any task is scheduled """
+        utils.login(self)
+
+    def on_stop(self):
+        """ on_stop is called when the TaskSet is stopping """
+        utils.logout(self)
+
+    @task(10)
+    def actions(self):
+        policy_id = utils.create_policy(self.client, self.token)
+        self.client.get("/actions", headers={"X-Api-Key": self.token})
+        perimeter_name = "test_action_" + str(uuid4())
+        req = self.client.post("/policies/{}/actions".format(policy_id),
+                               headers={"X-Api-Key": self.token},
+                               data={
+                                    "name": perimeter_name,
+                                    "description": "locust test",
+                                })
+        content = json.loads(req.content.decode("utf-8"))
+        subject_id = None
+        for subject_id, subject_value in content.get("actions", {}).items():
+            if subject_value.get("name") == perimeter_name:
+                break
+        self.client.delete("/policies/{}/actions/{}".format(policy_id, subject_id),
+                           headers={"X-Api-Key": self.token})
+        utils.delete_policy(self.client, self.token, policy_id)
+
+    @task(1)
+    def status(self):
+        self.client.get("/status", headers={"X-Api-Key": self.token})
+
+