public DocumentsViewModel GetViewModel()
{
var existingDocumentsViewModel = new ExistingDocumentsViewModel { CanDeleteDocuments = true, CanEditDocumentType = true, DocumentTypeId = (int)_documentType };
var attachedDocuments = _riskAssessmentAttachmentService.GetRiskAssessmentAttachedDocuments(_riskAssessmentId, _companyId);
foreach (var attachedDocument in attachedDocuments.Where(document => !document.Deleted))
{
existingDocumentsViewModel.PreviouslyAddedDocuments.Add(new PreviouslyAddedDocumentGridRowViewModel()
{
Description = attachedDocument.Description,
DocumentTypeName = attachedDocument.DocumentType.Name,
Id = attachedDocument.Id,
DocumentLibraryId = attachedDocument.DocumentLibraryId,
Filename = attachedDocument.Filename,
});
};
var model = new DocumentsViewModel()
{
CompanyId = _companyId,
RiskAssessmentId = _riskAssessmentId,
ExistingDocumentsViewModel = existingDocumentsViewModel,
DocumentDisplayMessages = _displayMessages.Select(x => new HtmlString(x))
};
return model;
}
public void When_get_index_Then_should_call_appropiate_methods()
{
// Given
var controller = GetTarget();
var viewModel = new DocumentsViewModel();
_viewModelFactory
.Setup(x => x.GetViewModel())
.Returns(viewModel);
// When
controller.Index(_riskAssessmentId, _companyId);
// Then
_viewModelFactory.VerifyAll();
}
public void When_get_index_Then_should_return_correct_view()
{
// Given
var controller = GetTarget();
_viewModelFactory
.Setup(x => x.WithRiskAssessmentId(_riskAssessmentId))
.Returns(_viewModelFactory.Object);
var viewModel = new DocumentsViewModel();
_viewModelFactory
.Setup(x => x.GetViewModel())
.Returns(viewModel);
// When
var result = controller.Index(_riskAssessmentId, _companyId) as ViewResult;
// Then
Assert.That(result.ViewName, Is.EqualTo("Index"));
}
public OverviewViewModel()
{
Documents = new DocumentsViewModel();
Documents.ExistingDocumentsViewModel = new ExistingDocumentsViewModel();
AccidentRecordNotificationMemberEmails = new List<AccidentRecordNotificationMemberEmail>();
}
public void When_get_index_Then_should_return_correct_viewmodel_type()
{
// Given
var controller = GetTarget();
var viewModel = new DocumentsViewModel();
_viewModelFactory
.Setup(x => x.GetViewModel())
.Returns(viewModel);
// When
var result = controller.Index(_riskAssessmentId, _companyId) as ViewResult;
// Then
Assert.That(result.Model, Is.TypeOf<DocumentsViewModel>());
}
