Added python based script test

Used sys._exit instead of exception raise to exit both scenarios

Updated click echo to always have [simvue]

Author: kzscisoft

simvue/run.py

@@ -189,7 +189,7 @@ def __exit__(
             if _error_msg:
                 _error_msg = f":\n{_error_msg}"
             click.secho(
-                "Simvue process executor terminated with non-zero exit status "
+                "[simvue] Process executor terminated with non-zero exit status "
                 f"{_non_zero}{_error_msg}",
                 fg="red",
                 bold=True,
@@ -294,7 +294,8 @@ def _heartbeat(
                             self._dispatcher.purge()
                             self._dispatcher.join()
                         self.set_status("terminated")
-                        raise RuntimeError("Run was aborted")
+                        click.secho("[simvue] Run was aborted.", fg="red", bold=True)
+                        os._exit(1)
 
                 if self._simvue:
                     self._simvue.send_heartbeat()
@@ -1232,7 +1233,7 @@ def save_file(
 
         if not file_size:
             click.secho(
-                "WARNING: saving zero-sized files not currently supported",
+                "[simvue] WARNING: saving zero-sized files not currently supported",
                 bold=True,
                 fg="yellow",
             )
@@ -1420,7 +1421,7 @@ def close(self) -> bool:
             if _error_msg:
                 _error_msg = f":\n{_error_msg}"
             click.secho(
-                "Simvue process executor terminated with non-zero exit status "
+                "[simvue] Process executor terminated with non-zero exit status "
                 f"{_non_zero}{_error_msg}",
                 fg="red",
                 bold=True,

tests/refactor/test_run_class.py

@@ -461,7 +461,7 @@ def test_save_object(
 
 
 @pytest.mark.run
-def test_abort_on_alert(create_plain_run: typing.Tuple[sv_run.Run, dict]) -> None:
+def test_abort_on_alert_process(create_plain_run: typing.Tuple[sv_run.Run, dict]) -> None:
     run, _ = create_plain_run
     run.config(resources_metrics_interval=1)
     run._heartbeat_interval = 1
@@ -473,6 +473,26 @@ def test_abort_on_alert(create_plain_run: typing.Tuple[sv_run.Run, dict]) -> Non
     if not run._status == "terminated":
         run.kill_all_processes()
         raise AssertionError("Run was not terminated")
+    
+
+@pytest.mark.run
+def test_abort_on_alert_python(create_plain_run: typing.Tuple[sv_run.Run, dict]) -> None:
+    run, _ = create_plain_run
+    run.config(resources_metrics_interval=1)
+    run._heartbeat_interval = 1
+    client = sv_cl.Client()
+    i = 0
+
+    while True:
+        time.sleep(1)
+        if i == 4:
+            client.abort_run(run._id, reason="testing abort")
+        i += 1
+        if i == 10:
+            break
+
+    assert i == 4
+    assert run._status == "terminated"
 
 
 @pytest.mark.run