ProgrammingGit/TreeTest.java at master · markusgeorge1207/ProgrammingGit · GitHub

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
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.security.NoSuchAlgorithmException;

import static org.junit.jupiter.api.Assertions.*;

public class TreeTest {
    private static final String basicTest = "objects/";
    private final String advancedTest = "objects/";
    private String testFile = "index.txt";

    @AfterEach
    void tearDown() {
        deleteDirectory(new File(basicTest));
        deleteDirectory(new File(advancedTest));
    }

    @Test
    void testAddDirectoryBasic() throws IOException {
        File fold = new File("objects/");
        fold.mkdirs();
        Tree tree = new Tree();

        for (int i = 0; i < 3; i++) {
            FileWriter fw = new FileWriter(new File(fold.getPath() + "/" + i));
            fw.write(i);
            fw.close();
        }

        tree.addDirectory ("objects/");
        tree.save();
        assertNotNull(tree.getSHA1());
    }

    @Test
    void testAddTreeEntry ()
    {
        Blob test = null;
        Tree tree = new Tree();

        try
        {
            test = new Blob ();
            tree.addTreeEntry ("tree : ", test.calculateSHA1(testFile), testFile);

        }
        catch (IOException e)
        {
            e.printStackTrace();
        }
        catch (NoSuchAlgorithmException e)
        {
            e.printStackTrace();
        }
        assertTrue(tree.getTreeList().isEmpty());
    }
    @Test
    void testRemoveTreeEntry ()
    {
        Blob test = null;
        Tree tree = new Tree();

        try
        {
            test = new Blob ();
            tree.addTreeEntry ("tree : ", test.calculateSHA1(testFile), testFile);

        }
        catch (IOException e)
        {
            e.printStackTrace();
        }
        catch (NoSuchAlgorithmException e)
        {
            e.printStackTrace();
        }
        tree.remove (testFile);
        assertTrue(tree.getTreeList().isEmpty());
    }

    private void createTempFiles(String directoryPath, int numFiles) throws IOException {
        File directory = new File(directoryPath);
        if (!directory.exists()) {
            directory.mkdir();
        }

        for (int i = 1; i <= numFiles; i++) {
            File file = new File(directory, "file" + i + ".txt");
            file.createNewFile();
        }
    }

    private void deleteDirectory(File directory) {
        if (directory.isDirectory()) {
            File[] files = directory.listFiles();
            if (files != null) {
                for (File file : files) {
                    deleteDirectory(file);
                }
            }
            directory.delete();
        }
    }
}