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/api/controller/api_unittest.py b/api/controller/api_unittest.py
index 95c0eaa..7afd454 100644
--- a/api/controller/api_unittest.py
+++ b/api/controller/api_unittest.py
@@ -14,91 +14,100 @@
from chromite.lib import cros_test_lib
-class CompileProtoTest(cros_test_lib.RunCommandTestCase,
- api_config.ApiConfigMixin):
- """CompileProto tests."""
+class CompileProtoTest(
+ cros_test_lib.RunCommandTestCase, api_config.ApiConfigMixin
+):
+ """CompileProto tests."""
- def setUp(self):
- self.request = api_pb2.CompileProtoRequest()
- self.response = api_pb2.CompileProtoResponse()
+ def setUp(self):
+ self.request = api_pb2.CompileProtoRequest()
+ self.response = api_pb2.CompileProtoResponse()
- def testCompileProto(self):
- """Quick CompileProto functional check."""
- self.rc.SetDefaultCmdResult(stdout=' M foo/bar.py')
- expected = [os.path.join(constants.CHROMITE_DIR, 'foo/bar.py')]
+ def testCompileProto(self):
+ """Quick CompileProto functional check."""
+ self.rc.SetDefaultCmdResult(stdout=" M foo/bar.py")
+ expected = [os.path.join(constants.CHROMITE_DIR, "foo/bar.py")]
- api_controller.CompileProto(self.request, self.response, self.api_config)
- returned = [f.path for f in self.response.modified_files]
+ api_controller.CompileProto(
+ self.request, self.response, self.api_config
+ )
+ returned = [f.path for f in self.response.modified_files]
- self.assertCountEqual(expected, returned)
- self.assertTrue(all(f in returned for f in expected))
- self.assertTrue(all(f in expected for f in returned))
+ self.assertCountEqual(expected, returned)
+ self.assertTrue(all(f in returned for f in expected))
+ self.assertTrue(all(f in expected for f in returned))
- def testValidateOnly(self):
- """Verify validate only calls do not execute logic."""
- api_controller.CompileProto(self.request, self.response,
- self.validate_only_config)
+ def testValidateOnly(self):
+ """Verify validate only calls do not execute logic."""
+ api_controller.CompileProto(
+ self.request, self.response, self.validate_only_config
+ )
- self.assertFalse(self.rc.call_count)
+ self.assertFalse(self.rc.call_count)
- def testMockSuccess(self):
- """Verify mock success calls do not execute logic."""
- api_controller.CompileProto(self.request, self.response,
- self.mock_call_config)
- self.assertTrue(len(self.response.modified_files))
- self.assertFalse(self.rc.call_count)
+ def testMockSuccess(self):
+ """Verify mock success calls do not execute logic."""
+ api_controller.CompileProto(
+ self.request, self.response, self.mock_call_config
+ )
+ self.assertTrue(len(self.response.modified_files))
+ self.assertFalse(self.rc.call_count)
class GetMethodsTest(cros_test_lib.MockTestCase, api_config.ApiConfigMixin):
- """GetMethods tests."""
+ """GetMethods tests."""
- def setUp(self):
- self.request = api_pb2.MethodGetRequest()
- self.response = api_pb2.MethodGetResponse()
+ def setUp(self):
+ self.request = api_pb2.MethodGetRequest()
+ self.response = api_pb2.MethodGetResponse()
- def testGetMethods(self):
- """Simple GetMethods check."""
- methods = ['foo', 'bar']
- self.PatchObject(router.Router, 'ListMethods', return_value=methods)
+ def testGetMethods(self):
+ """Simple GetMethods check."""
+ methods = ["foo", "bar"]
+ self.PatchObject(router.Router, "ListMethods", return_value=methods)
- api_controller.GetMethods(self.request, self.response, self.api_config)
+ api_controller.GetMethods(self.request, self.response, self.api_config)
- self.assertCountEqual(methods, [m.method for m in self.response.methods])
+ self.assertCountEqual(
+ methods, [m.method for m in self.response.methods]
+ )
- def testValidateOnly(self):
- """Check validate_only_config calls only validate."""
- patch = self.PatchObject(router.Router, 'ListMethods')
+ def testValidateOnly(self):
+ """Check validate_only_config calls only validate."""
+ patch = self.PatchObject(router.Router, "ListMethods")
- api_controller.GetMethods(self.request, self.response,
- self.validate_only_config)
+ api_controller.GetMethods(
+ self.request, self.response, self.validate_only_config
+ )
- patch.assert_not_called()
+ patch.assert_not_called()
class GetVersionTest(cros_test_lib.MockTestCase, api_config.ApiConfigMixin):
- """GetVersion tests."""
+ """GetVersion tests."""
- def setUp(self):
- self.PatchObject(api_controller, 'VERSION_MAJOR', new=1)
- self.PatchObject(api_controller, 'VERSION_MINOR', new=2)
- self.PatchObject(api_controller, 'VERSION_BUG', new=3)
+ def setUp(self):
+ self.PatchObject(api_controller, "VERSION_MAJOR", new=1)
+ self.PatchObject(api_controller, "VERSION_MINOR", new=2)
+ self.PatchObject(api_controller, "VERSION_BUG", new=3)
- self.request = api_pb2.VersionGetRequest()
- self.response = api_pb2.VersionGetResponse()
+ self.request = api_pb2.VersionGetRequest()
+ self.response = api_pb2.VersionGetResponse()
- def testGetVersion(self):
- """Simple GetVersion check."""
- api_controller.GetVersion(self.request, self.response, self.api_config)
+ def testGetVersion(self):
+ """Simple GetVersion check."""
+ api_controller.GetVersion(self.request, self.response, self.api_config)
- self.assertEqual(self.response.version.major, 1)
- self.assertEqual(self.response.version.minor, 2)
- self.assertEqual(self.response.version.bug, 3)
+ self.assertEqual(self.response.version.major, 1)
+ self.assertEqual(self.response.version.minor, 2)
+ self.assertEqual(self.response.version.bug, 3)
- def testValidateOnly(self):
- """Check validate_only_config calls only validate."""
- api_controller.GetVersion(self.request, self.response,
- self.validate_only_config)
+ def testValidateOnly(self):
+ """Check validate_only_config calls only validate."""
+ api_controller.GetVersion(
+ self.request, self.response, self.validate_only_config
+ )
- self.assertFalse(self.response.version.major)
- self.assertFalse(self.response.version.minor)
- self.assertFalse(self.response.version.bug)
+ self.assertFalse(self.response.version.major)
+ self.assertFalse(self.response.version.minor)
+ self.assertFalse(self.response.version.bug)