Merge pull request #13 from ueckoken/timedelta

use timedelta instead of primitive int

KM4K.py

@@ -4,6 +4,7 @@
 import os
 import sys
 import time
+from datetime import timedelta
 
 import nfc
 import redis
@@ -17,7 +18,7 @@
 
 
 # 有効期間1週間
-CACHE_EXPIRES_SECONDS = 60 * 60 * 24 * 7
+CACHE_EXPIRES_DELTA = timedelta(weeks=1)
 
 
 def read_nfc():
@@ -44,7 +45,7 @@ def start_system(isopen, okled_pin, ngled_pin, cache: redis.Redis, card: CardSDK
                 if is_registered_sso:
                     # 有効期限付きでRedisに保存
                     # 値は今のところ使わないので適当に1にしておいた
-                    cache.set(idm.decode(), 1, ex=CACHE_EXPIRES_SECONDS)
+                    cache.set(idm.decode(), 1, ex=CACHE_EXPIRES_DELTA)
                     verified = True
             if verified:
                 print("Registered (idm:" + idm.decode() + ")")

test_KM4K.py

@@ -2,6 +2,7 @@
 import contextlib
 import os
 import time
+from datetime import timedelta
 from unittest import TestCase
 from unittest.mock import Mock, create_autospec, patch
 
@@ -180,7 +181,7 @@ def test_start_system_unregistered_card_with_redis_cache(
         mocked_servo.lock.assert_called_once()
         card.verify.assert_called_once()
 
-    @patch("KM4K.CACHE_EXPIRES_SECONDS", 5)
+    @patch("KM4K.CACHE_EXPIRES_DELTA", timedelta(seconds=5))
     @patch("KM4K.servo", autospec=True)
     @patch("KM4K.read_nfc", side_effect=[b"456789", b"456789", InterruptedError])
     def test_start_system_with_redis_cache_expires(