Skip to content

Commit

Permalink
Simplify time_in_range and add test
Browse files Browse the repository at this point in the history
  • Loading branch information
drGrove committed Mar 10, 2022
1 parent df2dd08 commit 7670674
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 3 deletions.
5 changes: 2 additions & 3 deletions mtls_server/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -196,12 +196,11 @@ def create_dir_if_missing(path):
if not os.path.isdir(path):
os.makedirs(path)

def time_in_range(start, end, t):
def time_in_range(start: float, end: float, t: float) -> bool:
"""Return true if t is in the range [start,end]"""
if start <= end:
return start <= t <= end
else:
return start <= t or t <= end
return False

def has_user(gpg, fingerprint):
keys = gpg.list_keys(keys=fingerprint)
Expand Down
7 changes: 7 additions & 0 deletions test/test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,12 @@
import logging
import os
import tempfile
import time
import unittest

from mtls_server.utils import create_dir_if_missing
from mtls_server.utils import get_abs_path
from mtls_server.utils import time_in_range

logging.disable(logging.CRITICAL)
CLEANUP = os.environ.get('CLEANUP', '1')
Expand All @@ -29,3 +31,8 @@ def test_create_dir_if_missing(self):
self.assertTrue(os.path.isdir(new_dir))
if CLEANUP == '1':
self.TEMPDIR.cleanup()

def test_time_in_range(self):
self.assertTrue(time_in_range(1,5,2))
self.assertTrue(time_in_range(time.time()-5, time.time()+5, time.time()))
self.assertFalse(time_in_range(time.time()+5, time.time()-5, time.time()))

0 comments on commit 7670674

Please sign in to comment.