From de319f6e16254e11cfb3ec66e4489cab70ba59e2 Mon Sep 17 00:00:00 2001
From: Rayhan Hanaputra <hanaputrarayhan@gmail.com>
Date: Sat, 25 Oct 2025 23:26:35 +0700
Subject: [PATCH] test fix sla rui n itoid

---
 receiver/challenges/Carbeat.py | 11 ++++++++---
 receiver/challenges/Sheesh.py  | 17 +++++++++--------
 2 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/receiver/challenges/Carbeat.py b/receiver/challenges/Carbeat.py
index eb5c9cd..7a2c811 100644
--- a/receiver/challenges/Carbeat.py
+++ b/receiver/challenges/Carbeat.py
@@ -88,8 +88,9 @@ class Carbeat(Challenge):
             return False
 
                 
-    def register(self, proc, name, level, timeout=1.):
-        self._read_until(proc, "> ", timeout=timeout)
+    def register(self, proc, name, level, timeout=1., skip_initial_prompt=False):
+        if not skip_initial_prompt:
+            self._read_until(proc, "> ", timeout=timeout)
         self._send_line(proc, "1")
         self._read_until(proc, ": ", timeout=timeout)
         self._send_line(proc, name)
@@ -135,7 +136,11 @@ class Carbeat(Challenge):
             proc = self._spawn()
             self.logger.info("[1] process spawned ✅")
             
-            self.register(proc, "karbit", 1337)
+            # Wait for initial menu to appear with longer timeout
+            self._read_until(proc, "> ", timeout=10.0)
+            self.logger.info("[1.5] initial menu appeared ✅")
+            
+            self.register(proc, "karbit", 1337, skip_initial_prompt=True)
             self.logger.info("[2] user registered ✅")
             
             name = self.list_user(proc)
diff --git a/receiver/challenges/Sheesh.py b/receiver/challenges/Sheesh.py
index 6486705..107e2dd 100644
--- a/receiver/challenges/Sheesh.py
+++ b/receiver/challenges/Sheesh.py
@@ -174,14 +174,15 @@ class Sheesh(Challenge):
                 raise AssertionError("Program did not exit after option 4")
             self.logger.info("[ok] service exit on 4")
 
-            proc_alarm = self._spawn()
-            self._read_until(proc_alarm, "zzz", timeout=190.0)  # 180s + slack
-            try:
-                proc_alarm.wait(timeout=5.0)
-            except subprocess.TimeoutExpired:
-                proc_alarm.kill()
-                raise AssertionError("Alarm fired but process did not exit")
-            self.logger.info("[ok] alarm fired ('zzz') and process self-terminated")
+            # Skip alarm test during SLA checks (takes 180+ seconds)
+            # proc_alarm = self._spawn()
+            # self._read_until(proc_alarm, "zzz", timeout=190.0)  # 180s + slack
+            # try:
+            #     proc_alarm.wait(timeout=5.0)
+            # except subprocess.TimeoutExpired:
+            #     proc_alarm.kill()
+            #     raise AssertionError("Alarm fired but process did not exit")
+            # self.logger.info("[ok] alarm fired ('zzz') and process self-terminated")
 
             self.logger.info('Check passed for sheesh')
             return True