public void OptimizeRoute_IsStart_ShouldBeFirst(string trainColor, string stationStart, string stationEnd)
{
//Arrange
_stationServiceMock.GetStationByName(stationStart).Returns(_stationsRepository[stationStart]);
_stationServiceMock.GetStationByName(stationEnd).Returns(_stationsRepository[stationEnd]);
_stationServiceMock.GetValidStationLinks(trainColor).Returns(_linksRepository[trainColor]);
IOptimizationService optimizer = new OptimizationService(_stationServiceMock);
//Act
List <Station> path = optimizer.OptimizeRoute(trainColor, stationStart, stationEnd);
//Assert
Station start = path.FirstOrDefault();
Assert.NotNull(start);
Assert.Equal(stationStart, start.Name);
}
