Fixed WallController tests

VismaLietuva · Feb 9, 2020 · 754a6cf · 754a6cf
1 parent bbe097c
commit 754a6cf
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/api/Tests/Shrooms.Tests/Controllers/WebApi/WallControllerTests.cs b/src/api/Tests/Shrooms.Tests/Controllers/WebApi/WallControllerTests.cs
@@ -16,6 +16,7 @@
 using Shrooms.Domain.Services.Notifications;
 using Shrooms.Domain.Services.Permissions;
 using Shrooms.Domain.Services.Wall;
+using Shrooms.EntityModels.Models.Multiwall;
 using Shrooms.UnitTests.ModelMappings;
 using Shrooms.WebViewModels.Models.User;
 using Shrooms.WebViewModels.Models.Wall;
@@ -38,6 +39,7 @@ public void TestInitializer()
             _wallService = Substitute.For<IWallService>();
             _notificationService = Substitute.For<INotificationService>();
             var permissionService = Substitute.For<IPermissionService>();
+            permissionService.UserHasPermission(Arg.Any<UserAndOrganizationDTO>(), Arg.Any<string>()).Returns(true);
 
             _wallController = new WallController(ModelMapper.Create(), _wallService, _notificationService, permissionService);
             _wallController.ControllerContext = Substitute.For<HttpControllerContext>();
@@ -65,12 +67,12 @@ public async Task Wall_GetWall_List_Should_Return_View_Model()
                     Description = "Description",
                     IsFollowing = false
                 }
-            };  
+            };
 
             _wallService.GetWallsList(null, WallsListFilter.All).ReturnsForAnyArgs(Task.Run(() => walls));
 
             var response = await _wallController.GetWallList(WallsListFilter.All);
-         
+
             Assert.IsInstanceOf<OkNegotiatedContentResult<IEnumerable<WallListViewModel>>>(response);
         }
 
@@ -93,6 +95,7 @@ public async Task Wall_GetWall_Should_Return_View_Model()
                 Id = wallId
             };
 
+            _wallService.GetWall(wallId, null).Returns(new WallDto { Type = WallType.UserCreated });
             _wallService.GetWallDetails(0, null).ReturnsForAnyArgs(Task.Run(() => wall));
 
             var response = await _wallController.GetWall(wallId);
@@ -178,6 +181,7 @@ public async Task Wall_GetPagedWall_Should_Return_View_Model()
                 }
             };
 
+            _wallService.GetWall(wallId, null).Returns(new WallDto { Type = WallType.UserCreated });
             _wallService.GetWallPosts(page, ConstWebApi.DefaultPageSize, null, wallId).ReturnsForAnyArgs(Task.Run(() => posts));
 
             var response = await _wallController.GetPagedWall(wallId, page);