using System; using System.IO; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Common; using NzbDrone.Common.EnvironmentInfo; using NzbDrone.Common.Model; using NzbDrone.Core.Test.Framework; using NzbDrone.Core.Update; using NzbDrone.Core.Update.Commands; using NzbDrone.Test.Common; using NzbDrone.Test.Common.Categories; namespace NzbDrone.Core.Test.UpdateTests { [TestFixture] public class UpdateServiceFixture : CoreTest { private string _sandboxFolder; private readonly UpdatePackage _updatePackage = new UpdatePackage { FileName = "NzbDrone.vnext.0.8.1.385.zip", Url = "http://update.nzbdrone.com/v10/master/NzbDrone.master.2.0.0.0.zip", Version = new Version("0.8.1.385") }; [SetUp] public void Setup() { WindowsOnly(); Mocker.GetMock().SetupGet(c => c.TempFolder).Returns(TempFolder); Mocker.GetMock().Setup(c => c.AvailableUpdate()).Returns(_updatePackage); Mocker.GetMock().Setup(c => c.GetCurrentProcess()).Returns(new ProcessInfo { Id = 12 }); _sandboxFolder = Mocker.GetMock().Object.GetUpdateSandboxFolder(); } [Test] public void should_delete_sandbox_before_update_if_folder_exists() { Mocker.GetMock().Setup(c => c.FolderExists(_sandboxFolder)).Returns(true); Subject.Execute(new ApplicationUpdateCommand()); Mocker.GetMock().Verify(c => c.DeleteFolder(_sandboxFolder, true)); } [Test] public void should_not_delete_sandbox_before_update_if_folder_doesnt_exists() { Mocker.GetMock().Setup(c => c.FolderExists(_sandboxFolder)).Returns(false); Subject.Execute(new ApplicationUpdateCommand()); Mocker.GetMock().Verify(c => c.DeleteFolder(_sandboxFolder, true), Times.Never()); } [Test] public void Should_download_update_package() { var updateArchive = Path.Combine(_sandboxFolder, _updatePackage.FileName); Subject.Execute(new ApplicationUpdateCommand()); Mocker.GetMock().Verify(c => c.DownloadFile(_updatePackage.Url, updateArchive)); } [Test] public void Should_extract_update_package() { var updateArchive = Path.Combine(_sandboxFolder, _updatePackage.FileName); Subject.Execute(new ApplicationUpdateCommand()); Mocker.GetMock().Verify(c => c.Extract(updateArchive, _sandboxFolder)); } [Test] public void Should_copy_update_client_to_root_of_sandbox() { var updateClientFolder = Mocker.GetMock().Object.GetUpdateClientFolder(); Subject.Execute(new ApplicationUpdateCommand()); Mocker.GetMock().Verify(c => c.MoveFolder(updateClientFolder, _sandboxFolder)); } [Test] public void should_start_update_client() { Subject.Execute(new ApplicationUpdateCommand()); Mocker.GetMock() .Verify(c => c.Start(It.IsAny(), "12", null, null), Times.Once()); } [Test] public void when_no_updates_are_available_should_return_without_error_or_warnings() { Mocker.GetMock().Setup(c => c.AvailableUpdate()).Returns(null); Subject.Execute(new ApplicationUpdateCommand()); ExceptionVerification.AssertNoUnexcpectedLogs(); } [Test] [IntegrationTest] public void Should_download_and_extract_to_temp_folder() { UseRealHttp(); var updateSubFolder = new DirectoryInfo(Mocker.GetMock().Object.GetUpdateSandboxFolder()); updateSubFolder.Exists.Should().BeFalse(); Mocker.Resolve(); Mocker.SetConstant(Mocker.Resolve()); Subject.Execute(new ApplicationUpdateCommand()); updateSubFolder.Refresh(); updateSubFolder.Exists.Should().BeTrue(); updateSubFolder.GetDirectories("nzbdrone").Should().HaveCount(1); updateSubFolder.GetDirectories().Should().HaveCount(1); updateSubFolder.GetFiles().Should().NotBeEmpty(); } [TearDown] public void TearDown() { ExceptionVerification.IgnoreErrors(); } } }