Fixed: Fixed an issue where there could be a race condition during app update.

This commit is contained in:
kay.one 2012-02-26 13:22:35 -08:00
parent aa42111516
commit 39be4c567d
3 changed files with 16 additions and 7 deletions

View File

@ -57,6 +57,12 @@ namespace NzbDrone.Common
return process;
}
public virtual void WaitForExit(Process process)
{
Logger.Trace("Waiting for process {0} to exit.", process.ProcessName);
process.WaitForExit();
}
public virtual void Kill(int processId)
{
if (processId == 0 || !Process.GetProcesses().Any(p => p.Id == processId))

View File

@ -2,6 +2,7 @@
using System;
using System.Diagnostics;
using System.IO;
using FizzWare.NBuilder;
using FluentAssertions;
using Moq;
using NUnit.Framework;
@ -62,7 +63,7 @@ namespace NzbDrone.Core.Test.JobTests
Mocker.GetMock<DiskProvider>().Verify(c => c.DeleteFolder(SANDBOX_FOLDER, true), Times.Never());
}
[Test]
public void Should_download_update_package()
{
@ -126,7 +127,7 @@ namespace NzbDrone.Core.Test.JobTests
public void when_no_updates_are_available_should_return_without_error_or_warnings()
{
Mocker.GetMock<UpdateProvider>().Setup(c => c.GetAvilableUpdate(It.IsAny<Version>())).Returns((UpdatePackage)null);
StartUpdate();
ExceptionVerification.AssertNoUnexcpectedLogs();

View File

@ -47,7 +47,7 @@ namespace NzbDrone.Core.Jobs
}
public virtual void Start(ProgressNotification notification, int targetId, int secondaryTargetId)
{
{
notification.CurrentMessage = "Checking for updates";
var updatePackage = _updateProvider.GetAvilableUpdate(_enviromentProvider.Version);
@ -82,12 +82,14 @@ namespace NzbDrone.Core.Jobs
logger.Info("Starting update client");
var startInfo = new ProcessStartInfo
{
FileName = _enviromentProvider.GetUpdateClientExePath(),
Arguments = string.Format("{0} {1}", _enviromentProvider.NzbDroneProcessIdFromEnviroment, _configFileProvider.Guid)
};
FileName = _enviromentProvider.GetUpdateClientExePath(),
Arguments = string.Format("{0} {1}", _enviromentProvider.NzbDroneProcessIdFromEnviroment, _configFileProvider.Guid)
};
_processProvider.Start(startInfo);
var process = _processProvider.Start(startInfo);
notification.CurrentMessage = "Update in progress. NzbDrone will restart shortly.";
_processProvider.WaitForExit(process);
}
}
}