def test_flash_constructs_correct_args(self, logger): jlink = FlasherJLink(logger) executable = "JLink.exe" if sys.platform.startswith( "win") else "JLinkExe" expected_args = [ executable, "-if", "swd", "-speed", "auto", "-device", "test_jlink", "-SelectEmuBySN", "test_target_id", "-commanderscript", "not_tested" ] # pylint: disable=unused-argument def mock_start(args, exe): self.assertEqual(args[0], expected_args[0]) self.assertEqual(args[1], expected_args[1]) self.assertEqual(args[2], expected_args[2]) self.assertEqual(args[3], expected_args[3]) self.assertEqual(args[4], expected_args[4]) self.assertEqual(args[5], expected_args[5]) self.assertEqual(args[6], expected_args[6]) self.assertEqual(args[7], expected_args[7]) self.assertEqual(args[8], expected_args[8]) self.assertEqual(args[9], expected_args[9]) return 0, "" # pylint: disable=protected-access jlink._start_and_wait_flash = mock_start target = { "jlink_device_name": "test_jlink", "target_id": "test_target_id" } returncode = jlink.flash("test_source", target, None, "", False) self.assertEqual(returncode, 0)
def test_get_supported_targets(self, logger): jlink = FlasherJLink(logger) targets = jlink.get_supported_targets() self.assertTrue(isinstance(targets, list)) FlasherJLink.supported_targets = [] self.assertTrue(jlink.get_supported_targets(), [])
def test_flash_returns_non_zero_when_process_fails(self, logger): jlink = FlasherJLink(logger) # pylint: disable=unused-argument def mock_start(args, exe): return 1, "" # pylint: disable=protected-access jlink._start_and_wait_flash = mock_start target = { "jlink_device_name": "test_jlink", "target_id": "test_target_id" } with self.assertRaises(FlashError) as cm: jlink.flash("test_source", target, "", False) self.assertEqual(cm.exception.return_code, EXIT_CODE_FLASH_FAILED)
def test_write_file_contents(self): class MockFile(object): def __init__(self): self._data = b"" def write(self, data): self._data += data def flush(self): pass mock_file = MockFile() # pylint: disable=protected-access FlasherJLink._write_file_contents(mock_file, "binary.bin", False) self.assertEqual(mock_file._data, b"erase\nloadfile binary.bin\nr\ng\nexit\n") mock_file = MockFile() # pylint: disable=protected-access FlasherJLink._write_file_contents(mock_file, "binary.bin", True) self.assertEqual(mock_file._data, b"erase\nloadfile binary.bin\nexit\n")
def test_flash_returns_non_zero_when_target_is_invalid(self, logger): jlink = FlasherJLink(logger) target = {"target_id": "test_target_id"} with self.assertRaises(FlashError) as cm: jlink.flash("test_source", target, "", False) self.assertEqual(cm.exception.return_code, EXIT_CODE_FLASH_FAILED) target = {"jlink_device_name": "test_jlink"} with self.assertRaises(FlashError) as cm: jlink.flash("test_source", target, "", False) self.assertEqual(cm.exception.return_code, EXIT_CODE_FLASH_FAILED)
#!/usr/bin/env python """ Copyright 2016 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. """ from mbed_flasher.flashers.FlasherMbed import FlasherMbed as mbed_flasher from mbed_flasher.flashers.FlasherJLink import FlasherJLink as jlink_flasher # disable Invalid constant name warning, not a const # pylint: disable=C0103 AvailableFlashers = [] # Order matters since JLinkExe flash is preferred for JLink boards if jlink_flasher.is_executable_installed(): AvailableFlashers.append(jlink_flasher) AvailableFlashers.append(mbed_flasher)
def test_is_executable_installed_return_false_when_not_found( self, logger, mock_find): mock_find.return_value = None jlink = FlasherJLink(logger) self.assertEqual(jlink.is_executable_installed(), False)
def test_is_executable_installed_return_true_when_found( self, logger, mock_find): mock_find.return_value = "some/path" jlink = FlasherJLink(logger) self.assertEqual(jlink.is_executable_installed(), True)
def test_can_flash_returns_false(self, logger): jlink = FlasherJLink(logger) self.assertFalse(jlink.can_flash({})) self.assertFalse(jlink.can_flash({"device_type": "daplink"}))
def test_can_flash_when_device_type_matches(self, logger): jlink = FlasherJLink(logger) self.assertTrue(jlink.can_flash({"device_type": FlasherJLink.name}))
def test_get_available_devices(self, logger): jlink = FlasherJLink(logger) targets = jlink.get_available_devices() self.assertTrue(isinstance(targets, list))