def test_openocd_method(self, call_mock):
""" Test openocd method """
call_mock.return_value = 0
args = ['programmer.py', '/dev/null']
with mock.patch('sys.argv', args):
ret = flash()
self.assertEqual(ret, 0)
args += ['-q']
with mock.patch('sys.argv', args):
ret = flash()
self.assertEqual(ret, 0)
args2 = ['programmer.py']
with mock.patch('sys.argv', args2):
ret = reset()
self.assertEqual(ret, 0)
args2 += ['-q']
with mock.patch('sys.argv', args2):
ret = reset()
self.assertEqual(ret, 0)
args3 = ['programmer.py', '/path/to/firmware']
with mock.patch('sys.argv', args3):
ret = flash()
self.assertEqual(ret, 1)
call_mock.return_value = 42
with mock.patch('sys.argv', args2):
ret = reset()
self.assertEqual(ret, 42)
def test_cc2538_reset_method(self, call_mock, flash_mock):
""" Test cc2538 reset method """
call_mock.return_value = 0
flash_mock.return_value = 0
args = ['programmer.py']
with mock.patch('sys.argv', args):
ret = reset()
self.assertEqual(ret, 0)
def test_avrdude_reset(self, wait_mock, bootloader_mock):
""" Test avrdude reset """
bootloader_mock.return_value = 0
wait_mock.return_value = 0
args = ['programmer.py']
with mock.patch('sys.argv', args):
ret = reset()
self.assertEqual(ret, 0)
def test_linux_node_no_method(self):
""" Test Linux node no flash method """
args = ['programmer.py', '/path/to/firmware']
with mock.patch('sys.argv', args):
ret = flash()
self.assertEqual(ret, -1)
args2 = ['programmer.py']
with mock.patch('sys.argv', args2):
ret = reset()
self.assertEqual(ret, -1)
def test_control_node_method(self, call_mock):
""" Test control node method """
call_mock.return_value = 0
args = ['programmer.py', '-cn']
with mock.patch('sys.argv', args):
ret = reset()
self.assertEqual(ret, 0)
args2 = ['programmer.py', '-cn', '/dev/null']
with mock.patch('sys.argv', args2):
ret = flash()
self.assertEqual(ret, 0)
def test_linux_open_node_no_cn(self):
""" Test Linux open node without control node"""
args = ['programmer.py', '-cn']
with mock.patch('sys.argv', args):
with self.assertRaises(SystemExit):
reset()
