Use env vars to determine delay

Signed-off-by: Laura Couto <laurarccouto@gmail.com>

performance-test/src/performance_test/hooks.py

@@ -1,3 +1,4 @@
+import os
 from time import sleep
 from typing import Dict
 
@@ -6,7 +7,7 @@
 from pyspark import SparkConf
 from pyspark.sql import SparkSession
 
-HOOK_DELAY = 0
+HOOK_DELAY = int(os.getenv("HOOKS_DELAY", 0))
 
 class SparkHooks:
     @hook_impl

performance-test/src/performance_test/pipelines/expense_analysis/nodes.py

@@ -1,10 +1,11 @@
+import os
 from time import sleep
 
 import pyspark.sql.functions as F
 from pyspark.sql.window import Window
 
-DATASET_LOAD_DELAY = 0
-FILE_SAVE_DELAY = 0
+DATASET_LOAD_DELAY = int(os.getenv("DATASET_LOAD_DELAY", 0))
+FILE_SAVE_DELAY = int(os.getenv("FILE_SAVE_DELAY", 0))
 
 def analyze_expenses_per_party(congress_expenses):
     """Calculate total expense per party."""