From ef6af0b0adc95d7bf41a0351ba9b31917002aa89 Mon Sep 17 00:00:00 2001
From: Ankush Menat <ankush@frappe.io>
Date: Sun, 17 Mar 2024 16:31:34 +0530
Subject: [PATCH] test: setup signal and faulthandler to debug stuck tests
 (#25494)

---
 frappe/parallel_test_runner.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/frappe/parallel_test_runner.py b/frappe/parallel_test_runner.py
index b0d3244e18..49278028f4 100644
--- a/frappe/parallel_test_runner.py
+++ b/frappe/parallel_test_runner.py
@@ -1,6 +1,8 @@
+import faulthandler
 import json
 import os
 import re
+import signal
 import sys
 import time
 import unittest
@@ -108,6 +110,11 @@ class ParallelTestRunner:
 		return frappe.get_module(module_name)
 
 	def print_result(self):
+		# XXX: Added to debug tests getting stuck AFTER completion
+		# the process should terminate before this, we don't need to reset the signal.
+		signal.alarm(60)
+		faulthandler.register(signal.SIGALRM)
+
 		self.test_result.printErrors()
 		click.echo(self.test_result)
 		if self.test_result.failures or self.test_result.errors: