velopack/test/Velopack.Tests/UpdateManagerTests.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Runtime.CompilerServices;
using System.Text;
using System.Threading.Tasks;
using NuGet.Versioning;
using Velopack.Locators;
using Velopack.Sources;
using Velopack.Tests.TestHelpers;

namespace Velopack.Tests
{
    public class UpdateManagerTests
    {
        private readonly ITestOutputHelper _output;

        public UpdateManagerTests(ITestOutputHelper output)
        {
            _output = output;
        }

        [Fact]
        public void CheckForUpdatesFromLocal()
        {
            using var logger = _output.BuildLoggerFor<UpdateManagerTests>();
            using var _1 = Utility.GetTempDirectory(out var tempPath);

            string releasesSuffix = VelopackRuntimeInfo.IsOSX ? "-osx" : "";
            
            File.WriteAllText(Path.Combine(tempPath, "RELEASES" + releasesSuffix), """
3a2eadd15dd984e4559f2b4d790ec8badaeb6a39  MyCoolApp-1.1.0.nupkg  1040561
94689fede03fed7ab59c24337673a27837f0c3ec  MyCoolApp-1.0.0.nupkg  1004502
""");
            var locator = new TestVelopackLocator("MyCoolApp", "1.0.0", tempPath, logger);
            var um = new UpdateManager(tempPath, null, logger, locator);
            var info = um.CheckForUpdates();
            Assert.NotNull(info);
            Assert.True(new SemanticVersion(1, 1, 0) == info.TargetFullRelease.Version);
            Assert.Equal(0, info.DeltasToTarget.Count());
        }

        [Fact]
        public void CheckForUpdatesFromLocalWithChannel()
        {
            using var logger = _output.BuildLoggerFor<UpdateManagerTests>();
            using var _1 = Utility.GetTempDirectory(out var tempPath);
            File.WriteAllText(Path.Combine(tempPath, "RELEASES-experimental"), """
3a2eadd15dd984e4559f2b4d790ec8badaeb6a39  MyCoolApp-1.1.0.nupkg  1040561
94689fede03fed7ab59c24337673a27837f0c3ec  MyCoolApp-1.0.0.nupkg  1004502
""");
            var locator = new TestVelopackLocator("MyCoolApp", "1.0.0", tempPath, logger);
            var um = new UpdateManager(tempPath, "experimental", logger, locator);
            var info = um.CheckForUpdates();
            Assert.NotNull(info);
            Assert.True(new SemanticVersion(1, 1, 0) == info.TargetFullRelease.Version);
            Assert.Equal(0, info.DeltasToTarget.Count());
        }

        [Fact]
        public void CheckForUpdatesFromRemote()
        {
            using var logger = _output.BuildLoggerFor<UpdateManagerTests>();
            using var _1 = Utility.GetTempDirectory(out var tempPath);
            var releases = """
3a2eadd15dd984e4559f2b4d790ec8badaeb6a39  MyCoolApp-1.1.0.nupkg  1040561
94689fede03fed7ab59c24337673a27837f0c3ec  MyCoolApp-1.0.0.nupkg  1004502
""";
            var downloader = new FakeDownloader() { MockedResponseBytes = Encoding.UTF8.GetBytes(releases) };
            var locator = new TestVelopackLocator("MyCoolApp", "1.0.0", tempPath, logger);
            var um = new UpdateManager(new SimpleWebSource("http://any.com", "hello", downloader, logger), logger, locator);
            var info = um.CheckForUpdates();
            Assert.NotNull(info);
            Assert.True(new SemanticVersion(1, 1, 0) == info.TargetFullRelease.Version);
            Assert.Equal(0, info.DeltasToTarget.Count());
            Assert.Contains("/RELEASES-hello?", downloader.LastUrl);
        }

        [Fact]
        public void CheckForUpdatesFromLocalWithDelta()
        {
            using var logger = _output.BuildLoggerFor<UpdateManagerTests>();
            using var _1 = Utility.GetTempDirectory(out var tempPath);
            string releasesSuffix = VelopackRuntimeInfo.IsOSX ? "-osx" : "";
            File.WriteAllText(Path.Combine(tempPath, "RELEASES" + releasesSuffix), """
3a2eadd15dd984e4559f2b4d790ec8badaeb6a39  MyCoolApp-1.1.0.nupkg  1040561
3a2eadd15dd984e4559f2b4d790ec8badaeb6a39  MyCoolApp-1.2.0.nupkg  1040561
14db31d2647c6d2284882a2e101924a9c409ee67  MyCoolApp-1.2.0-delta.nupkg  80396
14db31d2647c6d2284882a2e101924a9c409ee67  MyCoolApp-1.1.0-delta.nupkg  80396
94689fede03fed7ab59c24337673a27837f0c3ec  MyCoolApp-1.0.0.nupkg  1004502
""");
            var locator = new TestVelopackLocator("MyCoolApp", "1.0.0", tempPath, logger);
            var um = new UpdateManager(tempPath, null, logger, locator);
            var info = um.CheckForUpdates();
            Assert.NotNull(info);
            Assert.True(new SemanticVersion(1, 2, 0) == info.TargetFullRelease.Version);
            Assert.Equal(2, info.DeltasToTarget.Count());
        }

        [Fact(Skip = "Consumes API Quota")]
        public void CheckForUpdatesGithub()
        {
            // https://github.com/caesay/SquirrelCustomLauncherTestApp
            using var logger = _output.BuildLoggerFor<UpdateManagerTests>();
            using var _1 = Utility.GetTempDirectory(out var tempPath);
            var locator = new TestVelopackLocator("MyCoolApp", "1.0.0", tempPath, logger);
            var source = new GithubSource("https://github.com/caesay/SquirrelCustomLauncherTestApp", null, false);
            var um = new UpdateManager(source, logger, locator);
            var info = um.CheckForUpdates();
            Assert.NotNull(info);
            Assert.True(new SemanticVersion(1, 0, 1) == info.TargetFullRelease.Version);
            Assert.Equal(0, info.DeltasToTarget.Count());
        }

        [Fact(Skip = "Consumes API Quota")]
        public void CheckForUpdatesGithubWithNonExistingChannel()
        {
            // https://github.com/caesay/SquirrelCustomLauncherTestApp
            using var logger = _output.BuildLoggerFor<UpdateManagerTests>();
            using var _1 = Utility.GetTempDirectory(out var tempPath);
            var locator = new TestVelopackLocator("MyCoolApp", "1.0.0", tempPath, logger);
            var source = new GithubSource("https://github.com/caesay/SquirrelCustomLauncherTestApp", null, false, "hello");
            var um = new UpdateManager(source, logger, locator);
            Assert.Throws<ArgumentException>(() => um.CheckForUpdates());
        }

        [Fact]
        public void NoUpdatesIfCurrentEqualsRemoteVersion()
        {
            using var logger = _output.BuildLoggerFor<UpdateManagerTests>();
            using var _1 = Utility.GetTempDirectory(out var tempPath);
            string releasesSuffix = VelopackRuntimeInfo.IsOSX ? "-osx" : "";
            File.WriteAllText(Path.Combine(tempPath, "RELEASES" + releasesSuffix), """
3a2eadd15dd984e4559f2b4d790ec8badaeb6a39  MyCoolApp-1.1.0.nupkg  1040561
94689fede03fed7ab59c24337673a27837f0c3ec  MyCoolApp-1.0.0.nupkg  1004502
""");
            var locator = new TestVelopackLocator("MyCoolApp", "1.1.0", tempPath, logger);
            var um = new UpdateManager(tempPath, null, logger, locator);
            var info = um.CheckForUpdates();
            Assert.Null(info);
        }

        [Fact]
        public void NoUpdatesIfCurrentGreaterThanRemoteVersion()
        {
            using var logger = _output.BuildLoggerFor<UpdateManagerTests>();
            using var _1 = Utility.GetTempDirectory(out var tempPath);
            string releasesSuffix = VelopackRuntimeInfo.IsOSX ? "-osx" : "";
            File.WriteAllText(Path.Combine(tempPath, "RELEASES" + releasesSuffix), """
3a2eadd15dd984e4559f2b4d790ec8badaeb6a39  MyCoolApp-1.1.0.nupkg  1040561
94689fede03fed7ab59c24337673a27837f0c3ec  MyCoolApp-1.0.0.nupkg  1004502
""");
            var locator = new TestVelopackLocator("MyCoolApp", "1.2.0", tempPath, logger);
            var um = new UpdateManager(tempPath, null, logger, locator);
            var info = um.CheckForUpdates();
            Assert.Null(info);
        }

        [Theory]
        [InlineData("Clowd", "3.4.287")]
        [InlineData("slack", "1.1.8")]
        public void DownloadsLatestFullVersion(string id, string version)
        {
            using var logger = _output.BuildLoggerFor<UpdateManagerTests>();
            using var _1 = Utility.GetTempDirectory(out var packagesDir);
            var repo = new FakeFixtureRepository(id, false);
            var source = new SimpleWebSource("http://any.com", null, repo, logger);
            var locator = new TestVelopackLocator(id, "1.0.0", packagesDir, logger);
            var um = new UpdateManager(source, logger, locator);

            var info = um.CheckForUpdates();
            Assert.NotNull(info);
            Assert.True(SemanticVersion.Parse(version) == info.TargetFullRelease.Version);
            Assert.Equal(0, info.DeltasToTarget.Count());

            um.DownloadUpdates(info);

            var target = Path.Combine(packagesDir, $"{id}-{version}-full.nupkg");
            Assert.True(File.Exists(target));
            um.VerifyPackageChecksum(info.TargetFullRelease);
        }

        [Theory]
        [InlineData("Clowd", "3.4.287", "3.4.292")]
        //[InlineData("slack", "1.1.8", "1.2.2")]
        public async Task DownloadsDeltasAndCreatesFullVersion(string id, string fromVersion, string toVersion)
        {
            using var logger = _output.BuildLoggerFor<UpdateManagerTests>();
            using var _1 = Utility.GetTempDirectory(out var packagesDir);
            var repo = new FakeFixtureRepository(id, true);
            var source = new SimpleWebSource("http://any.com", null, repo, logger);

            var basePkg = (await source.GetReleaseFeed()).Single(x => x.Version == SemanticVersion.Parse(fromVersion));
            var basePkgFixturePath = PathHelper.GetFixture(basePkg.OriginalFilename);
            var basePkgPath = Path.Combine(packagesDir, basePkg.OriginalFilename);
            File.Copy(basePkgFixturePath, basePkgPath);

            var updateExe = PathHelper.CopyUpdateTo(packagesDir);
            var locator = new TestVelopackLocator(id, fromVersion,
                packagesDir, null, null, updateExe, logger);
            var um = new UpdateManager(source, logger, locator);

            var info = await um.CheckForUpdatesAsync();
            Assert.NotNull(info);
            Assert.True(SemanticVersion.Parse(toVersion) == info.TargetFullRelease.Version);
            Assert.Equal(3, info.DeltasToTarget.Count());
            Assert.NotNull(info.BaseRelease);

            await um.DownloadUpdatesAsync(info);
            var target = Path.Combine(packagesDir, $"{id}-{toVersion}-full.nupkg");
            Assert.True(File.Exists(target));
        }
    }
}