| """Unit tests for saft_flashrom_util.py.""" |
| |
| import mock |
| import StringIO |
| import unittest |
| |
| from autotest_lib.client.common_lib import autotemp |
| from autotest_lib.client.cros.faft.utils import (os_interface, |
| saft_flashrom_util, |
| shell_wrapper) |
| |
| |
| class TestFlashromUtil(unittest.TestCase): |
| """Tests for saft_flashrom_util.flashrom_util().""" |
| |
| def setUp(self): |
| self._tempdir = autotemp.tempdir(unique_id='saft_flashrom_util_test') |
| self.addCleanup(self._tempdir.clean) |
| |
| self.os_if = os_interface.OSInterface( |
| self._tempdir.name, test_mode=True) |
| self.flashrom_util = saft_flashrom_util.flashrom_util(self.os_if) |
| |
| @mock.patch('subprocess.Popen') |
| def testTargetIsBroken(self, mock_subproc_popen): |
| """check_target should raise ShellError if flashrom is broken""" |
| |
| bad_flashrom = mock.Mock() |
| bad_flashrom.stdout = StringIO.StringIO('broken flashrom stdout') |
| bad_flashrom.stderr = StringIO.StringIO('broken flashrom stderr') |
| bad_flashrom.returncode = 1 |
| |
| mock_subproc_popen.return_value = bad_flashrom |
| with self.assertRaises(shell_wrapper.ShellError): |
| self.flashrom_util.check_target() |
| |
| @mock.patch('subprocess.Popen') |
| def testTargetIsNotBroken(self, mock_subproc_popen): |
| """check_target should return True if flashrom is working""" |
| |
| good_flashrom = mock.Mock() |
| good_flashrom.stdout = StringIO.StringIO('working flashrom stdout') |
| good_flashrom.stderr = StringIO.StringIO('working flashrom stderr') |
| good_flashrom.returncode = 0 |
| |
| mock_subproc_popen.return_value = good_flashrom |
| self.assertTrue(self.flashrom_util.check_target()) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
