LANCommander/LANCommander.SDK.Tests/SaveService.cs

using LANCommander.SDK.Models;
using System.Management.Automation.Language;

namespace LANCommander.SDK.Tests
{
    public class SaveClientTests
    {
        Guid SaveId;
        Client Client;

        public SaveClientTests() {
            Client = new Client("http://localhost:1337", "C:\\Games");
        }

        [Theory]
        [InlineData("{InstallDir}", "C:\\Games\\Age of Empires", "C:\\Games\\Age of Empires")]
        [InlineData("{InstallDir}\\baseq3", "C:\\Games\\Quake 3\\baseq3", "C:\\Games\\Quake 3")]
        [InlineData("{InstallDir}/baseq3/autoexec.cfg", "C:\\Games\\Quake 3\\baseq3\\autoexec.cfg", "C:\\Games\\Quake 3")]
        [InlineData("%SYSTEMDRIVE%", "C:", "C:\\Games\\Quake 3")]
        [InlineData("%PROGRAMDATA%", "C:\\ProgramData", "C:\\Games\\Quake 3")]
        [InlineData("%SYSTEMROOT%", "C:\\Windows", "C:\\Games\\Quake 3")]
        [InlineData("%USERNAME%", "{UserName}", "C:\\Games")]
        [InlineData("%LOCALAPPDATA%", "C:\\Users\\{UserName}\\AppData\\Local", "C:\\Games")]
        [InlineData("%TEMP%", "C:\\Users\\{UserName}\\AppData\\Local\\Temp", "C:\\Games")]
        public void GetLocalPathBasicsShouldWork(string input, string expected, string installDirectory)
        {
            // Tests to make sure GetLocalPath gets the correct local paths for a given input.
            // Useful to make sure that the full path to the file is returned properly.
            var result = Client.Saves.GetLocalPath(input, installDirectory);

            // To test anything that might have the username in the expected string
            expected = expected.Replace("{UserName}", Environment.UserName);

            Assert.Equal(expected, result);
        }

        [Theory]
        [InlineData("C:\\Games\\Age of Empires", "{InstallDir}", "C:\\Games\\Age of Empires")]
        [InlineData("C:\\", "%SystemDrive%", "C:\\Games\\")]
        [InlineData("C:\\Games\\Quake 3\\baseq3\\autoexec.cfg", "{InstallDir}\\baseq3\\autoexec.cfg", "C:\\Games\\Quake 3")]
        [InlineData("C:\\Users\\{UserName}\\AppData\\Roaming\\.nfs2e", "%APPDATA%\\.nfs2e", "C:\\Games")]
        [InlineData("C:\\Users\\{UserName}\\AppData\\Local\\.minecraft", "%LOCALAPPDATA%\\.minecraft", "C:\\Games")]
        [InlineData("C:\\Users\\{UserName}\\Documents\\My Games\\Praetorians", "%MyDocuments%\\My Games\\Praetorians", "C:\\Games")]

        public void GetActualPathBasicsShouldWork(string input, string expected, string installDirectory)
        {
            input = input.Replace("{UserName}", Environment.UserName);

            var result = Client.Saves.GetActualPath(input, installDirectory);

            Assert.Equal(expected, result);
        }

        [Theory]
        [InlineData("C:\\Games\\Quake 3\\baseq3\\autoexec.cfg", "{InstallDir}", "baseq3/autoexec.cfg", "C:\\Games\\Quake 3")]
        [InlineData("C:\\Games\\Quake 3\\baseq3\\autoexec.cfg", "{InstallDir}", "baseq3/autoexec.cfg", "C:\\Games\\Quake 3\\")]
        [InlineData("C:\\Games\\Age of Empires 2\\player.nfz", "{InstallDir}", "player.nfz", "C:\\Games\\Age of Empires 2\\")]
        [InlineData("C:\\Users\\{UserName}\\AppData\\Roaming\\.nfs2e\\Profiles\\Player1", "%APPDATA%\\.nfs2e", "Profiles/Player1", "C:\\Games\\Need for Speed 2\\")]
        [InlineData("C:\\Users\\{UserName}\\Documents\\My Games\\.nfs2e\\Profiles\\Player1", "%USERPROFILE%\\Documents\\My Games\\.nfs2e", "Profiles/Player1", "C:\\Games\\Need for Speed 2\\")]
        public void GetArchivePathBasicsShouldWork(string path, string workingDirectory, string expected, string installDirectory)
        {
            path = path.Replace("{UserName}", Environment.UserName);

            var result = Client.Saves.GetArchivePath(path, workingDirectory, installDirectory);

            Assert.Equal(expected, result);
        }

        [Fact]
        public void SimpleInstallDirectorySavePathsShouldWork()
        {
            // Arrange
            var installDirectory = Path.GetTempPath();
            var savePath = new SavePath
            {
                Id = Guid.NewGuid(),
                Path = "base\\autoexec.cfg",
                WorkingDirectory = "{InstallDir}",
                Type = Enums.SavePathType.File,
                IsRegex = false
            };

            // Act
            var entries = Client.Saves.GetFileSavePathEntries(savePath, installDirectory);

            // Assert
            Assert.Single(entries);

            var entry = entries.First();

            Assert.Equal($"base/autoexec.cfg", entry.ArchivePath);
            Assert.Equal("{InstallDir}/base/autoexec.cfg", entry.ActualPath);
        }

        [Fact]
        public void RegexInstallDirectorySavePathsShouldWork()
        {
            // Arrange
            var savePath = new SavePath
            {
                Id = Guid.NewGuid(),
                Path = "base\\.*.cfg",
                WorkingDirectory = "{InstallDir}",
                Type = Enums.SavePathType.File,
                IsRegex = true
            };
            var installDirectory = Path.Combine(Path.GetTempPath(), savePath.Id.ToString());

            Directory.CreateDirectory(installDirectory);
            Directory.CreateDirectory(Path.Combine(installDirectory, "base"));
            File.WriteAllText(Path.Combine(installDirectory, "base", "autoexec.cfg"), savePath.Id.ToString());
            File.WriteAllText(Path.Combine(installDirectory, "base", "player.cfg"), savePath.Id.ToString());

            // Act
            var entries = Client.Saves.GetFileSavePathEntries(savePath, installDirectory);

            // Assert
            Assert.Equal(2, entries.Count());
            Assert.True(File.Exists(Path.Combine($"{Path.GetTempPath()}\\{savePath.Id}\\base\\autoexec.cfg")));
            Assert.True(File.Exists(Path.Combine($"{Path.GetTempPath()}\\{savePath.Id}\\base\\player.cfg")));

            var autoexec = entries.First();
            var player = entries.Last();

            Assert.Equal("base/autoexec.cfg", autoexec.ArchivePath);
            Assert.Equal("{InstallDir}/base/autoexec.cfg", autoexec.ActualPath);

            Assert.Equal("base/player.cfg", player.ArchivePath);
            Assert.Equal("{InstallDir}/base/player.cfg", player.ActualPath);
        }
    }
}