61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
|
from datetime import datetime, timedelta
|
||
|
|
|
||
|
|
from data import model
|
||
|
|
from data.database import LogEntry3
|
||
|
|
from workers.logrotateworker import LogRotateWorker
|
||
|
|
|
||
|
|
from test.fixtures import *
|
||
|
|
|
||
|
|
|
||
|
|
def test_logrotateworker(initialized_db):
|
||
|
|
worker = LogRotateWorker()
|
||
|
|
logs = list(LogEntry3.select())
|
||
|
|
|
||
|
|
# Ensure there are logs.
|
||
|
|
assert logs
|
||
|
|
|
||
|
|
# Ensure we don't find any to archive that are new.
|
||
|
|
assert not worker._perform_archiving(LogEntry3, datetime.utcnow() - timedelta(weeks=6))
|
||
|
|
assert len(list(LogEntry3.select())) == len(logs)
|
||
|
|
|
||
|
|
# Archive all the logs.
|
||
|
|
assert worker._perform_archiving(LogEntry3, datetime.utcnow() + timedelta(days=1))
|
||
|
|
|
||
|
|
# Ensure all the logs were archived.
|
||
|
|
assert not list(LogEntry3.select())
|
||
|
|
|
||
|
|
|
||
|
|
def test_logrotateworker_withcutoff(initialized_db):
|
||
|
|
# Delete all existing logs.
|
||
|
|
LogEntry3.delete().execute()
|
||
|
|
|
||
|
|
# Create a new set of logs.
|
||
|
|
start_timestamp = datetime.now()
|
||
|
|
for day in range(0, 100):
|
||
|
|
model.log.log_action('push_repo', 'devtable', timestamp=start_timestamp + timedelta(days=day))
|
||
|
|
|
||
|
|
logs = list(LogEntry3.select())
|
||
|
|
|
||
|
|
# Sort and find a midpoint in the logs.
|
||
|
|
logs.sort(key=lambda l: l.datetime)
|
||
|
|
|
||
|
|
midpoint = logs[0:len(logs)/2]
|
||
|
|
assert midpoint
|
||
|
|
assert len(midpoint) < len(logs)
|
||
|
|
|
||
|
|
# Archive the earlier logs.
|
||
|
|
worker = LogRotateWorker()
|
||
|
|
assert worker._perform_archiving(LogEntry3, midpoint[-1].datetime)
|
||
|
|
|
||
|
|
# Ensure the earlier logs were archived.
|
||
|
|
for log in midpoint:
|
||
|
|
try:
|
||
|
|
LogEntry3.get(id=log.id)
|
||
|
|
assert False, "Found unexpected log"
|
||
|
|
except LogEntry3.DoesNotExist:
|
||
|
|
pass
|
||
|
|
|
||
|
|
# Ensure the later logs were not.
|
||
|
|
for log in logs[len(logs)/2:]:
|
||
|
|
LogEntry3.get(id=log.id)
|