Formatting: Format all python code with black.
This CL is probably not what you're looking for, it's only
automated formatting. Ignore it with
`git blame --ignore-rev <revision>` for this commit.
BUG=b:233893248
TEST=CQ
Change-Id: I66591d7a738d241aed3290138c0f68065ab10a6d
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/chromite/+/3879174
Reviewed-by: Mike Frysinger <vapier@chromium.org>
Tested-by: Alex Klein <saklein@chromium.org>
diff --git a/scripts/security_test_image_unittest.py b/scripts/security_test_image_unittest.py
index 94c6e85..79e73aa 100644
--- a/scripts/security_test_image_unittest.py
+++ b/scripts/security_test_image_unittest.py
@@ -13,36 +13,43 @@
class SecurityTestImageTest(cros_test_lib.MockTempDirTestCase):
- """Security test image script tests."""
+ """Security test image script tests."""
- def setUp(self):
- D = cros_test_lib.Directory
- filesystem = (
- D('board', ('recovery_image.bin',)),
- 'other_image.bin',
- )
- cros_test_lib.CreateOnDiskHierarchy(self.tempdir, filesystem)
+ def setUp(self):
+ D = cros_test_lib.Directory
+ filesystem = (
+ D("board", ("recovery_image.bin",)),
+ "other_image.bin",
+ )
+ cros_test_lib.CreateOnDiskHierarchy(self.tempdir, filesystem)
- def testParseArgs(self):
- """Argument parsing tests."""
- # pylint: disable=protected-access
- # Test no arguments.
- with self.assertRaises(SystemExit):
- security_test_image._ParseArgs([])
+ def testParseArgs(self):
+ """Argument parsing tests."""
+ # pylint: disable=protected-access
+ # Test no arguments.
+ with self.assertRaises(SystemExit):
+ security_test_image._ParseArgs([])
- # Test board is set but not used when we have the full image path.
- self.PatchObject(cros_build_lib, 'GetDefaultBoard', return_value='board')
- opts = security_test_image._ParseArgs(
- ['--image', os.path.join(self.tempdir, 'other_image.bin')])
- self.assertEqual('board', opts.board)
- self.assertEqual(opts.image,
- os.path.join(self.tempdir, 'other_image.bin'))
+ # Test board is set but not used when we have the full image path.
+ self.PatchObject(
+ cros_build_lib, "GetDefaultBoard", return_value="board"
+ )
+ opts = security_test_image._ParseArgs(
+ ["--image", os.path.join(self.tempdir, "other_image.bin")]
+ )
+ self.assertEqual("board", opts.board)
+ self.assertEqual(
+ opts.image, os.path.join(self.tempdir, "other_image.bin")
+ )
- # Test the board is fetched and used when using the default image basename.
- self.PatchObject(image_lib, 'GetLatestImageLink',
- return_value=os.path.join(self.tempdir, 'board'))
- opts = security_test_image._ParseArgs([])
- self.assertEqual('board', opts.board)
- self.assertEqual(opts.image,
- os.path.join(self.tempdir,
- 'board/recovery_image.bin'))
+ # Test the board is fetched and used when using the default image basename.
+ self.PatchObject(
+ image_lib,
+ "GetLatestImageLink",
+ return_value=os.path.join(self.tempdir, "board"),
+ )
+ opts = security_test_image._ParseArgs([])
+ self.assertEqual("board", opts.board)
+ self.assertEqual(
+ opts.image, os.path.join(self.tempdir, "board/recovery_image.bin")
+ )