public void When_GetViewModel_should_call_correct_methods()
{
// Given
var viewModel = new RiskAssessmentReviewsViewModel();
_reviewModelFactory
.Setup(x => x.GetViewModel())
.Returns(viewModel);
// When
_target
.WithCompanyId(_companyId)
.WithRiskAssessmentId(_riskAssessmentId)
.WithUser(_user.Object)
.GetViewModel();
// Then
_reviewModelFactory.VerifyAll();
}
public void When_GetViewModel_should_return_correct_result()
{
// Given
var viewModel = new RiskAssessmentReviewsViewModel();
_reviewModelFactory
.Setup(x => x.GetViewModel())
.Returns(viewModel);
// When
var result = _target
.WithCompanyId(_companyId)
.WithRiskAssessmentId(_riskAssessmentId)
.WithUser(_user.Object)
.GetViewModel();
// Then
Assert.That(result, Is.TypeOf<FireRiskAssessmentReviewsViewModel>());
Assert.That(result.ReviewViewModel, Is.Not.Null);
Assert.That(result.ReviewViewModel.RiskAssessmentType, Is.EqualTo(RiskAssessmentType.FRA));
}
