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
|
# pyOCD debugger
# Copyright (c) 2020 Arm Limited
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from __future__ import print_function
import argparse
import os
import sys
import traceback
import logging
from pyocd.core.helpers import ConnectHelper
from pyocd.probe.pydapaccess import DAPAccess
from pyocd.core.memory_map import MemoryType
from pyocd.commands.execution_context import CommandExecutionContext
from test_util import (
Test,
TestResult,
get_session_options,
get_target_test_params,
get_test_binary_path,
TEST_DIR,
)
TEST_USER_SCRIPT = os.path.join(TEST_DIR, "test_user_script.py")
class UserScriptTestResult(TestResult):
def __init__(self):
super(UserScriptTestResult, self).__init__(None, None, None)
self.name = "user_script"
class UserScriptTest(Test):
def __init__(self):
super(UserScriptTest, self).__init__("User Script Test", user_script_test)
def run(self, board):
try:
result = self.test_function(board.unique_id)
except Exception as e:
result = UserScriptTestResult()
result.passed = False
print("Exception %s when testing board %s" % (e, board.unique_id))
traceback.print_exc(file=sys.stdout)
result.board = board
result.test = self
return result
def user_script_test(board_id):
with ConnectHelper.session_with_chosen_probe(
unique_id=board_id, user_script=TEST_USER_SCRIPT, **get_session_options()) as session:
board = session.board
target = session.target
test_params = get_target_test_params(session)
session.probe.set_clock(test_params['test_clock'])
memory_map = target.get_memory_map()
boot_region = memory_map.get_boot_memory()
ram_region = memory_map.get_default_region_of_type(MemoryType.RAM)
binary_file = get_test_binary_path(board.test_binary)
test_pass_count = 0
test_count = 0
result = UserScriptTestResult()
# TEST basic functionality
print("\n------ Testing delegates ------")
# TODO verify user script delegates were called
target.reset_and_halt()
target.resume()
target.halt()
target.step()
test_count += 1
test_pass_count += 1
print("TEST PASSED")
# TEST user defined commands
print("\n------ Testing user defined commands ------")
context = CommandExecutionContext()
context.attach_session(session)
def test_command(cmd):
try:
print("\nTEST: %s" % cmd)
context.process_command_line(cmd)
except Exception:
print("TEST FAILED")
traceback.print_exc(file=sys.stdout)
return False
else:
print("TEST PASSED")
return True
# Verify command with float, int, str args.
if test_command("testcmd 3.14 0xbeef foobar"):
test_pass_count += 1
test_count += 1
# Verify varargs: all should be strings in the cmd's args
if test_command("anothertestcmd a b 1 2 fee fie foe"):
test_pass_count += 1
test_count += 1
print("\nTest Summary:")
print("Pass count %i of %i tests" % (test_pass_count, test_count))
if test_pass_count == test_count:
print("USER SCRIPT TEST PASSED")
else:
print("USER SCRIPT TEST FAILED")
target.reset()
result.passed = test_count == test_pass_count
return result
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='pyOCD user script test')
parser.add_argument('-d', '--debug', action="store_true", help='Enable debug logging')
parser.add_argument("-da", "--daparg", dest="daparg", nargs='+', help="Send setting to DAPAccess layer.")
args = parser.parse_args()
level = logging.DEBUG if args.debug else logging.INFO
logging.basicConfig(level=level)
DAPAccess.set_args(args.daparg)
# Set to debug to print some of the decisions made while flashing
session = ConnectHelper.session_with_chosen_probe(**get_session_options())
test = UserScriptTest()
result = [test.run(session.board)]
|
