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/sysmon/loop_unittest.py b/scripts/sysmon/loop_unittest.py
index 5433373..ba1e56e 100644
--- a/scripts/sysmon/loop_unittest.py
+++ b/scripts/sysmon/loop_unittest.py
@@ -16,46 +16,47 @@
class _MockTime(object):
- """Mock time and sleep.
+ """Mock time and sleep.
- Provides mock behavior for time.time() and time.sleep()
- """
-
- def __init__(self, sleep_delta):
- """Instantiate instance.
-
- Args:
- sleep_delta: Modify sleep time by this many seconds.
- But sleep will always be at least 1.
+ Provides mock behavior for time.time() and time.sleep()
"""
- self.current_time = 0
- self._sleep_delta = sleep_delta
- def time(self):
- return self.current_time
+ def __init__(self, sleep_delta):
+ """Instantiate instance.
- def sleep(self, secs):
- actual_sleep = max(secs + self._sleep_delta, 1)
- self.current_time += actual_sleep
- return actual_sleep
+ Args:
+ sleep_delta: Modify sleep time by this many seconds.
+ But sleep will always be at least 1.
+ """
+ self.current_time = 0
+ self._sleep_delta = sleep_delta
+
+ def time(self):
+ return self.current_time
+
+ def sleep(self, secs):
+ actual_sleep = max(secs + self._sleep_delta, 1)
+ self.current_time += actual_sleep
+ return actual_sleep
@contextlib.contextmanager
def _patch_time(sleep_delta):
- """Mock out time and sleep.
+ """Mock out time and sleep.
- Patches behavior for time.time() and time.sleep()
- """
- mock_time = _MockTime(sleep_delta)
- with mock.patch('time.time', mock_time.time), \
- mock.patch('time.sleep', mock_time.sleep):
- yield mock_time
+ Patches behavior for time.time() and time.sleep()
+ """
+ mock_time = _MockTime(sleep_delta)
+ with mock.patch("time.time", mock_time.time), mock.patch(
+ "time.sleep", mock_time.sleep
+ ):
+ yield mock_time
class TestForceSleep(cros_test_lib.TestCase):
- """Tests for _force_sleep."""
+ """Tests for _force_sleep."""
- def test__force_sleep_at_least_given_secs(self):
- with _patch_time(sleep_delta=-7) as mock_time:
- loop._force_sleep(10)
- self.assertGreaterEqual(mock_time.current_time, 10)
+ def test__force_sleep_at_least_given_secs(self):
+ with _patch_time(sleep_delta=-7) as mock_time:
+ loop._force_sleep(10)
+ self.assertGreaterEqual(mock_time.current_time, 10)