public void Setup()
{
_riskAssessmentId = 100;
_companyId = 500;
_viewModelFactory = new Mock<IFireRiskAssessmentReviewsViewModelFactory>();
target = GetTarget();
_viewModelFactory
.Setup(x => x.WithRiskAssessmentId(_riskAssessmentId))
.Returns(_viewModelFactory.Object);
_viewModelFactory
.Setup(x => x.WithCompanyId(_companyId))
.Returns(_viewModelFactory.Object);
var customPrincipal = target.CurrentUser;
_viewModelFactory
.Setup(x => x.WithUser(customPrincipal))
.Returns(_viewModelFactory.Object);
}
private ReviewController GetTarget()
{
var result = new ReviewController(_viewModelFactory.Object, null, null);
return TestControllerHelpers.AddUserToController(result);
}