Update fixes.

NzbDrone.Core.Test/JobTests/AppUpdateJobFixture.cs

@@ -0,0 +1,154 @@
+using System.Linq;
+using System;
+using System.Diagnostics;
+using System.IO;
+using FluentAssertions;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Common;
+using NzbDrone.Core.Model;
+using NzbDrone.Core.Providers;
+using NzbDrone.Core.Providers.Core;
+using NzbDrone.Core.Providers.Jobs;
+using NzbDrone.Core.Test.Framework;
+
+namespace NzbDrone.Core.Test.JobTests
+{
+    [TestFixture]
+    internal class AppUpdateJobFixture : CoreTest
+    {
+        private const string SANDBOX_FOLDER = @"C:\Temp\nzbdrone_update\";
+
+        private readonly Guid _clientGuid = Guid.NewGuid();
+
+        private readonly UpdatePackage updatePackage = new UpdatePackage
+        {
+            FileName = "NzbDrone.kay.one.0.6.0.2031.zip",
+            Url = "http://update.nzbdrone.com/_test/NzbDrone.zip",
+            Version = new Version("0.6.0.2031")
+        };
+
+        [SetUp]
+        public void Setup()
+        {
+            Mocker.GetMock<EnviromentProvider>().SetupGet(c => c.SystemTemp).Returns(@"C:\Temp\");
+            Mocker.GetMock<ConfigFileProvider>().SetupGet(c => c.Guid).Returns(_clientGuid);
+            Mocker.GetMock<UpdateProvider>().Setup(c => c.GetAvilableUpdate()).Returns(updatePackage);
+        }
+
+
+        [Test]
+        public void should_delete_sandbox_before_update_if_folder_exists()
+        {
+            Mocker.GetMock<DiskProvider>().Setup(c => c.FolderExists(SANDBOX_FOLDER)).Returns(true);
+
+            //Act
+            StartUpdate();
+
+            //Assert
+            Mocker.GetMock<DiskProvider>().Verify(c => c.DeleteFolder(SANDBOX_FOLDER, true));
+        }
+
+        [Test]
+        public void should_not_delete_sandbox_before_update_if_folder_doesnt_exists()
+        {
+            Mocker.GetMock<DiskProvider>().Setup(c => c.FolderExists(SANDBOX_FOLDER)).Returns(false);
+
+            //Act
+            StartUpdate();
+
+            //Assert
+            Mocker.GetMock<DiskProvider>().Verify(c => c.DeleteFolder(SANDBOX_FOLDER, true), Times.Never());
+        }
+
+        [Test]
+        public void Should_download_update_package()
+        {
+            var updateArchive = Path.Combine(SANDBOX_FOLDER, updatePackage.FileName);
+
+            //Act
+            StartUpdate();
+
+            //Assert
+            Mocker.GetMock<HttpProvider>().Verify(
+                    c => c.DownloadFile(updatePackage.Url, updateArchive));
+        }
+
+        [Test]
+        public void Should_extract_update_package()
+        {
+            var updateArchive = Path.Combine(SANDBOX_FOLDER, updatePackage.FileName);
+
+            //Act
+            StartUpdate();
+
+            //Assert
+            Mocker.GetMock<ArchiveProvider>().Verify(
+               c => c.ExtractArchive(updateArchive, SANDBOX_FOLDER));
+        }
+
+        [Test]
+        public void Should_copy_update_client_to_root_of_sandbox()
+        {
+            var updateClientFolder = Mocker.GetMock<EnviromentProvider>().Object.GetUpdateClientFolder();
+
+            //Act
+            StartUpdate();
+
+            //Assert
+            Mocker.GetMock<DiskProvider>().Verify(
+               c => c.CopyDirectory(updateClientFolder, SANDBOX_FOLDER));
+        }
+
+        [Test]
+        public void should_start_update_client()
+        {
+            //Setup
+            var updateClientPath = Mocker.GetMock<EnviromentProvider>().Object.GetUpdateClientExePath();
+
+            Mocker.GetMock<EnviromentProvider>()
+                .SetupGet(c => c.NzbDroneProcessIdFromEnviroment).Returns(12);
+
+            //Act
+            StartUpdate();
+
+            //Assert
+            Mocker.GetMock<ProcessProvider>().Verify(
+               c => c.Start(It.Is<ProcessStartInfo>(p =>
+                       p.FileName == updateClientPath &&
+                       p.Arguments == "12 " + _clientGuid.ToString())
+                       ));
+        }
+
+        [Test]
+        [Category(IntegrationTest)]
+        public void Should_download_and_extract_to_temp_folder()
+        {
+
+            Mocker.GetMock<EnviromentProvider>().SetupGet(c => c.SystemTemp).Returns(TempFolder);
+
+            var updateSubFolder = new DirectoryInfo(Mocker.GetMock<EnviromentProvider>().Object.GetUpdateSandboxFolder());
+
+
+            //Act
+            updateSubFolder.Exists.Should().BeFalse();
+
+            Mocker.Resolve<HttpProvider>();
+            Mocker.Resolve<DiskProvider>();
+            Mocker.Resolve<ArchiveProvider>();
+            StartUpdate();
+            updateSubFolder.Refresh();
+            //Assert
+
+            updateSubFolder.Exists.Should().BeTrue();
+            updateSubFolder.GetDirectories("nzbdrone").Should().HaveCount(1);
+            updateSubFolder.GetDirectories().Should().HaveCount(1);
+            updateSubFolder.GetFiles().Should().HaveCount(1);
+        }
+
+        private void StartUpdate()
+        {
+            Mocker.Resolve<AppUpdateJob>().Start(MockNotification, 0, 0);
+        }
+    }
+}