fix: Add IImageFolderViewModel to DI for testing ApplicationViewModel

justaprogrammer · Jan 20, 2019 · 726909c · 726909c
1 parent 6e443af
commit 726909c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/SonOfPicasso.UI.Tests/Extensions/TestExtensions.cs b/SonOfPicasso.UI.Tests/Extensions/TestExtensions.cs
@@ -6,6 +6,7 @@
 using SonOfPicasso.Core.Scheduling;
 using SonOfPicasso.Testing.Common;
 using SonOfPicasso.Testing.Common.Services;
+using SonOfPicasso.UI.Interfaces;
 using SonOfPicasso.UI.Scheduling;
 using SonOfPicasso.UI.Tests.Scheduling;
 using SonOfPicasso.UI.ViewModels;
@@ -18,16 +19,19 @@ public static ApplicationViewModel CreateApplicationViewModel<T>(this TestsBase<
             IFileSystem fileSystem = null,
             ISharedCache sharedCache = null,
             IImageLocationService imageLocationService = null,
-            ISchedulerProvider schedulerProvider = null)
+            ISchedulerProvider schedulerProvider = null,
+            IImageFolderViewModel imageFolderViewModel = null)
         {
             fileSystem = fileSystem ?? new MockFileSystem();
             imageLocationService = imageLocationService ?? Substitute.For<IImageLocationService>();
             schedulerProvider = schedulerProvider ?? new TestSchedulerProvider();
+            imageFolderViewModel = imageFolderViewModel ?? new ImageFolderViewModel();
 
             var serviceCollection = databaseReaderTests.GetServiceCollection()
                 .AddSingleton(fileSystem)
                 .AddSingleton(schedulerProvider)
                 .AddSingleton(imageLocationService)
+                .AddSingleton(imageFolderViewModel)
                 .AddSingleton(typeof(ApplicationViewModel));
 
             if (sharedCache != null)