aboutsummaryrefslogtreecommitdiff
path: root/catapult/devil/devil/utils/lazy/weak_constant_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'catapult/devil/devil/utils/lazy/weak_constant_test.py')
-rw-r--r--catapult/devil/devil/utils/lazy/weak_constant_test.py20
1 files changed, 6 insertions, 14 deletions
diff --git a/catapult/devil/devil/utils/lazy/weak_constant_test.py b/catapult/devil/devil/utils/lazy/weak_constant_test.py
index 643351d8..583fd07a 100644
--- a/catapult/devil/devil/utils/lazy/weak_constant_test.py
+++ b/catapult/devil/devil/utils/lazy/weak_constant_test.py
@@ -30,14 +30,11 @@ class DynamicSideEffect(object):
class WeakConstantTest(unittest.TestCase):
-
def testUninitialized(self):
"""Ensure that the first read calls the initializer."""
initializer = mock.Mock(return_value='initializer called')
test_constant = lazy.WeakConstant(initializer)
- self.assertEquals(
- 'initializer called',
- test_constant.read())
+ self.assertEquals('initializer called', test_constant.read())
initializer.assert_called_once_with()
def testInitialized(self):
@@ -46,23 +43,18 @@ class WeakConstantTest(unittest.TestCase):
test_constant = lazy.WeakConstant(initializer)
test_constant._initialized.set()
test_constant._val = 'initializer not called'
- self.assertEquals(
- 'initializer not called',
- test_constant.read())
+ self.assertEquals('initializer not called', test_constant.read())
self.assertFalse(initializer.mock_calls) # assert not called
def testFirstCallHangs(self):
"""Ensure that reading works even if the first initializer call hangs."""
- dyn = DynamicSideEffect([
- lambda: time.sleep(10),
- lambda: 'second try worked!'
- ])
+ dyn = DynamicSideEffect(
+ [lambda: time.sleep(10), lambda: 'second try worked!'])
initializer = mock.Mock(side_effect=dyn)
test_constant = lazy.WeakConstant(initializer)
- self.assertEquals(
- 'second try worked!',
- timeout_retry.Run(test_constant.read, 1, 1))
+ self.assertEquals('second try worked!',
+ timeout_retry.Run(test_constant.read, 1, 1))
initializer.assert_has_calls([mock.call(), mock.call()])