1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
import threading
import time
import unittest
from os import environ
from cachetools import LRUCache, cached, cachedmethod
@cached(cache=LRUCache(1), condition=threading.Condition(), info=True)
def func():
time.sleep(1.0)
if hasattr(func, "count"):
func.count += 1
else:
func.count = 1
return func.count
@unittest.skipUnless(environ.get("THREADING_TESTS", False), "THREADING_TESTS not set")
class ThreadingTest(unittest.TestCase):
NTHREADS = 10
cache = LRUCache(1)
cond = threading.Condition()
count = 0
@cachedmethod(
cache=lambda self: self.cache, condition=lambda self: self.cond, info=True
)
def meth(self):
time.sleep(1.0)
self.count += 1
return 42
def test_cached_stampede(self):
threads = [threading.Thread(target=func) for i in range(0, self.NTHREADS)]
for t in threads:
t.start()
for t in threads:
t.join()
self.assertEqual(func.count, 1)
info = func.cache_info()
self.assertEqual(info.hits, self.NTHREADS - 1)
self.assertEqual(info.misses, 1)
def test_cachedmethod_stampede(self):
threads = [threading.Thread(target=self.meth) for i in range(0, self.NTHREADS)]
for t in threads:
t.start()
for t in threads:
t.join()
self.assertEqual(self.count, 1)
info = self.meth.cache_info()
self.assertEqual(info.hits, self.NTHREADS - 1)
self.assertEqual(info.misses, 1)
|
