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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
# BSD LICENSE
#
# Copyright(c) 2010-2015 Intel Corporation. All rights reserved.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# * Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in
# the documentation and/or other materials provided with the
# distribution.
# * Neither the name of Intel Corporation nor the names of its
# contributors may be used to endorse or promote products derived
# from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
import sys
import os
import signal
import code
import time
import imp
from settings import load_global_setting, DEBUG_SETTING
from utils import get_subclasses, copy_instance_attr
from test_case import TestCase
console = None # global console object
debug_cmd = '' # global debug state
AliveSuite = None # global suite for run command
AliveModule = None # global module for reload
AliveCase = None # global case name for run command
def help_command():
console.push('print \'Help on debug module\'')
console.push('print \'DESCRIPTION\'')
console.push('print \'DTS debug module support few debug commands\'')
console.push('print \' - help(): help messages\'')
console.push('print \' - list(): list all connections\'')
console.push('print \' - connect(): bind to specified connection\'')
console.push('print \' - : connect(\"dut\")\'')
console.push('print \' - quit(): quit debug module\'')
console.push('print \' - exit(): exit processing procedure\'')
console.push('print \' - debug(): call python debug module for further debug\'')
console.push('print \' - rerun(): re-run the interrupted test case\'')
def list_command():
"""
List all connection sessions and can be reference of connect command.
"""
index = 0
from ssh_connection import CONNECTIONS
for connection in CONNECTIONS:
for name, session in connection.items():
console.push('print \'connect %d: %10s\'' % (index, name))
index += 1
def connect_command(connect):
"""
Connect to ssh session and give control to user.
"""
from ssh_connection import CONNECTIONS
for connection in CONNECTIONS:
for name, session in connection.items():
if name == connect:
session.session.interact()
def rerun_command():
"""
Rerun test case specified in command line
"""
global AliveSuite, AliveModule, AliveCase
new_module = imp.reload(AliveModule)
# save arguments required to initialize suite
duts = AliveSuite.__dict__['duts']
tester = AliveSuite.__dict__['tester']
target = AliveSuite.__dict__['target']
suite = AliveSuite.__dict__['suite_name']
for test_classname, test_class in get_subclasses(new_module, TestCase):
suite_obj = test_class(duts, tester, target, suite)
# copy all element from previous suite to reloaded suite
copy_instance_attr(AliveSuite, suite_obj)
# re-run specified test case
for case in suite_obj._get_test_cases(r'%s' % AliveCase):
if callable(case):
case()
def exit_command():
"""
Exit framework.
"""
global debug_cmd
debug_cmd = 'exit'
sys.exit(0)
def debug_command():
"""
Give control to python debugger pdb.
"""
global debug_cmd
debug_cmd = 'debug'
sys.exit(0)
def capture_handle(signum, frame):
"""
Capture keyboard interrupt in the process of send_expect.
"""
global debug_cmd
debug_cmd = 'waiting'
def keyboard_handle(signum, frame):
"""
Interrupt handler for SIGINT and call code module create python interpreter.
"""
global console
console = code.InteractiveConsole()
command = {}
command['list'] = list_command
command['exit'] = exit_command
command['debug'] = debug_command
command['help'] = help_command
command['connect'] = connect_command
command['rerun'] = rerun_command
console.push('print \"Use help command for detail information\"')
try:
code.interact(local=command)
except SystemExit:
# reopen sys.stdin for after exit function stdin will be closed
fd = os.open('/dev/stdin', 600)
sys.stdin = os.fdopen(fd, 'r')
global debug_cmd
if debug_cmd == 'debug':
# call pyton debugger
import pdb
pdb.set_trace()
elif debug_cmd == 'exit':
sys.exit(0)
debug_cmd = ''
def ignore_keyintr():
"""
Temporary disable interrupt handler.
"""
if load_global_setting(DEBUG_SETTING) != 'yes':
return
global debug_cmd
signal.siginterrupt(signal.SIGINT, True)
# if there's waiting request, first handler it
if debug_cmd == 'waiting':
keyboard_handle(signal.SIGINT, None)
return signal.signal(signal.SIGINT, capture_handle)
def aware_keyintr():
"""
Reenable interrupt handler.
"""
if load_global_setting(DEBUG_SETTING) != 'yes':
return
return signal.signal(signal.SIGINT, keyboard_handle)
|
