Mirrored_Repos/wabbajack
1
0
Fork 0
mirror of https://github.com/wabbajack-tools/wabbajack.git synced 2024-08-30 18:42:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed naming in Compression.BSA.Test

Browse Source
This commit is contained in:
erri120 2019-11-21 16:41:46 +01:00
parent 8342f00a53
commit abd2ef70d5
No known key found for this signature in database
GPG Key ID: A8C0A18D8D4D3135
1 changed files with 27 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
Compression.BSA.Test/BSATests.cs
View File

@ -18,27 +18,27 @@ namespace Compression.BSA.Test
[TestClass] [TestClass]
public class BSATests public class BSATests
{ {
private static string StagingFolder = "NexusDownloads"; private static string _stagingFolder = "NexusDownloads";
private static string BSAFolder = "BSAs"; private static string _bsaFolder = "BSAs";
private static string TestDir = "BSA Test Dir"; private static string _testDir = "BSA Test Dir";
private static string TempDir = "BSA Temp Dir"; private static string _tempDir = "BSA Temp Dir";
public TestContext TestContext { get; set; } public TestContext TestContext { get; set; }
private static WorkQueue Queue { get; set; } private static WorkQueue Queue { get; set; }
[ClassInitialize] [ClassInitialize]
public static void Setup(TestContext TestContext) public static void Setup(TestContext testContext)
{ {
Queue = new WorkQueue(); Queue = new WorkQueue();
Utils.LogMessages.Subscribe(f => TestContext.WriteLine(f)); Utils.LogMessages.Subscribe(f => testContext.WriteLine(f));
if (!Directory.Exists(StagingFolder)) if (!Directory.Exists(_stagingFolder))
Directory.CreateDirectory(StagingFolder); Directory.CreateDirectory(_stagingFolder);
if (!Directory.Exists(BSAFolder)) if (!Directory.Exists(_bsaFolder))
Directory.CreateDirectory(BSAFolder); Directory.CreateDirectory(_bsaFolder);
var mod_ids = new[] var modIDs = new[]
{ {
(Game.SkyrimSpecialEdition, 12604), // SkyUI (Game.SkyrimSpecialEdition, 12604), // SkyUI
(Game.Skyrim, 3863), // SkyUI (Game.Skyrim, 3863), // SkyUI
@ -46,10 +46,10 @@ namespace Compression.BSA.Test
(Game.Fallout4, 22223) // 10mm SMG (Game.Fallout4, 22223) // 10mm SMG
}; };
foreach (var info in mod_ids) foreach (var info in modIDs)
{ {
var filename = DownloadMod(info); var filename = DownloadMod(info);
var folder = Path.Combine(BSAFolder, info.Item1.ToString(), info.Item2.ToString()); var folder = Path.Combine(_bsaFolder, info.Item1.ToString(), info.Item2.ToString());
if (!Directory.Exists(folder)) if (!Directory.Exists(folder))
Directory.CreateDirectory(folder); Directory.CreateDirectory(folder);
FileExtractor.ExtractAll(Queue, filename, folder); FileExtractor.ExtractAll(Queue, filename, folder);
@ -66,7 +66,7 @@ namespace Compression.BSA.Test
var results = client.GetModFiles(info.Item1, info.Item2); var results = client.GetModFiles(info.Item1, info.Item2);
var file = results.FirstOrDefault(f => f.is_primary) ?? var file = results.FirstOrDefault(f => f.is_primary) ??
results.OrderByDescending(f => f.uploaded_timestamp).First(); results.OrderByDescending(f => f.uploaded_timestamp).First();
var src = Path.Combine(StagingFolder, file.file_name); var src = Path.Combine(_stagingFolder, file.file_name);
if (File.Exists(src)) return src; if (File.Exists(src)) return src;
@ -83,7 +83,7 @@ namespace Compression.BSA.Test
public static IEnumerable<object[]> BSAs() public static IEnumerable<object[]> BSAs()
{ {
return Directory.EnumerateFiles(BSAFolder, "*", DirectoryEnumerationOptions.Recursive) return Directory.EnumerateFiles(_bsaFolder, "*", DirectoryEnumerationOptions.Recursive)
.Where(f => Consts.SupportedBSAs.Contains(Path.GetExtension(f))) .Where(f => Consts.SupportedBSAs.Contains(Path.GetExtension(f)))
.Select(nm => new object[] {nm}); .Select(nm => new object[] {nm});
} }
@ -95,29 +95,29 @@ namespace Compression.BSA.Test
{ {
TestContext.WriteLine($"From {bsa}"); TestContext.WriteLine($"From {bsa}");
TestContext.WriteLine("Cleaning Output Dir"); TestContext.WriteLine("Cleaning Output Dir");
if (Directory.Exists(TempDir)) Directory.Delete(TempDir, true); if (Directory.Exists(_tempDir)) Directory.Delete(_tempDir, true);
//if (Directory.Exists(ArchiveTempDir)) Directory.Delete(ArchiveTempDir, true); //if (Directory.Exists(ArchiveTempDir)) Directory.Delete(ArchiveTempDir, true);
Directory.CreateDirectory(TempDir); Directory.CreateDirectory(_tempDir);
TestContext.WriteLine($"Reading {bsa}"); TestContext.WriteLine($"Reading {bsa}");
string TempFile = Path.Combine("tmp.bsa"); string tempFile = Path.Combine("tmp.bsa");
using (var a = BSADispatch.OpenRead(bsa)) using (var a = BSADispatch.OpenRead(bsa))
{ {
a.Files.PMap(Queue, file => a.Files.PMap(Queue, file =>
{ {
var abs_name = Path.Combine(TempDir, file.Path); var absName = Path.Combine(_tempDir, file.Path);
ViaJson(file.State); ViaJson(file.State);
if (!Directory.Exists(Path.GetDirectoryName(abs_name))) if (!Directory.Exists(Path.GetDirectoryName(absName)))
Directory.CreateDirectory(Path.GetDirectoryName(abs_name)); Directory.CreateDirectory(Path.GetDirectoryName(absName));
using (var fs = File.OpenWrite(abs_name)) using (var fs = File.OpenWrite(absName))
{ {
file.CopyDataTo(fs); file.CopyDataTo(fs);
} }
Assert.AreEqual(file.Size, new FileInfo(abs_name).Length); Assert.AreEqual(file.Size, new FileInfo(absName).Length);
}); });
Console.WriteLine($"Building {bsa}"); Console.WriteLine($"Building {bsa}");
@ -126,17 +126,17 @@ namespace Compression.BSA.Test
{ {
a.Files.PMap(Queue, file => a.Files.PMap(Queue, file =>
{ {
var abs_path = Path.Combine(TempDir, file.Path); var absPath = Path.Combine(_tempDir, file.Path);
using (var str = File.OpenRead(abs_path)) using (var str = File.OpenRead(absPath))
{ {
w.AddFile(ViaJson(file.State), str); w.AddFile(ViaJson(file.State), str);
} }
}); });
w.Build(TempFile); w.Build(tempFile);
} }
Console.WriteLine($"Verifying {bsa}"); Console.WriteLine($"Verifying {bsa}");
using (var b = BSADispatch.OpenRead(TempFile)) using (var b = BSADispatch.OpenRead(tempFile))
{ {
Console.WriteLine($"Performing A/B tests on {bsa}"); Console.WriteLine($"Performing A/B tests on {bsa}");