Fix smoketest (partial)

Signed-off-by: Mihai Criveti <[email protected]>

Author: crivetimihai

smoketest.py

@@ -314,16 +314,18 @@ def step_8_invoke_tool():
     logging.info("✅ Tool invocation returned time")
 
 
-
 def step_9_version_health():
     health = request("GET", "/health").json()["status"].lower()
     assert health in ("ok", "healthy"), f"Unexpected health status: {health}"
     ver = request("GET", "/version").json()["app"]["name"]
     logging.info("✅ Health OK – app %s", ver)
 
 
+def step_10_cleanup_gateway(gid: int | None = None):
+    if gid is None:
+        logging.warning("🧹  No gateway ID; nothing to delete")
+        return
 
-def step_10_cleanup_gateway(gid: int):
     request("DELETE", f"/gateways/{gid}")
     assert all(g["id"] != gid for g in request("GET", "/gateways").json())
     logging.info("✅ Gateway deleted")
@@ -388,8 +390,11 @@ def main():
                 fn(args.restart_time_server)  # type: ignore[arg-type]
             elif name == "register_gateway":
                 gid = fn()  # type: ignore[func-returns-value]
-            elif name == "cleanup_gateway" and gid is not None:
-                fn(gid)  # type: ignore[arg-type]
+            elif name == "cleanup_gateway":
+                if gid is None:
+                    logging.warning("🧹  Skipping gateway‐deletion: no gateway was ever registered")
+                else:
+                    fn(gid)  # type: ignore[arg-type]
             else:
                 fn()
         logging.info("\n✅✅  ALL STEPS PASSED")
@@ -401,6 +406,5 @@ def main():
         cleanup()
 
 
-
 if __name__ == "__main__":
     main()