22
33import itertools
44import time
5- from unittest import TestCase
65
76from dogpile .cache import util
87from dogpile .cache .api import NO_VALUE
8+ from dogpile .testing import assert_raises_message
9+ from dogpile .testing import eq_
10+ from dogpile .testing import winsleep
11+ from dogpile .testing .fixtures import _GenericBackendFixture
912from dogpile .util import compat
10- from . import eq_
11- from . import winsleep
12- from ._fixtures import _GenericBackendFixture
1313
1414
15- class DecoratorTest (_GenericBackendFixture , TestCase ):
15+ class DecoratorTest (_GenericBackendFixture ):
1616 backend = "dogpile.cache.memory"
1717
1818 def _fixture (
@@ -189,7 +189,7 @@ def generate_key_with_reversed_order(*args):
189189 eq_ (go .get (3 , 1 ), ["1 2" , "2 1" ])
190190
191191
192- class KeyGenerationTest ( TestCase ) :
192+ class KeyGenerationTest :
193193 def _keygen_decorator (self , namespace = None , ** kw ):
194194 canary = []
195195
@@ -229,7 +229,14 @@ def one(a, b):
229229 pass
230230
231231 gen = canary [0 ]
232- self .assertRaises (ValueError , gen , 1 , b = 2 )
232+ assert_raises_message (
233+ ValueError ,
234+ "dogpile.cache's default key creation function "
235+ "does not accept keyword arguments" ,
236+ gen ,
237+ 1 ,
238+ b = 2 ,
239+ )
233240
234241 def test_kwarg_kegen_keygen_fn (self ):
235242 decorate , canary = self ._kwarg_keygen_decorator ()
@@ -471,7 +478,7 @@ def test_sha1_key_mangler_bytes_py3k(self):
471478 )
472479
473480
474- class CacheDecoratorTest (_GenericBackendFixture , TestCase ):
481+ class CacheDecoratorTest (_GenericBackendFixture ):
475482 backend = "mock"
476483
477484 def test_cache_arg (self ):
@@ -656,7 +663,7 @@ def func(a, b, c=True, *args, **kwargs):
656663 cached_func = reg .cache_on_arguments ()(func )
657664 cached_signature = compat .inspect_getargspec (cached_func )
658665
659- self . assertEqual (signature , cached_signature )
666+ eq_ (signature , cached_signature )
660667
661668 def test_cache_multi_preserve_sig (self ):
662669 reg = self ._region ()
@@ -668,4 +675,4 @@ def func(a, b, c=True, *args, **kwargs):
668675 cached_func = reg .cache_multi_on_arguments ()(func )
669676 cached_signature = compat .inspect_getargspec (cached_func )
670677
671- self . assertEqual (signature , cached_signature )
678+ eq_ (signature , cached_signature )
0 commit comments