blob: e80a043bd9c57ffa380c13f2179a002a53000c9d [file] [log] [blame] [edit]
#!/usr/bin/env python
# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import mox
import threading
import time
import unittest
import common
from autotest_lib.client.common_lib import decorators
class InContextTest(mox.MoxTestBase):
""" Unit tests for the in_context decorator. """
@decorators.in_context('lock')
def inc_count(self):
""" Do a slow, racy read/write. """
temp = self.count
time.sleep(0.0001)
self.count = temp + 1
def testDecorator(self):
""" Test that the decorator works by using it with a lock. """
self.count = 0
self.lock = threading.RLock()
iters = 100
num_threads = 20
# Note that it is important for us to go through all this bother to call
# a method in_context N times rather than call a method in_context that
# does something N times, because by doing the former, we acquire the
# context N times (1 time for the latter).
thread_body = lambda f, n: [f() for i in xrange(n)]
threads = [threading.Thread(target=thread_body,
args=(self.inc_count, iters))
for i in xrange(num_threads)]
for thread in threads:
thread.start()
for thread in threads:
thread.join()
self.assertEquals(iters * num_threads, self.count)
class CachedPropertyTest(unittest.TestCase):
def testIt(self):
"""cached_property"""
class Example(object):
def __init__(self, v=0):
self.val = v
@decorators.cached_property
def prop(self):
self.val = self.val + 1
return self.val
ex = Example()
self.assertEquals(1, ex.prop)
self.assertEquals(1, ex.prop)
ex2 = Example(v=5)
self.assertEquals(6, ex2.prop)
if __name__ == '__main__':
unittest.main()