def testSudoRunCommandCalled(self):
"""Test that SudoRunCommand is called when log level > NOTICE."""
expected_cmd = ['dd', 'if=foo', 'of=bar', 'bs=4M', 'iflag=fullblock',
'oflag=sync']
usb_imager = flash.USBImager('dummy_device', 'board', 'foo')
run_mock = self.PatchObject(cros_build_lib, 'SudoRunCommand')
self.PatchObject(logging.Logger, 'getEffectiveLevel',
return_value=logging.WARNING)
usb_imager.CopyImageToDevice('foo', 'bar')
# Check that SudoRunCommand() is called correctly.
run_mock.assert_any_call(expected_cmd, debug_level=logging.NOTICE,
print_cmd=False)
def testUsbImagerOperationCalled(self):
"""Test that flash.UsbImagerOperation is called when log level <= NOTICE."""
expected_cmd = ['dd', 'if=foo', 'of=bar', 'bs=4M', 'iflag=fullblock',
'oflag=sync']
usb_imager = flash.USBImager('dummy_device', 'board', 'foo')
run_mock = self.PatchObject(flash.UsbImagerOperation, 'Run')
self.PatchObject(logging.Logger, 'getEffectiveLevel',
return_value=logging.NOTICE)
usb_imager.CopyImageToDevice('foo', 'bar')
# Check that flash.UsbImagerOperation.Run() is called correctly.
run_mock.assert_called_with(cros_build_lib.SudoRunCommand, expected_cmd,
debug_level=logging.NOTICE, update_period=0.5)