982ebcc217
from https://github.com/parkouss/dlmanager Note this technically should come before the first patch, but mozreview won't show useful interdiffs if I do that, so I'll reorder the patches before landing.
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
import pytest
|
|
import os
|
|
import tempfile
|
|
import time
|
|
import six
|
|
|
|
from dlmanager import fs
|
|
from dlmanager.persist_limit import PersistLimit
|
|
|
|
|
|
class TempCreator(object):
|
|
def __init__(self):
|
|
self.tempdir = tempfile.mkdtemp()
|
|
|
|
def list(self):
|
|
return os.listdir(self.tempdir)
|
|
|
|
def create_file(self, name, size, delay):
|
|
fname = os.path.join(self.tempdir, name)
|
|
with open(fname, 'wb') as f:
|
|
f.write(six.b('a' * size))
|
|
# equivalent to touch, but we apply a delay for the test
|
|
atime = time.time() + delay
|
|
os.utime(fname, (atime, atime))
|
|
|
|
|
|
@pytest.yield_fixture
|
|
def temp():
|
|
tmp = TempCreator()
|
|
yield tmp
|
|
fs.remove(tmp.tempdir)
|
|
|
|
|
|
@pytest.mark.parametrize("size_limit,file_limit,files", [
|
|
# limit_file is always respected
|
|
(10, 5, "bcdef"),
|
|
(10, 3, "def"),
|
|
# if size_limit or file_limit is 0, nothing is removed
|
|
(0, 5, "abcdef"),
|
|
(5, 0, "abcdef"),
|
|
# limit_size works
|
|
(35, 1, "def"),
|
|
])
|
|
def test_persist_limit(temp, size_limit, file_limit, files):
|
|
temp.create_file("a", 10, -6)
|
|
temp.create_file("b", 10, -5)
|
|
temp.create_file("c", 10, -4)
|
|
temp.create_file("d", 10, -3)
|
|
temp.create_file("e", 10, -2)
|
|
temp.create_file("f", 10, -1)
|
|
|
|
persist_limit = PersistLimit(size_limit, file_limit)
|
|
persist_limit.register_dir_content(temp.tempdir)
|
|
persist_limit.remove_old_files()
|
|
|
|
assert ''.join(sorted(temp.list())) == ''.join(sorted(files))
|