1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
$NetBSD: patch-isc__dhcp__leases_test__lease.py,v 1.1 2021/09/09 00:21:13 gutteridge Exp $
Support reference datetime for examining historical dhcpd.leases
https://github.com/MartijnBraam/python-isc-dhcp-leases/pull/32
https://github.com/MartijnBraam/python-isc-dhcp-leases/pull/33
--- isc_dhcp_leases/test_lease.py.orig 2018-04-13 13:10:47.000000000 +0000
+++ isc_dhcp_leases/test_lease.py
@@ -1,7 +1,7 @@
+import datetime
from unittest import TestCase
-from isc_dhcp_leases.iscdhcpleases import Lease
+from isc_dhcp_leases.iscdhcpleases import Lease, utc
from freezegun import freeze_time
-from datetime import datetime
__author__ = 'Martijn Braam <martijn@brixit.nl>'
@@ -19,14 +19,14 @@ class TestLease(TestCase):
'client-hostname': '"Satellite-C700"'
}
- @freeze_time("2015-07-6 8:15:0")
def test_init(self):
lease = Lease("192.168.0.1", self.lease_data)
self.assertEqual(lease.ip, "192.168.0.1")
self.assertEqual(lease.hardware, "ethernet")
self.assertEqual(lease.ethernet, "60:a4:4c:b5:6a:dd")
self.assertEqual(lease.hostname, "Satellite-C700")
- self.assertEqual(lease.start, datetime(2013, 12, 10, 12, 57, 4))
+ self.assertEqual(
+ lease.start, datetime.datetime(2013, 12, 10, 12, 57, 4, tzinfo=utc))
self.assertIsNone(lease.end)
self.assertTrue(lease.valid)
self.assertFalse(lease.active)
@@ -36,21 +36,28 @@ class TestLease(TestCase):
lease = Lease("192.168.0.1", self.lease_data)
self.assertEqual(repr(lease), '<Lease 192.168.0.1 for 60:a4:4c:b5:6a:dd (Satellite-C700)>')
- @freeze_time("2015-07-6 8:15:0")
- def test_valid(self):
- lease = Lease("192.168.0.1", self.lease_data)
+ def _test_valid(self, now=None):
+ lease = Lease("192.168.0.1", self.lease_data, now=now)
self.assertTrue(lease.valid) # Lease is forever
- lease.end = datetime(2015, 7, 6, 13, 57, 4)
+ lease.end = datetime.datetime(2015, 7, 6, 13, 57, 4, tzinfo=utc)
self.assertTrue(lease.valid) # Lease is within start and end
- lease.end = datetime(2015, 7, 6, 6, 57, 4)
+ lease.end = lease.end - datetime.timedelta(hours=7)
self.assertFalse(lease.valid) # Lease is ended
- lease.start = datetime(2015, 7, 6, 12, 57, 4)
- lease.end = datetime(2015, 7, 6, 13, 57, 4)
+ lease.start = datetime.datetime(2015, 7, 6, 12, 57, 4, tzinfo=utc)
+ lease.end = lease.start + datetime.timedelta(hours=1)
self.assertFalse(lease.valid) # Lease is in the future
+ @freeze_time("2015-07-6 8:15:0")
+ def test_valid_frozen(self):
+ self._test_valid()
+
+ def test_valid_historical(self):
+ self._test_valid(
+ now=datetime.datetime(2015, 7, 6, 8, 15, 0, tzinfo=utc))
+
def test_eq(self):
lease_a = Lease("192.168.0.1", self.lease_data)
lease_b = Lease("192.168.0.1", self.lease_data)
@@ -83,8 +90,12 @@ class TestLease(TestCase):
lease = Lease("192.168.0.1", self.lease_data)
self.assertTrue(lease.valid) # Lease is forever
- lease.end = datetime(2015, 7, 6, 6, 57, 4)
+ lease.end = datetime.datetime(2015, 7, 6, 6, 57, 4, tzinfo=utc)
self.assertFalse(lease.valid) # Lease is ended
- lease.end = datetime(2015, 7, 6, 9, 57, 4)
+ lease.end = lease.end + datetime.timedelta(hours=3)
self.assertTrue(lease.valid) # Lease is not expired
+
+ def test_naive_time(self):
+ with self.assertRaises(ValueError):
+ Lease("192.168.0.1", self.lease_data, now=datetime.datetime.now())
|
