add timeout on procLog socket for CPU usage test

old-commit-hash: 9a78378b81

selfdrive/test/test_cpu_usage.py

@@ -62,24 +62,30 @@ def print_cpu_usage(first_proc, last_proc):
   print(result)
   return r
 
+def all_running():
+  running = manager.get_running()
+  return all(p in running and running[p].is_alive() for p in manager.car_started_processes)
+
 return_code = 1
 def test_thread():
+  try:
     global return_code
-  proc_sock = messaging.sub_sock('procLog', conflate=True)
+    proc_sock = messaging.sub_sock('procLog', conflate=True, timeout=1000)
 
     # wait until everything's started and get first sample
     time.sleep(30)
     first_proc = messaging.recv_sock(proc_sock, wait=True)
+    if first_proc is None or not all_running():
+      print("\n\nTEST FAILED: all car started processes not running\n\n")
+      raise Exception
 
     # run for a minute and get last sample
     time.sleep(60)
     last_proc = messaging.recv_sock(proc_sock, wait=True)
-
-  running = manager.get_running()
-  all_running = all(p in running and running[p].is_alive() for p in manager.car_started_processes)
     return_code = print_cpu_usage(first_proc, last_proc)
-  if not all_running:
+    if not all_running():
       return_code = 1
+  finally:
     _thread.interrupt_main()
 
 if __name__ == "__main__":