public UnitOfWork(ApplicationDbContext context)
{
_context = context;
Policies = new PolicyRepository(context);
Coverages = new CoverageRepository(context);
Risks = new RiskRepository(context);
UserPolicies = new UserPolicyRepository(context);
Users = new UserRepository(context);
}
public void Setup()
{
_settledBetCsvFileReader = new Mock <ICsvFileProvider <SettledBet> >();
_unsettledBetCsvFileReader = new Mock <ICsvFileProvider <UnsettledBet> >();
_filePathsProvider = new Mock <IFilePathsProvider>();
_filePathsProvider.Setup(p => p.GetSettledBetsFilePath()).Returns(SettledBetsFilePath);
_filePathsProvider.Setup(p => p.GetUnsettledBetsFilePath()).Returns(UnsettledBetsFilePath);
_repository = new RiskRepository(_filePathsProvider.Object,
_settledBetCsvFileReader.Object,
_unsettledBetCsvFileReader.Object);
}
public void GetRisks_InvalidPageNumber_Raise_Exception()
{
//Arrange
var riskDataSourceMock = new Mock <IRiskDataSource>();
riskDataSourceMock.Setup(x => x.GetRisks()).Returns(() => _validRiskResponse);
var riskRepository = new RiskRepository(riskDataSourceMock.Object);
//Act
var response = riskRepository.GetRisks("Id", Sortorder.Ascending, -1, 5);
//Assert
Assert.AreEqual(response.Count, _validRiskResponse.Count);
}
public void GetRisks_RaiseException_Returns_Error()
{
//Arrange
var riskDataSourceMock = new Mock <IRiskDataSource>();
riskDataSourceMock.Setup(x => x.GetRisks()).Throws <Exception>();
var riskRepository = new RiskRepository(riskDataSourceMock.Object);
//Act
var response = riskRepository.GetRisks("Id", Sortorder.Ascending, 1, 5);
//Assert
Assert.AreEqual(response, _nilRisks);
}
public void GetRisks_When_NoRiskData_Returns_Null()
{
//Arrange
var riskDataSourceMock = new Mock <IRiskDataSource>();
riskDataSourceMock.Setup(x => x.GetRisks()).Returns(() => _nilRisks);
var riskRepository = new RiskRepository(riskDataSourceMock.Object);
//Act
var response = riskRepository.GetRisks("Id", Sortorder.Ascending, 1, 5);
//Assert
Assert.AreEqual(response, _nilRisks);
}
public void GetRisks_WithPagination_Returns_Success()
{
//Arrange
var riskDataSourceMock = new Mock <IRiskDataSource>();
riskDataSourceMock.Setup(x => x.GetRisks()).Returns(() => _validRiskResponse);
var riskRepository = new RiskRepository(riskDataSourceMock.Object);
//Act
var response = riskRepository.GetRisks("Id", Sortorder.Ascending, 1, 5);
//Assert
Assert.AreEqual(response.Count, _validRiskResponse.Count);
}
public RiskVM GetRisk(int riskId)
{
var model = new RiskVM();
var transformer = new RiskTransformer();
using (var db = new ManagementToolEntities())
{
var resp = new RiskRepository(db);
var risk = resp.GetFirstOrDefault(r => r.RiskId == riskId);
model = transformer.Transform(risk);
}
return(model);
}
public List <RiskVM> GetProjectRisks(int projectId)
{
var model = new List <RiskVM>();
var transformer = new RiskTransformer();
using (var db = new ManagementToolEntities())
{
var resp = new RiskRepository(db);
IEnumerable <DBModels.Risk> risks = resp.Get(r => r.ProjectId == projectId);
model = transformer.Transform(risks);
}
return(model);
}
public bool UpdateRisk(RiskVM model)
{
bool success = false;
using (var db = new ManagementToolEntities())
{
var resp = new RiskRepository(db);
var risk = resp.GetFirstOrDefault(r => r.RiskId == model.Id);
risk.Status = model.Status.ToString();
risk.Description = model.Description;
risk.Title = model.Title;
resp.Update(risk);
success = db.SaveChanges() > 0;
}
return(success);
}
public bool CreateRisk(RiskVM model)
{
bool success = false;
using (var db = new ManagementToolEntities())
{
var resp = new RiskRepository(db);
var risk = new Risk
{
Title = model.Title,
Description = model.Description,
ProjectId = model.ProjectId,
Status = model.Status.ToString()
};
resp.Insert(risk);
success = db.SaveChanges() > 0;
}
return(success);
}
public RiskService(RiskRepository risk, RiskOptionRepository riskOption, RiskNewsRepository riskNews)
{
_risk = risk;
_riskOption = riskOption;
_riskNews = riskNews;
}
public void SetUp()
{
RiskDataProviderMock = new Mock <IRiskDataProvider>();
RiskRepository = new RiskRepository(RiskDataProviderMock.Object);
}
