UnityDataTools/UnityDataTool.Tests/ExpectedDataGenerator.cs at 7edf43a74281cf94d57fe7fe6bd22c9018d2814f · Unity-Technologies/UnityDataTools · 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
using System.IO;
using Microsoft.Data.Sqlite;
using UnityDataTools.FileSystem;
using UnityDataTools.TestCommon;

namespace UnityDataTools.UnityDataTool.Tests;

// Collect and record the current output returned by the same UnityDataTool commands
// that the tests will run.  Once saved these become the reference data, and if the output
// changes the tests will fail.  So this can be repeated if there is an "expected" change
// in the output.
public static class ExpectedDataGenerator
{
    public static void Generate(Context context)
    {
        var expectedData = context.ExpectedData;

        UnityFileSystem.Init();
        using (var archive = UnityFileSystem.MountArchive(Path.Combine(context.UnityDataFolder, "assetbundle"), "/"))
        {
            expectedData.Add("NodeCount", archive.Nodes.Count);

            foreach (var n in archive.Nodes)
            {
                expectedData.Add(n.Path + "-Size", n.Size);
                expectedData.Add(n.Path + "-Flags", n.Flags);
            }
        }
        UnityFileSystem.Cleanup();

        Program.Main(new string[] { "analyze", Path.Combine(context.UnityDataFolder), "-r" });

        using var db = new SqliteConnection($"Data Source={Path.Combine(Directory.GetCurrentDirectory(), "database.db")};Version=3;New=True;Foreign Keys=False;");
        db.Open();

        using (var cmd = db.CreateCommand())
        {
            cmd.CommandText =
                @"SELECT
                    (SELECT COUNT(*) FROM animation_clips),
                    (SELECT COUNT(*) FROM asset_bundles),
                    (SELECT COUNT(*) FROM assets),
                    (SELECT COUNT(*) FROM audio_clips),
                    (SELECT COUNT(*) FROM meshes),
                    (SELECT COUNT(*) FROM objects),
                    (SELECT COUNT(*) FROM refs),
                    (SELECT COUNT(*) FROM serialized_files),
                    (SELECT COUNT(*) FROM shader_subprograms),
                    (SELECT COUNT(*) FROM shaders),
                    (SELECT COUNT(*) FROM shader_keywords),
                    (SELECT COUNT(*) FROM shader_subprogram_keywords),
                    (SELECT COUNT(*) FROM textures),
                    (SELECT COUNT(*) FROM types)";

            using var reader = cmd.ExecuteReader();

            reader.Read();

            expectedData.Add("animation_clips_count", reader.GetInt32(0));
            expectedData.Add("asset_bundles_count", reader.GetInt32(1));
            expectedData.Add("assets_count", reader.GetInt32(2));
            expectedData.Add("audio_clips_count", reader.GetInt32(3));
            expectedData.Add("meshes_count", reader.GetInt32(4));
            expectedData.Add("objects_count", reader.GetInt32(5));
            expectedData.Add("refs_count", reader.GetInt32(6));
            expectedData.Add("serialized_files_count", reader.GetInt32(7));
            expectedData.Add("shader_subprograms_count", reader.GetInt32(8));
            expectedData.Add("shaders_count", reader.GetInt32(9));
            expectedData.Add("shader_keywords_count", reader.GetInt32(10));
            expectedData.Add("shader_subprogram_keywords_count", reader.GetInt32(11));
            expectedData.Add("textures_count", reader.GetInt32(12));
            expectedData.Add("types_count", reader.GetInt32(13));
        }

        var csprojFolder = Directory.GetParent(context.TestDataFolder).Parent.Parent.Parent.FullName;
        var outputFolder = Path.Combine(csprojFolder, "ExpectedData", context.UnityDataVersion);

        expectedData.Save(outputFolder);

        // Also take a snapshot of the output of running "dump" commands on the test file "assetbundle"
        Directory.CreateDirectory(outputFolder);

        var dumpPath = Path.Combine(outputFolder, "dump");
        Directory.CreateDirectory(dumpPath);
        Program.Main(new string[] { "dump", Path.Combine(context.UnityDataFolder, "assetbundle"), "-o", dumpPath });

        dumpPath = Path.Combine(outputFolder, "dump-s");
        Directory.CreateDirectory(dumpPath);
        Program.Main(new string[] { "dump", Path.Combine(context.UnityDataFolder, "assetbundle"), "-o", dumpPath, "-s" });
    }
}