UpdateProvider will delete update package after installation.

This commit is contained in:
Mark McDowall 2011-11-21 18:53:22 -08:00
parent 50ee3f8d4d
commit 0224db7440
2 changed files with 4 additions and 0 deletions

View File

@ -143,6 +143,9 @@ namespace NzbDrone.Update.Test
Mocker.GetMock<DiskProvider>() Mocker.GetMock<DiskProvider>()
.Setup(c => c.CopyDirectory(UPDATE_FOLDER, TARGET_FOLDER)); .Setup(c => c.CopyDirectory(UPDATE_FOLDER, TARGET_FOLDER));
Mocker.GetMock<DiskProvider>()
.Setup(c => c.DeleteFolder(UPDATE_FOLDER, true));
Mocker.Resolve<UpdateProvider>().Start(TARGET_FOLDER); Mocker.Resolve<UpdateProvider>().Start(TARGET_FOLDER);
} }

View File

@ -79,6 +79,7 @@ namespace NzbDrone.Update.Providers
try try
{ {
_diskProvider.CopyDirectory(_enviromentProvider.GetUpdatePackageFolder(), targetFolder); _diskProvider.CopyDirectory(_enviromentProvider.GetUpdatePackageFolder(), targetFolder);
_diskProvider.DeleteFolder(_enviromentProvider.GetUpdatePackageFolder(), true);
} }
catch (Exception e) catch (Exception e)
{ {