using BusinessLogic.Interfaces.Services; using FluentAssertions; using log4net; using Microsoft.VisualStudio.TestTools.UnitTesting; using Moq; using System.Collections.Generic; using ViewModel = Admin.Models.UserFundGroup.EditViewModel; using ViewModelBuilder = Admin.Classes.ViewModelBuilders.UserFundGroup.EditViewModelBuilder; namespace Admin.UnitTests.Classes.ViewModelBuilders.UserFundGroup { [TestClass] public class EditViewModelBuilderTests { private readonly Mock _mockLogger = new Mock(); private readonly Mock _mockUserService = new Mock(); private readonly Mock _mockFundGroupService = new Mock(); private readonly Mock _mockUserFundGroupService = new Mock(); private ViewModelBuilder _viewModelBuilder; [TestInitialize] public void TestInitialise() { _viewModelBuilder = new ViewModelBuilder( _mockLogger.Object, _mockUserService.Object, _mockFundGroupService.Object, _mockUserFundGroupService.Object); } private void SetupUserService(Mock service) { service.Setup(x => x.GetUser(It.IsAny())).Returns(new BusinessLogic.Entities.User() { Disabled = false, DisplayName = string.Empty, ExpiryDays = 0, LastLogin = null, UserFundGroups = null, UserId = 0, UserName = string.Empty, UserRoles = null }); } private void SetupFundGroupService(Mock service) { service.Setup(x => x.GetAllFundGroups()).Returns(new List() { { new BusinessLogic.Entities.FundGroup() { FundGroupFunds = null, FundGroupId = 0, Name = "Fund Group 1", UserFundGroups = null } }, { new BusinessLogic.Entities.FundGroup() { FundGroupFunds = null, FundGroupId = 1, Name = "Fund Group 2", UserFundGroups = null } } }); } private void SetupUserFundGroupService(Mock service) { service.Setup(x => x.GetByUserId(It.IsAny())).Returns(new List() { { new BusinessLogic.Entities.UserFundGroup() { FundGroup = null, FundGroupId = 0, User = null, UserFundGroupId = 0, UserId = 0 } }, { new BusinessLogic.Entities.UserFundGroup() { FundGroup = null, FundGroupId = 1, User = null, UserFundGroupId = 1, UserId = 0 } } }); } [TestMethod] public void OnBuildWithoutParamReturnsNull() { // Arrange // Act var result = _viewModelBuilder.Build(); // Assert result.Should().BeNull(); } [TestMethod] public void OnBuildWithParamReturnsViewModel() { // Arrange SetupUserService(_mockUserService); SetupFundGroupService(_mockFundGroupService); SetupUserFundGroupService(_mockUserFundGroupService); // Act var result = _viewModelBuilder.Build(2); // Assert result.Should().BeOfType(typeof(ViewModel)); } [TestMethod] public void OnBuildWithParamNullUserReturnsViewModel() { // Arrange SetupFundGroupService(_mockFundGroupService); SetupUserFundGroupService(_mockUserFundGroupService); // Act var result = _viewModelBuilder.Build(1); // Assert result.Should().BeOfType(typeof(ViewModel)); } [TestMethod] public void OnBuildWithParamNullUserFundGroupsReturnsViewModel() { // Arrange SetupUserService(_mockUserService); SetupFundGroupService(_mockFundGroupService); // Act var result = _viewModelBuilder.Build(1); // Assert result.Should().BeOfType(typeof(ViewModel)); } } }