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/cros_generate_os_release_unittest.py b/scripts/cros_generate_os_release_unittest.py
index 8fb8cfe..c46c40c 100644
--- a/scripts/cros_generate_os_release_unittest.py
+++ b/scripts/cros_generate_os_release_unittest.py
@@ -13,56 +13,56 @@
class CrosGenerateOsReleaseTest(cros_test_lib.TempDirTestCase):
- """Tests GenerateOsRelease."""
+ """Tests GenerateOsRelease."""
- def setUp(self):
- # Use a fresh tempdir as the root for each test case.
- self.osrelease = os.path.join(self.tempdir, 'etc', 'os-release')
- self.osreleased = os.path.join(self.tempdir, 'etc', 'os-release.d')
- osutils.SafeMakedirs(self.osreleased)
+ def setUp(self):
+ # Use a fresh tempdir as the root for each test case.
+ self.osrelease = os.path.join(self.tempdir, "etc", "os-release")
+ self.osreleased = os.path.join(self.tempdir, "etc", "os-release.d")
+ osutils.SafeMakedirs(self.osreleased)
- def testOnlyOsRelease(self):
- """Tests the script without /etc/os-release."""
- osutils.WriteFile(os.path.join(self.osreleased, 'TEST'), 'hello')
- cros_generate_os_release.GenerateOsRelease(self.tempdir)
- self.assertEqual('TEST=hello\n', osutils.ReadFile(self.osrelease))
+ def testOnlyOsRelease(self):
+ """Tests the script without /etc/os-release."""
+ osutils.WriteFile(os.path.join(self.osreleased, "TEST"), "hello")
+ cros_generate_os_release.GenerateOsRelease(self.tempdir)
+ self.assertEqual("TEST=hello\n", osutils.ReadFile(self.osrelease))
- def testOnlyOsReleaseD(self):
- """Tests the script without /etc/os-release.d."""
- osutils.RmDir(self.osreleased)
- osutils.WriteFile(self.osrelease, 'TEST=bonjour\n')
+ def testOnlyOsReleaseD(self):
+ """Tests the script without /etc/os-release.d."""
+ osutils.RmDir(self.osreleased)
+ osutils.WriteFile(self.osrelease, "TEST=bonjour\n")
- cros_generate_os_release.GenerateOsRelease(self.tempdir)
- self.assertEqual('TEST=bonjour\n', osutils.ReadFile(self.osrelease))
+ cros_generate_os_release.GenerateOsRelease(self.tempdir)
+ self.assertEqual("TEST=bonjour\n", osutils.ReadFile(self.osrelease))
- def testFailOnDuplicate(self):
- """Tests with a field set both in os-release and os-release.d/."""
- osutils.WriteFile(os.path.join(self.osreleased, 'TEST'), 'hello')
- osutils.WriteFile(self.osrelease, 'TEST=bonjour')
+ def testFailOnDuplicate(self):
+ """Tests with a field set both in os-release and os-release.d/."""
+ osutils.WriteFile(os.path.join(self.osreleased, "TEST"), "hello")
+ osutils.WriteFile(self.osrelease, "TEST=bonjour")
- self.assertRaises(cros_build_lib.DieSystemExit,
- cros_generate_os_release.GenerateOsRelease, self.tempdir)
+ self.assertRaises(
+ cros_build_lib.DieSystemExit,
+ cros_generate_os_release.GenerateOsRelease,
+ self.tempdir,
+ )
- def testNormal(self):
- """Normal scenario: both os-release and os-release.d are present."""
- osutils.WriteFile(os.path.join(self.osreleased, 'TEST1'), 'hello')
- osutils.WriteFile(self.osrelease, 'TEST2=bonjour')
+ def testNormal(self):
+ """Normal scenario: both os-release and os-release.d are present."""
+ osutils.WriteFile(os.path.join(self.osreleased, "TEST1"), "hello")
+ osutils.WriteFile(self.osrelease, "TEST2=bonjour")
- default_params = {'TEST1': 'hello2',
- 'TEST3': 'hola'}
+ default_params = {"TEST1": "hello2", "TEST3": "hola"}
- cros_generate_os_release.GenerateOsRelease(self.tempdir,
- default_params=default_params)
- output = osutils.ReadFile(self.osrelease).splitlines()
- output.sort()
- self.assertEqual(['TEST1=hello',
- 'TEST2=bonjour',
- 'TEST3=hola'],
- output)
+ cros_generate_os_release.GenerateOsRelease(
+ self.tempdir, default_params=default_params
+ )
+ output = osutils.ReadFile(self.osrelease).splitlines()
+ output.sort()
+ self.assertEqual(["TEST1=hello", "TEST2=bonjour", "TEST3=hola"], output)
- def testDefaultsOnly(self):
- """Make sure we always emit defaults even without any os-release.d frags."""
- osutils.RmDir(self.osreleased)
- cros_generate_os_release.GenerateOsRelease(self.tempdir, {'FOO': 'bar'})
- self.assertExists(self.osrelease)
- self.assertEqual('FOO=bar\n', osutils.ReadFile(self.osrelease))
+ def testDefaultsOnly(self):
+ """Make sure we always emit defaults even without any os-release.d frags."""
+ osutils.RmDir(self.osreleased)
+ cros_generate_os_release.GenerateOsRelease(self.tempdir, {"FOO": "bar"})
+ self.assertExists(self.osrelease)
+ self.assertEqual("FOO=bar\n", osutils.ReadFile(self.osrelease))