Adding stress test script

Author: dwang3851

.gitignore

@@ -27,3 +27,4 @@ pip-selfcheck.json
 
 # Backup files
 *.bak
+

clients/job11/autograde-Makefile

@@ -0,0 +1,3 @@
+all:
+	tar xvf autograde.tar
+	(cd random; ./driver.sh)

clients/job11/autograde.tar

@@ -0,0 +1,11 @@
+def maxsubarray(nums):
+    n = len(nums)
+    if n == 1:
+        return nums[0]
+    lookup = [0] * n
+    lookup[0] = nums[0]
+    res = nums[0]
+    for i in range(1, n):
+        lookup[i] = max(lookup[i-1]+nums[i], nums[i])
+        res = max(res, lookup[i])
+    return res

clients/job11/maxsubarray.py

@@ -0,0 +1,8 @@
+home = /opt/homebrew/opt/[email protected]/bin
+implementation = CPython
+version_info = 3.10.13.final.0
+virtualenv = 20.25.0
+include-system-site-packages = false
+base-prefix = /opt/homebrew/opt/[email protected]/Frameworks/Python.framework/Versions/3.10
+base-exec-prefix = /opt/homebrew/opt/[email protected]/Frameworks/Python.framework/Versions/3.10
+base-executable = /opt/homebrew/opt/[email protected]/bin/python3.10

dump.rdb

@@ -0,0 +1,76 @@
+import argparse
+import subprocess
+import time
+import os
+import sys
+from config import Config
+
+def run_stress_test(num_submissions, submission_delay, autograder_image, output_file, tango_port, tango_path, job_name, job_path):
+    with open(output_file, 'a') as f:        
+        f.write(f"Stress testing with {num_submissions} submissions\n")
+        for i in range(1, num_submissions + 1):
+            command = [
+            'python3', os.path.join(tango_path, 'clients/tango-cli.py'),
+            '-P', str(tango_port),
+            '-k', 'test',
+            '-l', job_name,
+            '--runJob', job_path,
+            '--image', autograder_image
+            ]
+            subprocess.run(command, stdout = f, stderr = f)
+            f.write(f"Submission {i} submitted \n")  
+
+            if submission_delay > 0:
+                time.sleep(submission_delay)
+    
+    sys.exit()
+
+def get_metrics(output_file):
+    if Config.LOGFILE == None:
+        print("Make sure logs are recorded in a log file")
+
+    job_times = []
+    with open(Config.LOGFILE, 'r') as f:  
+        for line in f:
+            if "finished after " in line:
+                start = line.find("finished after ") + len("finished after ")
+                seconds = int(line[start:].split()[0])
+                job_times.append(seconds)
+    
+    with open(output_file, 'a') as f:
+        if len(job_times) == 0:
+            print("No jobs have been completed")
+        else:
+            avg = sum(job_times)/len(job_times)
+            f.write(f"Average job time is {avg} seconds \n")
+    
+    sys.exit()
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Stress test script for Tango")
+    parser.add_argument('--num_submissions', type=int, default=10, help="Number of submissions")
+    parser.add_argument('--submission_delay', type=float, default=1.0, help="Delay between submissions")
+    parser.add_argument('--autograder_image', type=str, required=True, help="Autograder image")
+    parser.add_argument('--output_file', type=str, default='stress_test.out', help="Output file")
+    parser.add_argument('--tango_port', type=int, default=4567, help="Tango server port")
+    parser.add_argument('--tango_path', type=str, required=True, help="Path to Tango")
+    parser.add_argument('--job_name', type=str, required=True, help="Name of the job")
+    parser.add_argument('--job_path', type=str, required=True, help="Path to the job")
+    parser.add_argument('--get_metrics', type=bool, default = False, help= "Set to true to get metrics, does not create new jobs")
+
+    args = parser.parse_args()
+
+    if args.get_metrics:
+        get_metrics(args.output_file)
+    else:
+        run_stress_test(
+            args.num_submissions,
+            args.submission_delay,
+            args.autograder_image,
+            args.output_file,
+            args.tango_port,
+            args.tango_path,
+            args.job_name,
+            args.job_path
+        )

pyvenv.cfg

@@ -159,6 +159,7 @@ def run(self):
         """run - Step a job through its execution sequence"""
         try:
             # Hash of return codes for each step
+            start_time = time.time()
             ret = {}
             ret["waitvm"] = None
             ret["copyin"] = None
@@ -319,7 +320,8 @@ def run(self):
             # Job termination. Notice that Tango considers
             # things like runjob timeouts and makefile errors to be
             # normal termination and doesn't reschedule the job.
-            self.log.info("Success: job %s:%d finished" % (self.job.name, self.job.id))
+            elapsed_secs = time.time() - start_time
+            self.log.info("Success: job %s:%d finished after %d seconds" % (self.job.name, self.job.id, elapsed_secs))
 
             # Move the job from the live queue to the dead queue
             # with an explanatory message