wabbajack/Wabbajack.BuildServer.Test/IndexedFilesTests.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using Wabbajack.BuildServer.Model.Models;
using Wabbajack.BuildServer.Models.Jobs;
using Wabbajack.Common;
using Wabbajack.Lib;
using Wabbajack.Lib.Downloaders;
using Wabbajack.Lib.FileUploader;
using Xunit;
using Xunit.Abstractions;
using Xunit.Priority;

namespace Wabbajack.BuildServer.Test
{
    [Collection("ServerTests")]
    public class IndexedFilesTests : ABuildServerSystemTest
    {

        [Fact, Priority(1)]
        public async Task CanIngestExportedInis()
        {
            var to = Fixture.ServerTempFolder.Combine("IniIngest");
            await @"sql\DownloadStates".RelativeTo(AbsolutePath.EntryPoint).CopyDirectoryToAsync(to);
            var result = await _authedClient.GetStringAsync(MakeURL("indexed_files/ingest/IniIngest"));
            Assert.Equal("5", result);
        }

        [Fact, Priority(2)]
        public async Task CanQueryViaHash()
        {
            var hashes = new HashSet<Hash>
            {
                Hash.FromHex("097ad17ef4b9f5b7"),
                Hash.FromHex("96fb53c3dc6397d2"),
                Hash.FromHex("97a6d27b7becba19")
            };

            foreach (var hash in hashes)
            {
                Utils.Log($"Testing Archive {hash}");
                var ini = await ClientAPI.GetModIni(hash);
                Assert.NotNull(ini);
                Assert.NotNull(DownloadDispatcher.ResolveArchive(ini.LoadIniString()));
            }
        }

        [Fact]
        public async Task CanNotifyOfInis()
        {
            var archive =
                new Archive
                {
                    State = new NexusDownloader.State
                    {
                        Game = Game.SkyrimSpecialEdition,
                        ModID = long.MaxValue >> 3,
                        FileID = long.MaxValue >> 3,
                    },
                    Name = Guid.NewGuid().ToString()
                };
            Assert.True(await AuthorAPI.UploadPackagedInis(new[] {archive}));

            var SQL = Fixture.GetService<SqlService>();
            var job = await SQL.GetJob();
            Assert.NotNull(job);
            Assert.IsType<IndexJob>(job.Payload);
            var payload = (IndexJob)job.Payload;
            
            Assert.IsType<NexusDownloader.State>(payload.Archive.State);

            var casted = (NexusDownloader.State)payload.Archive.State;
            Assert.Equal(Game.SkyrimSpecialEdition, casted.Game);
            
            // Insert the record into SQL
            await SQL.AddDownloadState(Hash.FromHex("00e8bbbf591f61a3"), casted);

            // Enqueue the same file again
            Assert.True(await AuthorAPI.UploadPackagedInis(new[] {archive}));
            
            // File is aleady indexed so nothing gets enqueued
            Assert.Null(await SQL.GetJob());
        }

        public IndexedFilesTests(ITestOutputHelper output, SingletonAdaptor<BuildServerFixture> fixture) : base(output, fixture)
        {
            
        }
    }
}