1
1
import time
2
- from unittest import mock
3
2
4
3
from pylibmc import Client
5
4
@@ -13,12 +12,8 @@ def __init__(self, *args, **kwargs):
13
12
14
13
def set (self , * args , ** kwargs ):
15
14
key , value , _time = args
16
- with mock .patch .object (Client , "set" ) as set_mock :
17
- set_mock .return_value = True
18
- res = super ().set (* args , ** kwargs )
19
- assert set_mock .call_args .args == args
20
15
self ._cache [key ] = (value , _time )
21
- return res
16
+ return True
22
17
23
18
def get (self , * args , ** kwargs ):
24
19
(key ,) = args
@@ -28,11 +23,7 @@ def get(self, *args, **kwargs):
28
23
del self ._cache [key ]
29
24
_res = None
30
25
31
- with mock .patch .object (Client , "get" ) as get_mock :
32
- get_mock .return_value = _res [0 ] if _res else None
33
- res = super ().get (* args , ** kwargs )
34
- assert get_mock .call_args .args == args
35
- return res
26
+ return _res [0 ] if _res else None
36
27
37
28
def touch (self , * args , ** kwargs ):
38
29
key , _time = args
@@ -41,12 +32,7 @@ def touch(self, *args, **kwargs):
41
32
if _res :
42
33
self ._cache .update ({key : (_res [0 ], _time )})
43
34
value = True
44
-
45
- with mock .patch .object (Client , "touch" ) as touch_mock :
46
- touch_mock .return_value = value
47
- res = super ().touch (* args , ** kwargs )
48
- assert touch_mock .call_args .args == args
49
- return res
35
+ return value
50
36
51
37
def delete (self , * args , ** kwargs ):
52
38
(key ,) = args
@@ -55,15 +41,7 @@ def delete(self, *args, **kwargs):
55
41
del self ._cache [key ]
56
42
else :
57
43
value = False
58
- with mock .patch .object (Client , "delete" ) as delete_mock :
59
- delete_mock .return_value = value
60
- res = super ().delete (* args , ** kwargs )
61
- assert delete_mock .call_args .args == args
62
- return res
44
+ return value
63
45
64
46
def disconnect_all (self , * args , ** kwargs ):
65
- with mock .patch .object (Client , "disconnect_all" ) as disconnect_all_mock :
66
- disconnect_all_mock .return_value = None
67
- res = super ().disconnect_all (* args , ** kwargs )
68
- assert disconnect_all_mock .call_args .args == args
69
- return res
47
+ return None
0 commit comments