public void TeamTraits_SetTo_DriverResult()
{
// Arrange
DriverResult driverResult = new DriverResult();
TeamTrait teamTrait1 = new TeamTrait {
Trait = new Trait {
QualyPace = 1
}
};
TeamTrait teamTrait2 = new TeamTrait {
Trait = new Trait {
QualyPace = 2
}
};
List <TeamTrait> teamTraits = new List <TeamTrait> {
teamTrait1, teamTrait2
};
Weather weather = Weather.Sunny;
// Act
RaceService.SetTeamTraitMods(driverResult, teamTraits, weather);
// Assert
int expected = 3; // QualyMod should add up all QualyPace which results in an expected of 3
Assert.Equal(expected, driverResult.QualyMod);
}
public async Task AddTraitToTeam(Team team, Trait trait)
{
TeamTrait newTrait = new TeamTrait {
Team = team, Trait = trait
};
await Context.AddAsync(newTrait);
}
public async Task <IActionResult> RemoveTeamTrait(int teamId, int traitId)
{
Team team = await Context.Teams
.Include(te => te.TeamTraits)
.FirstAsync(te => te.Id == teamId);
Trait trait = await Context.Traits
.FirstAsync(tr => tr.TraitId == traitId);
if (team == null || trait == null)
{
return(NotFound());
}
TeamTrait removetrait = team.TeamTraits
.First(ttr => ttr.TraitId == traitId);
Context.Remove(removetrait);
await Context.SaveChangesAsync();
return(RedirectToAction(nameof(TeamTraits), new { id = teamId }));
}
public async Task <IActionResult> TeamTraits(int id, [Bind("TraitId")] int traitId)
{
Team team = await _teams.GetTeamById(id);
Trait trait = await Context.Traits.FirstAsync(tr => tr.TraitId == traitId);
if (team is null || trait is null)
{
return(NotFound());
}
TeamTrait newTrait = new TeamTrait {
Team = team, Trait = trait
};
DataService.Update(team);
await Context.AddAsync(newTrait);
await Context.SaveChangesAsync();
return(RedirectToAction(nameof(TeamTraits), new { id }));
}
