summaryrefslogtreecommitdiff
path: root/src/server-tests/test_util__pydev.py
blob: 87e9f7b4f9c2dd7cbc0d0921a265a3134263d079 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
"""
PYDev test utilities
"""
import sys
import logging

log = logging.getLogger('rhizi')

def py_dev():
    try:  # enable pydev remote debugging
        import pydevd
        pydevd.settrace(trace_only_current_thread=False)
    except ImportError as e:
        pass


def debug__pydev_pd_arg(f_main):
    """
    PyDev Eclipse remote debug decorator:
       - follow PyDev's remote debugging instructions: http://pydev.org/manual_adv_remote_debugger.html
       - decorate main test function with @debug__pydev_pd_arg
       - pass the -pd command line argument
    """

    def main_decorated(*args, **kwargs):
        if len(sys.argv) > 1 and sys.argv[1] == '-pd':
            try:  # enable pydev remote debugging
                import pydevd
                pydevd.settrace(trace_only_current_thread=False)
            except ImportError as e:
                log.exception('failed to import pydevd')

            sys.argv.pop() # hide extra arg from main()
        f_main(*args, **kwargs)

    return main_decorated