wabbajack/Wabbajack.Server.Test/ArchiveDownloadsTests.cs

using System;
using System.Linq;
using System.Threading.Tasks;
using Wabbajack.Common;
using Wabbajack.Lib;
using Wabbajack.Lib.Downloaders;
using Wabbajack.Server.DataLayer;
using Xunit;
using Xunit.Abstractions;

namespace Wabbajack.BuildServer.Test
{
    public class ArchiveDownloadsTests : ABuildServerSystemTest
    {
        public ArchiveDownloadsTests(ITestOutputHelper output, SingletonAdaptor<BuildServerFixture> fixture) : base(output, fixture)
        {
        }

        [Fact]
        public async Task CanEnqueueDequeueAndUpdateDownloads()
        {
            await ClearDownloaderQueue();
            var state = new HTTPDownloader.State("http://www.google.com");
            var archive = new Archive(state);
            
            var service = Fixture.GetService<SqlService>();
            var id = await service.EnqueueDownload(archive);

            var toRun = await service.GetNextPendingDownload();
            
            Assert.Equal(id, toRun.Id);
            
            await toRun.Finish(service);
            await service.UpdatePendingDownload(toRun);

            toRun = await service.GetNextPendingDownload();
            Assert.Null(toRun);

            var allStates = await service.GetAllArchiveDownloads();
            Assert.Contains(state.PrimaryKeyString, allStates.Select(s => s.PrimaryKeyString));
        }

        [Fact]
        public async Task DontReenqueueDownloadedfiles()
        {
            var hash = Hash.FromLong(Random.Next(int.MinValue, int.MaxValue));
            await ClearDownloaderQueue();
            var _sql = Fixture.GetService<SqlService>();
            var archive = new Archive(new HTTPDownloader.State("http://www.google.com")) {Size = 42, Hash = hash,};
            await _sql.EnqueueDownload(archive);
            
            var download = await _sql.GetNextPendingDownload();
            await download.Finish(_sql);
            Assert.Null(await _sql.GetNextPendingDownload());
            
            var found = await _sql.GetArchiveDownload(archive.State.PrimaryKeyString, archive.Hash, archive.Size);
            Assert.NotNull(found);

            var next = await _sql.GetOrEnqueueArchive(archive);
            Assert.Null(await _sql.GetNextPendingDownload());

            await ClearDownloaderQueue();

        }

        private async Task ClearDownloaderQueue()
        {
            var service = Fixture.GetService<SqlService>();
            while (true)
            {
                var job = await service.GetNextPendingDownload();
                if (job == null) break;

                await job.Fail(service, "Canceled");
            }
        }
    }
}