timeout_hook_plugin.py

from __future__ import absolute_import, division, print_function, unicode_literals

from functools import partial


class Hooks:

    def pytest_about_to_timeout(self, config):
        pass


_already_mocked = False

def pytest_configure(config):
    config.pluginmanager.add_hookspecs(Hooks)
    import pytest_timeout

    global _already_mocked
    if not _already_mocked:
        config._original_dump_stacks = pytest_timeout.dump_stacks
        pytest_timeout.dump_stacks = partial(_mock_dump_stacks, config)


def _mock_dump_stacks(config):
    try:
        config.hook.pytest_about_to_timeout(config=config)
    finally:
        config._original_dump_stacks()