mirror of https://github.com/micromata/borgbackup-butler.git

Kai Reinhard
28.10.2018 34de8e3ff4872c7e48e78d1fa102aa029ac2b261
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
package de.micromata.borgbutler.cache;
 
import de.micromata.borgbutler.config.BorgRepoConfig;
import de.micromata.borgbutler.data.Archive;
import org.junit.jupiter.api.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
 
import java.io.File;
import java.io.IOException;
import java.lang.reflect.Field;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.attribute.FileTime;
import java.util.ArrayList;
import java.util.List;
 
import static org.junit.jupiter.api.Assertions.*;
 
 
public class ArchiveFilelistCacheTest {
    private static Logger log = LoggerFactory.getLogger(ArchiveFilelistCacheTest.class);
 
    @Test
    void readWriteTest() throws Exception {
        List<FilesystemItem> list = createList(1000000);
        ArchiveFilelistCache cache = new ArchiveFilelistCache(new File("out"), 100);
        cache.removeAllCacheFiles();
        BorgRepoConfig repoConfig = new BorgRepoConfig();
        repoConfig.setRepo("repo");
        Archive archive = createArchive("2018-12-10");
        log.info("Saving " + list.size() + " items to out dir.");
        cache.save(repoConfig, archive, list);
        log.info("Saving done.");
        log.info("Loading items from out dir.");
        List<FilesystemItem> filesystemItems = cache.load(repoConfig, archive);
        log.info("Loading " + filesystemItems.size() + " items done.");
        assertEquals(list.size(), filesystemItems.size());
        for (int i = 0; i < filesystemItems.size(); i++) {
            assertEquals(list.get(i).getPath(), filesystemItems.get(i).getPath());
        }
        cache.removeAllCacheFiles();
    }
 
    @Test
    void readWriteEmptyTest() throws Exception {
        List<FilesystemItem> list = new ArrayList<>();
        ArchiveFilelistCache cache = new ArchiveFilelistCache(new File("out"), 100);
        cache.removeAllCacheFiles();
        BorgRepoConfig repoConfig = new BorgRepoConfig();
        repoConfig.setRepo("repo");
        Archive archive = createArchive("2018-12-09");
        assertNull(cache.load(repoConfig, archive));
        cache.save(repoConfig, archive, list);
        List<FilesystemItem> filesystemItems = cache.load(repoConfig, archive);
        assertNull(cache.load(repoConfig, archive));
        cache.removeAllCacheFiles();
    }
 
    @Test
    void cleanUpMaximumSizeTest() throws Exception {
        List<FilesystemItem> list = createList(1000000);
        ArchiveFilelistCache cache = new ArchiveFilelistCache(new File("out"), 3);
        cache.removeAllCacheFiles();
        BorgRepoConfig repoConfig = new BorgRepoConfig();
        repoConfig.setRepo("repo");
 
        long millis = System.currentTimeMillis();
 
        Archive archive = createArchive("2018-11-20");
        cache.save(repoConfig, archive, list);
        File oldestFile = cache.getFile(repoConfig, archive);
        setLastModificationTime(oldestFile, millis - 10 * 3600000); // Fake lastModifiedTime - 10 h
 
        archive = createArchive("2018-11-21");
        cache.save(repoConfig, archive, list);
        File newestFile = cache.getFile(repoConfig, archive);
        setLastModificationTime(newestFile, millis - 60000); // Fake lastModifiedTime - 1min
 
        archive = createArchive("2018-11-22");
        cache.save(repoConfig, archive, list);
        File file = cache.getFile(repoConfig, archive);
        setLastModificationTime(file, millis - 3600000); // Fake lastModifiedTime - 1 hour
 
        assertTrue(oldestFile.exists());
        assertTrue(newestFile.exists());
        assertTrue(file.exists());
 
        cache.cleanUp();
        assertFalse(oldestFile.exists());
        assertFalse(file.exists());
        assertTrue(newestFile.exists());
        cache.removeAllCacheFiles();
    }
 
    @Test
    void cleanUpExpiredTest() throws Exception {
        List<FilesystemItem> list = createList(1000);
        ArchiveFilelistCache cache = new ArchiveFilelistCache(new File("out"), 3);
        cache.removeAllCacheFiles();
        BorgRepoConfig repoConfig = new BorgRepoConfig();
        repoConfig.setRepo("repo");
 
        long millis = System.currentTimeMillis();
 
        Archive archive = createArchive("2018-10-20");
        cache.save(repoConfig, archive, list);
        File notExpiredFile = cache.getFile(repoConfig, archive);
        setLastModificationTime(notExpiredFile, millis - 6 * 24 * 3600000); // Fake lastModifiedTime - 10 h
 
        archive = createArchive("2018-10-21");
        cache.save(repoConfig, archive, list);
        File expiredFile = cache.getFile(repoConfig, archive);
        setLastModificationTime(expiredFile, millis - 8 * 24 * 3600000); // Fake lastModifiedTime - 10 h
 
        assertTrue(expiredFile.exists());
        assertTrue(notExpiredFile.exists());
 
        cache.cleanUp();
        assertFalse(expiredFile.exists());
        assertTrue(notExpiredFile.exists());
        cache.removeAllCacheFiles();
    }
 
    private List<FilesystemItem> createList(int number) throws Exception {
        List<FilesystemItem> list = new ArrayList<>();
        for (int i = 0; i < 1000000; i++) {
            list.add(create(i));
        }
        return list;
    }
 
    private FilesystemItem create(int i) throws Exception {
        FilesystemItem item = new FilesystemItem();
        set(item, "type", "-").set(item, "mode", "drwxr-xr-x")
                .set(item, "user", "kai").set(item, "group", "user")
                .set(item, "path", "/Users/kai/Test" + i + ".java").set(item, "size", 1000);
        return item;
    }
 
    private ArchiveFilelistCacheTest set(Object obj, String field, Object value) throws Exception {
        Field f1 = obj.getClass().getDeclaredField(field);
        f1.setAccessible(true);
        f1.set(obj, value);
        return this;
    }
 
    private Archive createArchive(String time) throws Exception {
        Archive archive = new Archive();
        set(archive, "name", "archive-" + time);
        set(archive, "time", time);
        return archive;
    }
 
    private void setLastModificationTime(File file, long lastModificationTime) throws IOException {
        Path path = file.toPath();
        FileTime fileTime = FileTime.fromMillis(lastModificationTime);
        Files.setAttribute(path, "lastModifiedTime", fileTime);
    }
}