@@ -20,16 +20,6 @@ def test_ctor(self):
2020 inst = self ._makeOne ('lock' )
2121 self .assertEqual (inst .lock , 'lock' )
2222
23- class Test_now (unittest .TestCase ):
24- def _callFUT (self ):
25- from .. import now
26- return now ()
27-
28- def test_it (self ):
29- from pytz import UTC
30- result = self ._callFUT ()
31- self .assertEqual (result .tzinfo , UTC )
32-
3323class TestLockOwnerSchema (unittest .TestCase ):
3424 def _makeOne (self ):
3525 from .. import LockOwnerSchema
@@ -210,18 +200,18 @@ def test_ctor(self):
210200 self .assertEqual (inst .comment , 'comment' )
211201
212202 def test_refresh (self ):
213- import datetime
203+ from .. import now
214204 inst = self ._makeOne ()
215- now = datetime . datetime . utcnow ()
216- inst .refresh (when = now )
217- self .assertEqual (inst .last_refresh , now )
205+ NOW = now ()
206+ inst .refresh (when = NOW )
207+ self .assertEqual (inst .last_refresh , NOW )
218208
219209 def test_refresh_with_timeout (self ):
220- import datetime
210+ from .. import now
221211 inst = self ._makeOne ()
222- now = datetime . datetime . utcnow ()
223- inst .refresh (timeout = 30 , when = now )
224- self .assertEqual (inst .last_refresh , now )
212+ NOW = now ()
213+ inst .refresh (timeout = 30 , when = NOW )
214+ self .assertEqual (inst .last_refresh , NOW )
225215 self .assertEqual (inst .timeout , 30 )
226216
227217 def test_expires_timeout_is_None (self ):
@@ -231,11 +221,12 @@ def test_expires_timeout_is_None(self):
231221
232222 def test_expires_timeout_is_int (self ):
233223 import datetime
224+ from .. import now
234225 inst = self ._makeOne ()
235226 inst .timeout = 30
236- now = datetime . datetime . utcnow ()
237- inst .last_refresh = now
238- self .assertEqual (inst .expires (), now + datetime .timedelta (seconds = 30 ))
227+ NOW = now ()
228+ inst .last_refresh = NOW
229+ self .assertEqual (inst .expires (), NOW + datetime .timedelta (seconds = 30 ))
239230
240231 def test_is_valid_expires_timeout_is_None (self ):
241232 inst = self ._makeOne ()
@@ -244,31 +235,34 @@ def test_is_valid_expires_timeout_is_None(self):
244235
245236 def test_is_valid_expires_timeout_is_int (self ):
246237 import datetime
238+ from .. import now
247239 inst = self ._makeOne ()
248240 inst .timeout = 30
249- now = datetime . datetime . utcnow ()
250- future = now + datetime .timedelta (seconds = 60 )
251- inst .last_refresh = now
252- self .assertTrue (inst .is_valid (now ))
253- self .assertFalse (inst .is_valid (future ))
241+ NOW = now ()
242+ FUTURE = NOW + datetime .timedelta (seconds = 60 )
243+ inst .last_refresh = NOW
244+ self .assertTrue (inst .is_valid (NOW ))
245+ self .assertFalse (inst .is_valid (FUTURE ))
254246
255247 def test_is_valid_expires_resource_id_exists (self ):
256248 import datetime
249+ from .. import now
257250 inst = self ._makeOne ()
258251 inst .timeout = 30
259- now = datetime . datetime . utcnow ()
260- inst .last_refresh = now
252+ NOW = now ()
253+ inst .last_refresh = NOW
261254 inst .__objectmap__ = DummyObjectMap ([1 ])
262- self .assertTrue (inst .is_valid (now ))
255+ self .assertTrue (inst .is_valid (NOW ))
263256
264257 def test_is_valid_expires_resource_id_notexist (self ):
265258 import datetime
259+ from .. import now
266260 inst = self ._makeOne ()
267261 inst .timeout = 30
268- now = datetime . datetime . utcnow ()
269- inst .last_refresh = now
262+ NOW = now ()
263+ inst .last_refresh = NOW
270264 inst .__objectmap__ = DummyObjectMap ([])
271- self .assertFalse (inst .is_valid (now ))
265+ self .assertFalse (inst .is_valid (NOW ))
272266
273267 def test_depth_wo_infinite (self ):
274268 inst = self ._makeOne (infinite = False )
0 commit comments