Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 8ac1d04c authored by Jizheng Chu's avatar Jizheng Chu
Browse files

Add timeout for teardown_test 

Currently teardown test will stuck at stop stack when ValueError
occured in neighbor test. Add timeout to teardown will allow us to
capture snoop log and further debug neighbor test failures.

Bug: 208712935
Test: cert/run
Tag: #gd-refactor
Change-Id: I0dab000a888e23c2f76b0a342039ecb0ebe64aef
parent 6bf2e507

system/blueberry/tests/gd/cert/gd_base_test.py

+25 −4
Original line number Diff line number Diff line
@@ -39,8 +39,26 @@ from mobly import asserts, signals 
from mobly import base_test





class Timeout:



    def __init__(self, seconds=1, error_message='Timeout'):

        self.seconds = seconds

        self.error_message = error_message



    def handle_timeout(self, signum, frame):

        raise TimeoutError(self.error_message)



    def __enter__(self):

        signal.signal(signal.SIGALRM, self.handle_timeout)

        signal.alarm(self.seconds)



    def __exit__(self, type, value, traceback):

        signal.alarm(0)





class GdBaseTestClass(base_test.BaseTestClass):



    FUNCTION_CALL_TIMEOUT_SECONDS = 5

    SUBPROCESS_WAIT_TIMEOUT_SECONDS = 10



    def setup_class(self, dut_module, cert_module):
@@ -117,12 +135,15 @@ class GdBaseTestClass(base_test.BaseTestClass): 
    def teardown_test(self):

        stack = ""

        try:

            with Timeout(seconds=self.FUNCTION_CALL_TIMEOUT_SECONDS):

                stack = "CERT"

                self.cert.rootservice.StopStack(facade_rootservice.StopStackRequest())

                stack = "DUT"

                self.dut.rootservice.StopStack(facade_rootservice.StopStackRequest())

        except RpcError as rpc_error:

            asserts.fail("Failed to stop {} stack, RpcError={!r}".format(stack, rpc_error))

        except TimeoutError:

            logging.error("Failed to stop {} stack in {} s".format(stack, self.FUNCTION_CALL_TIMEOUT_SECONDS))

        finally:

            # Destroy GD device objects

            self._controller_manager.unregister_controllers()