aboutsummaryrefslogtreecommitdiff
path: root/src/test/java/moe/nea/zwirn/ZwirnTest.java
blob: c0db320d54954f540853069a65d073638b5f10b4 (plain)
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
package moe.nea.zwirn;

import net.fabricmc.stitch.commands.tinyv2.*;
import org.junit.jupiter.api.Test;

import java.io.IOException;
import java.nio.file.FileSystems;
import java.nio.file.Path;
import java.util.Arrays;
import java.util.HashMap;

class ZwirnTest {

    @Test
    void diffTinyFile() throws Exception {
        var base = getBaseFile();
        var overlay = getOverlayFile();
        var merged = Zwirn.mergeTinyFile(base, overlay, "official");
        var unmerged = Zwirn.createOverlayTinyFile(base, merged, Arrays.asList("official", "named"), "official");
        TinyV2Writer.write(overlay, Path.of("overlay.tiny"));
        TinyV2Writer.write(unmerged, Path.of("unmerged.tiny"));
    }

    @Test
    void mergeMCP() throws IOException {
        try (var fs = FileSystems.newFileSystem(Path.of("mcp.zip"))) {
            var merged = Zwirn.enrichSeargeWithMCP(TinyV2Reader.read(Path.of("searge.tiny")), fs.getPath("/"));
            TinyV2Writer.write(merged, Path.of("mcp.tiny"));
        }
    }


    TinyFile getBaseFile() {
        return new TinyFile(
                new TinyHeader(
                        Arrays.asList("official", "intermediary", "named"),
                        2, 0, new HashMap<>()
                ),
                Arrays.asList(
                        new TinyClass(
                                Arrays.asList("A", "ClassA", "SomeClass"),
                                Arrays.asList(
                                        new TinyMethod(
                                                "a",
                                                Arrays.asList("a", "method_a", "doSomething"),
                                                Arrays.asList(
                                                        new TinyMethodParameter(
                                                                0,
                                                                Arrays.asList("a", "param_a_a", "somethingToOperateOn"),
                                                                Arrays.asList()
                                                        )
                                                ),
                                                Arrays.asList(),
                                                Arrays.asList("method comment")
                                        )
                                ),
                                Arrays.asList(new TinyField("a",
                                        Arrays.asList("a", "field_a", "myField"), Arrays.asList("Field comment"))),
                                Arrays.asList("some comment")
                        ),
                        new TinyClass(
                                Arrays.asList("C", "ClassC", "SomeOtherClass"),
                                Arrays.asList(),
                                Arrays.asList(),
                                Arrays.asList()
                        )
                )
        );
    }

    @Test
    void mergeTinyFile() throws IOException {
        var base = getBaseFile();
        var overlay = getOverlayFile();
        var merged = Zwirn.mergeTinyFile(base, overlay, "official");
        TinyV2Writer.write(merged, Path.of("out.tiny"));
    }

    TinyFile getOverlayFile() {
        return new TinyFile(
                new TinyHeader(
                        Arrays.asList("official", "named"),
                        2, 0, new HashMap<>()
                ),
                Arrays.asList(
                        new TinyClass(
                                Arrays.asList("A", "SomeClassButNamedBetter"),
                                Arrays.asList(),
                                Arrays.asList(new TinyField("a",
                                        Arrays.asList("a", "myFieldButNamedCool"), Arrays.asList("Better comment"))),
                                Arrays.asList()
                        ),
                        new TinyClass(
                                Arrays.asList("B", "OtherClass")
                        )
                )
        );
    }
}