Query test names from elasticsearch

1 files changed, 13 insertions, 25 deletions

diff --git a/elastic/generate-config.py b/elastic/generate-config.py
index 44cc083..abb8d7c 100755
--- a/elastic/generate-config.py
+++ b/elastic/generate-config.py
@@ -1,32 +1,10 @@
 #!/usr/bin/env python3
 
 import pystache
+import requests
+import json
 
-#print(pystache.render('Hi {{person}}!', {'person': 'Mom'}))
-
-tests = [
-    "test_encrypt_raw_10M",
-    "test_encrypt_raw_100k",
-    "test_encrypt_raw_10k",
-    "test_decrypt_raw_100k",
-    "test_decrypt_raw_10M",
-    "test_decrypt_raw_10k",
-    "test_decrypt_raw_1M",
-    "test_decrypt_raw_500k",
-    "test_encrypt_raw_1M",
-    "test_encrypt_raw_500k",
-    "test_initialization",
-    "test_async_create_1000_10k",
-    "test_async_create_100_100k",
-    "test_create_1000_10k",
-    "test_create_100_100k",
-    "test_create_20_500k",
-    "test_decrypt_doc_100k",
-    "test_decrypt_doc_10M",
-    "test_decrypt_doc_10k",
-    "test_decrypt_doc_1M"
-    ]
-
+URL='https://moose.leap.se:9200'
 
 
 # Generate visualizations
@@ -75,5 +53,15 @@ def generate_dashboards():
 
 # Main
 
+# Get all test names
+with open('./query/test_names.json') as f:
+    response = requests.get(URL + '/_search', data=f)
+    buckets = response.json()['aggregations']['test_names']['buckets']
+
+tests = []
+for d in buckets:
+    name = d['key']
+    tests.append(name)
+
 generate_dashboards()
 generate_visualizations()