chore: fix wrong stressor init in validator

Signed-off-by: Huamin Chen <[email protected]>

e2e/tools/validator/src/validator/cli/__init__.py

@@ -431,7 +431,9 @@ def stress(cfg: config.Validator, script_path: str, report_dir: str):
 
     click.secho("  * Running stress test ...", fg="green")
     remote = Remote(cfg.remote)
-    stress_test = remote.run_script(script_path)
+    total_runtime_seconds = cfg.stressor.total_runtime_seconds
+    curve_type = cfg.stressor.curve_type
+    stress_test = remote.run_script(script_path = script_path, total_runtime_seconds = total_runtime_seconds, curve_type = curve_type)
     res.start_time = stress_test.start_time
     res.end_time = stress_test.end_time
 

e2e/tools/validator/src/validator/stresser/__init__.py

@@ -27,8 +27,6 @@ def __init__(self, config: config.Remote):
         self.user = config.user
         self.port = config.port
         self.password = config.password
-        self.total_runtime_seconds = config.total_runtime_seconds
-        self.curve_type = config.curve_type
 
         self.ssh_client = paramiko.SSHClient()
         self.ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
@@ -65,13 +63,13 @@ def copy(self, script_path, target_script):
         self.ssh_client.exec_command(f"chmod +x {target_script}")
         logger.info("copying script %s to remote - %s - successful", script_path, target_script)
 
-    def run_script(self, script_path: str) -> ScriptResult:
+    def run_script(self, script_path: str, total_runtime_seconds: int = 1200, curve_type: str = 'default') -> ScriptResult:
         self.connect()
         logger.info("Running script %s ...", script_path)
 
         # ruff: noqa: S108 (Suppressed hard-coded path because we want to intentionally copy stress.sh inside `/tmp` dir)
         target_script = "/tmp/stress.sh"
-        cli_options = f"-t {self.total_runtime_seconds} -c {self.curve_type}"
+        cli_options = f"-t {total_runtime_seconds} -c {curve_type}"
         command = f"{target_script} {cli_options}"
         self.copy(script_path, target_script)
 

e2e/tools/validator/validations.yaml

@@ -2,9 +2,6 @@ config:
   mapping:
     actual: metal
     predicted: vm
-  stressor:
-    total_runtime_seconds: 1200
-    curve_type: default
 
 validations:
   - name: node-rapl - kepler-package

e2e/tools/validator/validator.yaml.sample

@@ -20,4 +20,8 @@ prometheus:
   rate_interval: 20s  # Rate interval for Promql, default is 20s, typically 4 x $scrape_interval
   steps: 3s  # Step duration for Prometheus range queries
 
+stressor:
+    total_runtime_seconds: 1200
+    curve_type: default
+
 validations_file: ./validations.yaml  # Path to the validations file, default is ./validations.yaml