def test_decoder_cache_shrinking(tmpdir): cache_dir = str(tmpdir) solver_mock = SolverMock() another_solver = SolverMock('another_solver') cache = DecoderCache(cache_dir=cache_dir) cache.wrap_solver(solver_mock)(**get_solver_test_args()) # Ensure differing time stamps (depending on the file system the timestamp # resolution might be as bad as 1 day). for path in cache.get_files(): timestamp = os.stat(path).st_atime timestamp -= 60 * 60 * 24 * 2 # 2 days os.utime(path, (timestamp, timestamp)) cache.wrap_solver(another_solver)(**get_solver_test_args()) cache_size = cache.get_size_in_bytes() assert cache_size > 0 cache.shrink(cache_size - 1) # check that older cached result was removed assert SolverMock.n_calls[solver_mock] == 1 cache.wrap_solver(another_solver)(**get_solver_test_args()) cache.wrap_solver(solver_mock)(**get_solver_test_args()) assert SolverMock.n_calls[solver_mock] == 2 assert SolverMock.n_calls[another_solver] == 1
def test_decoder_cache_size_includes_overhead(tmpdir): cache_dir = str(tmpdir) solver_mock = SolverMock() cache = DecoderCache(cache_dir=cache_dir) cache.wrap_solver(solver_mock)(**get_solver_test_args()) fragment_size = get_fragment_size(cache_dir) actual_size = sum(os.stat(p).st_size for p in cache.get_files()) assert actual_size % fragment_size != 0, ( 'Test succeeded by chance. Adjust get_solver_test_args() to produce ' 'date not aligned with the files system fragment size.') assert cache.get_size_in_bytes() % fragment_size == 0
def test_corrupted_decoder_cache(tmpdir): cache_dir = str(tmpdir) cache = DecoderCache(cache_dir=cache_dir) solver_mock = SolverMock() cache.wrap_solver(solver_mock)(**get_solver_test_args()) assert SolverMock.n_calls[solver_mock] == 1 # corrupt the cache for path in cache.get_files(): with open(path, 'w') as f: f.write('corrupted') cache.wrap_solver(solver_mock)(**get_solver_test_args()) assert SolverMock.n_calls[solver_mock] == 2