diff --git a/NzbDrone.Common.Test/DiskProviderTests.cs b/NzbDrone.Common.Test/DiskProviderTests.cs index 8729ae4b4..d2c2a950a 100644 --- a/NzbDrone.Common.Test/DiskProviderTests.cs +++ b/NzbDrone.Common.Test/DiskProviderTests.cs @@ -35,11 +35,7 @@ namespace NzbDrone.Common.Test diskProvider.CopyDirectory(BinFolder.FullName, BinFolderCopy.FullName); //Assert - BinFolder.Refresh(); - BinFolderCopy.Refresh(); - - BinFolder.GetFiles("*.*", SearchOption.AllDirectories) - .Should().HaveSameCount(BinFolderCopy.GetFiles("*.*", SearchOption.AllDirectories)); + VerifyCopy(); } [Test] @@ -51,16 +47,24 @@ namespace NzbDrone.Common.Test diskProvider.CopyDirectory(BinFolder.FullName, BinFolderCopy.FullName); //Delete Random File - BinFolderCopy.GetFiles().First().Delete(); + BinFolderCopy.Refresh(); + BinFolderCopy.GetFiles("*.*", SearchOption.AllDirectories).First().Delete(); diskProvider.CopyDirectory(BinFolder.FullName, BinFolderCopy.FullName); //Assert + VerifyCopy(); + } + + private void VerifyCopy() + { BinFolder.Refresh(); BinFolderCopy.Refresh(); - BinFolder.GetFiles("*.*", SearchOption.AllDirectories) - .Should().HaveSameCount(BinFolderCopy.GetFiles("*.*", SearchOption.AllDirectories)); + BinFolderCopy.GetFiles("*.*", SearchOption.AllDirectories) + .Should().HaveSameCount(BinFolder.GetFiles("*.*", SearchOption.AllDirectories)); + + BinFolderCopy.GetDirectories().Should().HaveSameCount(BinFolder.GetDirectories()); } } } diff --git a/NzbDrone.Common/DiskProvider.cs b/NzbDrone.Common/DiskProvider.cs index a3f49be2f..340ad3ff7 100644 --- a/NzbDrone.Common/DiskProvider.cs +++ b/NzbDrone.Common/DiskProvider.cs @@ -67,7 +67,12 @@ namespace NzbDrone.Common targetFolder.Create(); } - foreach (var file in sourceFolder.GetFiles("*.*", SearchOption.AllDirectories)) + foreach (var subDir in sourceFolder.GetDirectories()) + { + CopyDirectory(subDir.FullName, Path.Combine(target, subDir.Name)); + } + + foreach (var file in sourceFolder.GetFiles("*.*", SearchOption.TopDirectoryOnly)) { var destFile = Path.Combine(target, file.Name); file.CopyTo(destFile, true);