From 5b2ce45d570730fe88f4f7c70d476e08221c8cc7 Mon Sep 17 00:00:00 2001
From: Pierrick HYMBERT <pierrick.hymbert@gmail.com>
Date: Fri, 23 Feb 2024 12:32:30 +0100
Subject: [PATCH] server: test: display server logs in case of failure

---
 examples/server/tests/features/environment.py | 24 ++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/examples/server/tests/features/environment.py b/examples/server/tests/features/environment.py
index de2f9c982..05ede6463 100644
--- a/examples/server/tests/features/environment.py
+++ b/examples/server/tests/features/environment.py
@@ -1,4 +1,3 @@
-import multiprocessing
 import os
 import socket
 import subprocess
@@ -17,6 +16,16 @@ def before_scenario(context, scenario):
 
 
 def after_scenario(context, scenario):
+    if scenario.status == "failed":
+        print(f"\x1b[33;101mSCENARIO FAILED: {scenario.name} server logs:\x1b[0m\n\n")
+        if os.path.isfile('llama.log'):
+            with closing(open('llama.log', 'r')) as f:
+                for line in f:
+                    print(line)
+
+    if not pid_exists(context.server_process.pid):
+        assert False, f"Server not running pid={context.server_process.pid} ..."
+
     print(f"stopping server pid={context.server_process.pid} ...")
     context.server_process.kill()
     # Wait few for socket to free up
@@ -40,3 +49,16 @@ def is_server_listening(server_fqdn, server_port):
     with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as sock:
         result = sock.connect_ex((server_fqdn, server_port))
         return result == 0
+
+
+def pid_exists(pid):
+    """Check whether pid exists in the current process table."""
+    import errno
+    if pid < 0:
+        return False
+    try:
+        os.kill(pid, 0)
+    except OSError as e:
+        return e.errno == errno.EPERM
+    else:
+        return True