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
|
"""
mbed SDK
Copyright (c) 2011-2018 ARM Limited
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.
Author: Przemyslaw Wirkus <Przemyslaw.Wirkus@arm.com>
"""
import os
from shutil import copy
from .host_test_plugins import HostTestPluginBase
class HostTestPluginCopyMethod_MPS2(HostTestPluginBase):
# MPS2 specific flashing / binary setup funcitons
name = 'HostTestPluginCopyMethod_MPS2'
type = 'CopyMethod'
stable = True
capabilities = ['mps2']
required_parameters = ['image_path', 'destination_disk']
def __init__(self):
""" ctor
"""
HostTestPluginBase.__init__(self)
def mps2_copy(self, image_path, destination_disk):
"""! mps2 copy method for "mbed enabled" devices.
This copies the file on the MPS2 keeping the same extension but
renaming it "mbed.extension"
@param image_path Path to file to be copied
@param destination_disk Path to destination (mbed mount point)
@details this uses shutil copy to copy the file.
@return Returns True if copy (flashing) was successful
"""
result = True
# Keep the same extension in the test spec and on the MPS2
_, extension = os.path.splitext(image_path);
destination_path = os.path.join(destination_disk, "mbed" + extension)
try:
copy(image_path, destination_path)
# sync command on mac ignores command line arguments.
if os.name == 'posix':
result = self.run_command('sync -f %s' % destination_path, shell=True)
except Exception as e:
self.print_plugin_error("shutil.copy('%s', '%s')"% (image_path, destination_path))
self.print_plugin_error("Error: %s"% str(e))
result = False
return result
def setup(self, *args, **kwargs):
""" Configure plugin, this function should be called before plugin execute() method is used.
"""
return True
def execute(self, capability, *args, **kwargs):
"""! Executes capability by name.
@details Each capability may directly just call some command line program or execute building pythonic function
@return Returns True if 'capability' operation was successful
"""
if not kwargs['image_path']:
self.print_plugin_error("Error: image path not specified")
return False
if not kwargs['destination_disk']:
self.print_plugin_error("Error: destination disk not specified")
return False
# This optional parameter can be used if TargetID is provided (-t switch)
target_id = kwargs.get('target_id', None)
pooling_timeout = kwargs.get('polling_timeout', 60)
result = False
if self.check_parameters(capability, *args, **kwargs):
# Capability 'default' is a dummy capability
if kwargs['image_path'] and kwargs['destination_disk']:
if capability == 'mps2':
image_path = os.path.normpath(kwargs['image_path'])
destination_disk = os.path.normpath(kwargs['destination_disk'])
# Wait for mount point to be ready
# if mount point changed according to target_id use new mount point
# available in result (_, destination_disk) of check_mount_point_ready
result, destination_disk = self.check_mount_point_ready(destination_disk, target_id=target_id, timeout=pooling_timeout) # Blocking
if result:
result = self.mps2_copy(image_path, destination_disk)
return result
def load_plugin():
""" Returns plugin available in this module
"""
return HostTestPluginCopyMethod_MPS2()
|
