forked from third-party-mirrors/zfs_autobackup
148 lines
4.0 KiB
Python
148 lines
4.0 KiB
Python
from basetest import *
|
|
|
|
|
|
class Thing:
|
|
def __init__(self, timestamp):
|
|
self.timestamp=timestamp
|
|
|
|
def __str__(self):
|
|
# age=now-self.timestamp
|
|
struct=time.gmtime(self.timestamp)
|
|
return("{}".format(time.strftime("%Y-%m-%d %H:%M:%S",struct)))
|
|
|
|
|
|
class TestThinner(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
|
|
return super().setUp()
|
|
|
|
def test_incremental(self):
|
|
ok=['2023-01-01 10:09:50',
|
|
'2024-01-01 20:06:35',
|
|
'2025-01-01 09:59:44',
|
|
'2026-01-01 18:06:41',
|
|
'2026-03-08 02:27:07',
|
|
'2026-04-07 02:29:04',
|
|
'2026-05-07 18:39:31',
|
|
'2026-06-06 06:06:14',
|
|
'2026-07-06 03:53:12',
|
|
'2026-08-05 06:23:43',
|
|
'2026-09-04 21:13:46',
|
|
'2026-10-04 00:50:48',
|
|
'2026-11-03 01:52:55',
|
|
'2026-12-03 15:04:25',
|
|
'2027-01-01 09:02:16',
|
|
'2027-01-02 09:59:16',
|
|
'2027-01-28 09:54:49',
|
|
'2027-02-01 08:59:47',
|
|
'2027-02-04 03:24:33',
|
|
'2027-02-11 01:51:49',
|
|
'2027-02-18 04:09:25',
|
|
'2027-02-19 14:21:39',
|
|
'2027-02-20 13:41:38',
|
|
'2027-02-21 07:33:50',
|
|
'2027-02-22 07:39:18',
|
|
'2027-02-23 07:52:18',
|
|
'2027-02-24 02:16:31',
|
|
'2027-02-24 02:17:08',
|
|
'2027-02-24 05:26:13',
|
|
'2027-02-24 12:56:41']
|
|
|
|
|
|
|
|
#some arbitrary date
|
|
now=1589229252
|
|
#we want deterministic results
|
|
random.seed(1337)
|
|
thinner=Thinner("5,10s1min,1d1w,1w1m,1m12m,1y5y")
|
|
things=[]
|
|
|
|
#thin incrementally while adding
|
|
for i in range(0,5000):
|
|
|
|
#increase random amount of time and maybe add a thing
|
|
now=now+random.randint(0,3600*24)
|
|
if random.random()>=0:
|
|
things.append(Thing(now))
|
|
|
|
(keeps, removes)=thinner.thin(things, now=now)
|
|
things=keeps
|
|
|
|
|
|
result=[]
|
|
for thing in things:
|
|
result.append(str(thing))
|
|
|
|
print("Thinner result:")
|
|
pprint.pprint(result)
|
|
|
|
self.assertEqual(result, ok)
|
|
|
|
|
|
def test_full(self):
|
|
|
|
ok=['2022-02-24 15:54:37',
|
|
'2023-01-01 10:09:50',
|
|
'2024-01-01 20:06:35',
|
|
'2025-01-01 09:59:44',
|
|
'2026-01-01 18:06:41',
|
|
'2026-03-01 23:23:58',
|
|
'2026-03-08 02:27:07',
|
|
'2026-04-07 02:29:04',
|
|
'2026-05-07 18:39:31',
|
|
'2026-06-06 06:06:14',
|
|
'2026-07-06 03:53:12',
|
|
'2026-08-05 06:23:43',
|
|
'2026-09-04 21:13:46',
|
|
'2026-10-04 00:50:48',
|
|
'2026-11-03 01:52:55',
|
|
'2026-12-03 15:04:25',
|
|
'2027-01-01 09:02:16',
|
|
'2027-01-02 09:59:16',
|
|
'2027-01-25 20:00:35',
|
|
'2027-01-28 09:54:49',
|
|
'2027-02-01 08:59:47',
|
|
'2027-02-04 03:24:33',
|
|
'2027-02-11 01:51:49',
|
|
'2027-02-18 04:09:25',
|
|
'2027-02-19 14:21:39',
|
|
'2027-02-20 13:41:38',
|
|
'2027-02-21 07:33:50',
|
|
'2027-02-22 07:39:18',
|
|
'2027-02-23 07:52:18',
|
|
'2027-02-24 02:16:31',
|
|
'2027-02-24 02:17:08',
|
|
'2027-02-24 05:26:13',
|
|
'2027-02-24 12:56:41']
|
|
|
|
|
|
|
|
#some arbitrary date
|
|
now=1589229252
|
|
#we want deterministic results
|
|
random.seed(1337)
|
|
thinner=Thinner("5,10s1min,1d1w,1w1m,1m12m,1y5y")
|
|
things=[]
|
|
|
|
for i in range(0,5000):
|
|
|
|
#increase random amount of time and maybe add a thing
|
|
now=now+random.randint(0,3600*24)
|
|
if random.random()>=0:
|
|
things.append(Thing(now))
|
|
|
|
(things, removes)=thinner.thin(things, now=now)
|
|
|
|
result=[]
|
|
for thing in things:
|
|
result.append(str(thing))
|
|
|
|
print("Thinner result:")
|
|
pprint.pprint(result)
|
|
|
|
self.assertEqual(result, ok)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |