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/faux.py b/api/faux.py
index a24bc76..2743de5 100644
--- a/api/faux.py
+++ b/api/faux.py
@@ -10,106 +10,112 @@
def all_responses(faux_result_factory):
- """A decorator to handle all mock responses.
+ """A decorator to handle all mock responses.
- This is syntactic sugar for handling all of the mock response types in a
- single function.
+ This is syntactic sugar for handling all of the mock response types in a
+ single function.
- Args:
- faux_result_factory: A function with the same signature as a regular
- API endpoint function that populates the output with success or error
- results, as requested, without executing the endpoint.
- """
- assert faux_result_factory
+ Args:
+ faux_result_factory: A function with the same signature as a regular
+ API endpoint function that populates the output with success or error
+ results, as requested, without executing the endpoint.
+ """
+ assert faux_result_factory
- # Get the decorators for each of the mock types so we can compose them.
- success_fn = success(faux_result_factory)
- err_fn = error(faux_result_factory)
+ # Get the decorators for each of the mock types so we can compose them.
+ success_fn = success(faux_result_factory)
+ err_fn = error(faux_result_factory)
- def _decorator(func):
- return err_fn(success_fn(func))
+ def _decorator(func):
+ return err_fn(success_fn(func))
- return _decorator
+ return _decorator
def all_empty(func):
- """Decorator to handle all mock responses with an empty output."""
- return empty_error(empty_success(func))
+ """Decorator to handle all mock responses with an empty output."""
+ return empty_error(empty_success(func))
def success(faux_result_factory):
- """A decorator to handle mock call responses.
+ """A decorator to handle mock call responses.
- Args:
- faux_result_factory: A function with the same signature as a regular
- API endpoint function that populates the output
- """
- assert faux_result_factory
+ Args:
+ faux_result_factory: A function with the same signature as a regular
+ API endpoint function that populates the output
+ """
+ assert faux_result_factory
- def decorator(func):
+ def decorator(func):
+ @functools.wraps(func)
+ def _success(input_proto, output_proto, config, *args, **kwargs):
+ if config.mock_call:
+ faux_result_factory(
+ input_proto, output_proto, config, *args, **kwargs
+ )
+ return controller.RETURN_CODE_SUCCESS
- @functools.wraps(func)
- def _success(input_proto, output_proto, config, *args, **kwargs):
- if config.mock_call:
- faux_result_factory(input_proto, output_proto, config, *args, **kwargs)
- return controller.RETURN_CODE_SUCCESS
+ return func(input_proto, output_proto, config, *args, **kwargs)
- return func(input_proto, output_proto, config, *args, **kwargs)
+ return _success
- return _success
-
- return decorator
+ return decorator
def empty_success(func):
- """A decorator to handle mock success responses with empty outputs."""
- @functools.wraps(func)
- def _empty_success(input_proto, output_proto, config, *args, **kwargs):
- if config.mock_call:
- return controller.RETURN_CODE_SUCCESS
+ """A decorator to handle mock success responses with empty outputs."""
- return func(input_proto, output_proto, config, *args, **kwargs)
+ @functools.wraps(func)
+ def _empty_success(input_proto, output_proto, config, *args, **kwargs):
+ if config.mock_call:
+ return controller.RETURN_CODE_SUCCESS
- return _empty_success
+ return func(input_proto, output_proto, config, *args, **kwargs)
+
+ return _empty_success
def error(faux_error_factory):
- """A decorator to handle mock error responses."""
- assert faux_error_factory
+ """A decorator to handle mock error responses."""
+ assert faux_error_factory
- def decorator(func):
+ def decorator(func):
+ @functools.wraps(func)
+ def _error(input_proto, output_proto, config, *args, **kwargs):
+ if config.mock_error:
+ faux_error_factory(
+ input_proto, output_proto, config, *args, **kwargs
+ )
+ return controller.RETURN_CODE_UNSUCCESSFUL_RESPONSE_AVAILABLE
- @functools.wraps(func)
- def _error(input_proto, output_proto, config, *args, **kwargs):
- if config.mock_error:
- faux_error_factory(input_proto, output_proto, config, *args, **kwargs)
- return controller.RETURN_CODE_UNSUCCESSFUL_RESPONSE_AVAILABLE
+ return func(input_proto, output_proto, config, *args, **kwargs)
- return func(input_proto, output_proto, config, *args, **kwargs)
+ return _error
- return _error
-
- return decorator
+ return decorator
def empty_error(func):
- """A decorator to handle mock error responses with empty outputs."""
- @functools.wraps(func)
- def _empty_error(input_proto, output_proto, config, *args, **kwargs):
- if config.mock_error:
- return controller.RETURN_CODE_UNRECOVERABLE
+ """A decorator to handle mock error responses with empty outputs."""
- return func(input_proto, output_proto, config, *args, **kwargs)
+ @functools.wraps(func)
+ def _empty_error(input_proto, output_proto, config, *args, **kwargs):
+ if config.mock_error:
+ return controller.RETURN_CODE_UNRECOVERABLE
- return _empty_error
+ return func(input_proto, output_proto, config, *args, **kwargs)
+
+ return _empty_error
+
def empty_completed_unsuccessfully_error(func):
- """A decorator to handle mock unsuccessful response with empty outputs."""
- @functools.wraps(func)
- def _empty_error(input_proto, output_proto, config, *args, **kwargs):
- if config.mock_error:
- return controller.RETURN_CODE_COMPLETED_UNSUCCESSFULLY
+ """A decorator to handle mock unsuccessful response with empty outputs."""
- return func(input_proto, output_proto, config, *args, **kwargs)
+ @functools.wraps(func)
+ def _empty_error(input_proto, output_proto, config, *args, **kwargs):
+ if config.mock_error:
+ return controller.RETURN_CODE_COMPLETED_UNSUCCESSFULLY
- return _empty_error
+ return func(input_proto, output_proto, config, *args, **kwargs)
+
+ return _empty_error