using System; using System.Collections.Generic; using System.Linq; using System.Threading.Tasks; using Alphaleonis.Win32.Filesystem; using Microsoft.VisualStudio.TestTools.UnitTesting; using Wabbajack.Common; using Wabbajack.Lib; using Wabbajack.Lib.Downloaders; using Wabbajack.Lib.NexusApi; namespace Wabbajack.Test { [TestClass] public class EndToEndTests { private const string DOWNLOAD_FOLDER = "downloads"; private TestUtils utils = new TestUtils(); public TestContext TestContext { get; set; } public WorkQueue Queue { get; set; } [TestInitialize] public void TestInitialize() { Queue = new WorkQueue(); Consts.TestMode = true; utils = new TestUtils(); utils.Game = Game.SkyrimSpecialEdition; Utils.LogMessages.Subscribe(f => TestContext.WriteLine(f.ToString())); if (!Directory.Exists(DOWNLOAD_FOLDER)) Directory.CreateDirectory(DOWNLOAD_FOLDER); } [TestCleanup] public void Cleanup() { Queue.Dispose(); } [TestMethod] public async Task CreateModlist() { var profile = utils.AddProfile("Default"); var mod = utils.AddMod(); await DownloadAndInstall( "https://github.com/ModOrganizer2/modorganizer/releases/download/v2.2.1/Mod.Organizer.2.2.1.7z", "Mod.Organizer.2.2.1.7z", utils.MO2Folder); File.WriteAllLines(Path.Combine(utils.DownloadsFolder, "Mod.Organizer.2.2.1.7z.meta"), new List { "[General]", "directURL=https://github.com/ModOrganizer2/modorganizer/releases/download/v2.2.1/Mod.Organizer.2.2.1.7z" }); await DownloadAndInstall(Game.SkyrimSpecialEdition, 12604, "SkyUI"); utils.Configure(); var modlist = await CompileAndInstall(profile); utils.VerifyAllFiles(); var loot_folder = Path.Combine(utils.InstallFolder, "LOOT Config Files"); if (Directory.Exists(loot_folder)) Utils.DeleteDirectory(loot_folder); var compiler = new MO2Compiler( mo2Folder: utils.InstallFolder, mo2Profile: profile, outputFile: profile + ExtensionManager.Extension); compiler.MO2DownloadsFolder = Path.Combine(utils.DownloadsFolder); compiler.ShowReportWhenFinished = false; Assert.IsTrue(await compiler.Begin()); } private async Task DownloadAndInstall(string url, string filename, string mod_name = null) { var src = Path.Combine(DOWNLOAD_FOLDER, filename); if (!File.Exists(src)) { var state = DownloadDispatcher.ResolveArchive(url); await state.Download(new Archive { Name = "Unknown"}, src); } if (!Directory.Exists(utils.DownloadsFolder)) { Directory.CreateDirectory(utils.DownloadsFolder); } File.Copy(src, Path.Combine(utils.DownloadsFolder, filename)); await FileExtractor.ExtractAll(Queue, src, mod_name == null ? utils.MO2Folder : Path.Combine(utils.ModsFolder, mod_name)); } private async Task DownloadAndInstall(Game game, int modid, string mod_name) { utils.AddMod(mod_name); var client = await NexusApiClient.Get(); var resp = await client.GetModFiles(game, modid); var file = resp.files.First(f => f.is_primary); var src = Path.Combine(DOWNLOAD_FOLDER, file.file_name); var ini = string.Join("\n", new List { "[General]", $"gameName={game.MetaData().MO2ArchiveName}", $"modID={modid}", $"fileID={file.file_id}" }); if (!File.Exists(src)) { var state = (AbstractDownloadState)await DownloadDispatcher.ResolveArchive(ini.LoadIniString()); await state.Download(src); } if (!Directory.Exists(utils.DownloadsFolder)) { Directory.CreateDirectory(utils.DownloadsFolder); } var dest = Path.Combine(utils.DownloadsFolder, file.file_name); File.Copy(src, dest); await FileExtractor.ExtractAll(Queue, src, Path.Combine(utils.ModsFolder, mod_name)); File.WriteAllText(dest + ".meta", ini); } private async Task CompileAndInstall(string profile) { var compiler = await ConfigureAndRunCompiler(profile); await Install(compiler); return compiler.ModList; } private async Task Install(MO2Compiler compiler) { var modlist = AInstaller.LoadFromFile(compiler.ModListOutputFile); var installer = new MO2Installer( archive: compiler.ModListOutputFile, modList: modlist, outputFolder: utils.InstallFolder, downloadFolder: utils.DownloadsFolder); installer.GameFolder = utils.GameFolder; await installer.Begin(); } private async Task ConfigureAndRunCompiler(string profile) { var compiler = new MO2Compiler( mo2Folder: utils.MO2Folder, mo2Profile: profile, outputFile: profile + ExtensionManager.Extension); compiler.ShowReportWhenFinished = false; Assert.IsTrue(await compiler.Begin()); return compiler; } } }