public Budgets Read(Func <Budgets, bool> predicate)
{
using (var dbcontext = new GoosExamplePRDEntities())
{
var firstBudget = dbcontext.Budgets.FirstOrDefault(predicate);
return(firstBudget);
}
}
public void Save(Budgets entity)
{
using (var dbcontext = new GoosExamplePRDEntities())
{
var budgetFromDb = dbcontext.Budgets.FirstOrDefault(x => x.YearMonth == entity.YearMonth);
if (budgetFromDb == null)
{
dbcontext.Budgets.Add(entity);
}
else
{
budgetFromDb.Amount = entity.Amount;
}
dbcontext.SaveChanges();
}
}
private static void CleanTableByTags()
{
var tags = ScenarioContext.Current.ScenarioInfo.Tags
.Where(x => x.StartsWith("Clean"))
.Select(x => x.Replace("Clean", ""));
if (!tags.Any())
{
return;
}
using (var dbcontext = new GoosExamplePRDEntities())
{
foreach (var tag in tags)
{
dbcontext.Database.ExecuteSqlCommand($"TRUNCATE TABLE [{tag}]");
}
dbcontext.SaveChangesAsync();
}
}
