UniGetUI/src/UniGetUI.Core.Settings.Tests/SecureSettingsTests.cs at 083f3692bcd2d9decf3f7c436515ba01feb93fc8 · Devolutions/UniGetUI · 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
109
110
111
using System.Reflection;
using UniGetUI.Core.Tools;
using SecureSettingsStore = UniGetUI.Core.SettingsEngine.SecureSettings.SecureSettings;

namespace UniGetUI.Core.SettingsEngine.Tests;

public sealed class SecureSettingsTests : IDisposable
{
    private readonly string _testRoot;

    public SecureSettingsTests()
    {
        _testRoot = Path.Combine(Path.GetTempPath(), $"UniGetUI-SecureSettingsTests-{Guid.NewGuid():N}");
        Directory.CreateDirectory(_testRoot);
        SecureSettingsStore.TEST_SecureSettingsRootOverride = _testRoot;
        ClearSecureSettingsCache();
    }

    public void Dispose()
    {
        ClearSecureSettingsCache();
        SecureSettingsStore.TEST_SecureSettingsRootOverride = null;

        if (Directory.Exists(_testRoot))
        {
            Directory.Delete(_testRoot, true);
        }
    }

    [Theory]
    [InlineData(SecureSettingsStore.K.AllowCLIArguments, "AllowCLIArguments")]
    [InlineData(SecureSettingsStore.K.AllowImportingCLIArguments, "AllowImportingCLIArguments")]
    [InlineData(SecureSettingsStore.K.AllowPrePostOpCommand, "AllowPrePostInstallCommands")]
    [InlineData(SecureSettingsStore.K.AllowImportPrePostOpCommands, "AllowImportingPrePostInstallCommands")]
    [InlineData(SecureSettingsStore.K.ForceUserGSudo, "ForceUserGSudo")]
    [InlineData(SecureSettingsStore.K.AllowCustomManagerPaths, "AllowCustomManagerPaths")]
    public void ResolveKey_ReturnsExpectedMappings(SecureSettingsStore.K key, string expected)
    {
        Assert.Equal(expected, SecureSettingsStore.ResolveKey(key));
    }

    [Fact]
    public void ResolveKey_ThrowsForUnsetAndUnknownKeys()
    {
        Assert.Throws<InvalidDataException>(() =>
            SecureSettingsStore.ResolveKey(SecureSettingsStore.K.Unset)
        );
        Assert.Throws<KeyNotFoundException>(() =>
            SecureSettingsStore.ResolveKey((SecureSettingsStore.K)999)
        );
    }

    [Fact]
    public void Get_ReturnsFalseWhenSettingDoesNotExist()
    {
        Assert.False(SecureSettingsStore.Get(SecureSettingsStore.K.AllowCLIArguments));
        Assert.False(Directory.Exists(GetCurrentUserSettingsDirectory()));
    }

    [Fact]
    public void ApplyForUser_CreatesAndRemovesSanitizedFile()
    {
        const string username = "test:user?";
        const string setting = "setting<with>invalid|chars";

        Assert.Equal(0, SecureSettingsStore.ApplyForUser(username, setting, true));
        Assert.True(File.Exists(GetSettingsFilePath(username, setting)));

        Assert.Equal(0, SecureSettingsStore.ApplyForUser(username, setting, false));
        Assert.False(File.Exists(GetSettingsFilePath(username, setting)));
    }

    [Fact]
    public void Get_RefreshesCachedValueAfterApplyForUserWrites()
    {
        string username = Environment.UserName;
        string setting = SecureSettingsStore.ResolveKey(SecureSettingsStore.K.AllowCLIArguments);

        Assert.False(SecureSettingsStore.Get(SecureSettingsStore.K.AllowCLIArguments));

        Assert.Equal(0, SecureSettingsStore.ApplyForUser(username, setting, true));
        Assert.True(File.Exists(GetSettingsFilePath(username, setting)));
        Assert.True(SecureSettingsStore.Get(SecureSettingsStore.K.AllowCLIArguments));

        Assert.Equal(0, SecureSettingsStore.ApplyForUser(username, setting, false));
        Assert.False(File.Exists(GetSettingsFilePath(username, setting)));
        Assert.False(SecureSettingsStore.Get(SecureSettingsStore.K.AllowCLIArguments));
    }

    private string GetCurrentUserSettingsDirectory() =>
        Path.Combine(_testRoot, CoreTools.MakeValidFileName(Environment.UserName));

    private string GetSettingsFilePath(string username, string setting) =>
        Path.Combine(
            _testRoot,
            CoreTools.MakeValidFileName(username),
            CoreTools.MakeValidFileName(setting)
        );

    private static void ClearSecureSettingsCache()
    {
        FieldInfo? cacheField = typeof(SecureSettingsStore).GetField(
            "_cache",
            BindingFlags.NonPublic | BindingFlags.Static
        );
        Assert.NotNull(cacheField);

        var cache = Assert.IsType<Dictionary<string, bool>>(cacheField.GetValue(null));
        cache.Clear();
    }
}