public void Should_AddAnimalToTheTrain_When_AddingAnimalsToTheTrain()
{
//Arrange
Train train = new Train();
Animal animal = new Animal(AnimalType.Herbivore, AnimalSize.Big);
//Act
train.AddAnimalToTrain(animal.FoodType, animal.AnimalSize);
bool AnimalAddedToTrain = (train.Animals[0].FoodType == animal.FoodType && train.Animals[0].AnimalSize == animal.AnimalSize);
//Assert
Assert.True(AnimalAddedToTrain);
}
public void Should_NotAddCarn_When_TheWagonAlreadyHasAnCarn()
{
//Arrange
train.AddAnimalToTrain(AnimalType.Carnivore, AnimalSize.Small);
train.AddAnimalToTrain(AnimalType.Carnivore, AnimalSize.Small);
train.AddAnimalToTrain(AnimalType.Carnivore, AnimalSize.Small);
train.AddAnimalToTrain(AnimalType.Carnivore, AnimalSize.Small);
train.AddAnimalToTrain(AnimalType.Carnivore, AnimalSize.Small);
train.AddAnimalToTrain(AnimalType.Carnivore, AnimalSize.Big);
train.AddAnimalsToWagons();
//Act
foreach (Wagon wagon in train.LoadWagons())
{
int carnivoresInWagon = 0;
foreach (Animal animal in wagon.Animals)
{
if (animal.FoodType == AnimalType.Carnivore)
{
carnivoresInWagon += 1;
}
}
//Assert
Assert.IsFalse(carnivoresInWagon > 1);
}
}