added wdcc test

cehbrecht · Apr 18, 2024 · dd50ab6 · dd50ab6
1 parent 64ed532
commit dd50ab6
Showing 1 changed file with 43 additions and 19 deletions.
diff --git a/tests/storm/locustfile.py b/tests/storm/locustfile.py
@@ -1,38 +1,27 @@
-from locust import FastHttpUser, between, task
+from locust import FastHttpUser, between, task, tag
 
 import json
 
+TOPIC = "pids"
+RABBITMQ_URL = "http://localhost:15672"
+
 
 # API endpoint for publishing messages to a queue
 # https://www.freekb.net/Article?id=3108
-PUBLISH_URL = "/api/exchanges/%2f/pids/publish"
+PUBLISH_URL = f"/api/exchanges/%2f/{TOPIC}/publish"
 
 
 class RabbitMQUser(FastHttpUser):
-    host = "http://localhost:15672"
+    host = RABBITMQ_URL
     wait_time = between(1, 5)
 
-    @task
-    def send_cmip6_message(self):
-        # Define JSON message to be sent
-        message = {
-            "aggregation_level": "FILE",
-            "file_name": "temp.nc",
-            "file_size": 18028171468,
-            "file_version": 1,
-            "checksum": "312cb59854334b340d4d046310d028d08e0e71cfaa31ac031b383e1008594d42",
-            "checksum_method": "SHA256",
-            "is_part_of": "hdl:21.14100/37528ff1-2653-3c50-b670-7c00ba96fd6e",
-            "url_original_data": "http://vesg.ipsl.upmc.fr/thredds/fileServer/cmip6/temp.nc",
-            "url_replica": "http://esgf3.dkrz.de/thredds/fileServer/cmip6/temp.nc",
-        }
-
+    def send_message(self, routing_key, message):
         data = json.dumps(message)
 
         # Message payload
         payload = {
             "properties": {},
-            "routing_key": "cmip6.test",
+            "routing_key": routing_key,
             "payload": data,
             "payload_encoding": "string",
         }
@@ -49,3 +38,38 @@ def send_cmip6_message(self):
                 response.failure(
                     f"'routed' had an unexpected value: {response.js['routed']}"
                 )
+
+    @tag("cmip6")
+    @task(10)
+    def send_cmip6_message(self):
+        # Define JSON message to be sent
+        message = {
+            "aggregation_level": "FILE",
+            "file_name": "temp.nc",
+            "file_size": 18028171468,
+            "file_version": 1,
+            "checksum": "312cb59854334b340d4d046310d028d08e0e71cfaa31ac031b383e1008594d42",
+            "checksum_method": "SHA256",
+            "is_part_of": "hdl:21.14100/37528ff1-2653-3c50-b670-7c00ba96fd6e",
+            "url_original_data": "http://vesg.ipsl.upmc.fr/thredds/fileServer/cmip6/temp.nc",
+            "url_replica": "http://esgf3.dkrz.de/thredds/fileServer/cmip6/temp.nc",
+        }
+
+        self.send_message("cmip6.test", message)
+
+    @tag("wdcc")
+    @task
+    def send_wdcc_message(self):
+        # Define JSON message to be sent
+        message = {
+            "handle": "hdl:21.14106/test_abc_1",
+            "url_landing_page": "https://www.wdc-climate.de/ui/entry?acronym=temperature",
+            "is_part_of": "hdl:21.14106/test_temperature",
+            "publisher": "WDCC at DKRZ",
+            "aggregation_level": "dataset",
+            "title": "temperature",
+            "entry_id": "2426195",
+            "please_allow_datasets_without_parents": False,
+        }
+
+        self.send_message("wdcc.test", message)