public void Get_all_animals_in_trainWagon()
{
// Arrange
Train train = new Train();
for (int i = 0; i < 3; i++)
{
train.AddAnimalsToWagon(new Animal(Animal.AnimalType.Plants, Animal.AnimalSize.Medium));
}
train.AddAnimalsToWagon(new Animal(Animal.AnimalType.Meat, Animal.AnimalSize.Large));
//Act
int actualAmountOfAnimals = train.GetAnimalFromWagon(0).Count;
// Assert
Assert.AreEqual(3, actualAmountOfAnimals);
}
public void Should_AddAnimalsToTheNewWagon_When_AddingAnimalsToTheWagons()
{
//Arrange
Train train = new Train();
List <Animal> animals = new List <Animal>();
animals.Add(new Animal(AnimalType.Herbivore, AnimalSize.Big));
//Act
train.AddAnimalsToWagon(animals);
bool animal1Added = (train.Wagons[0].Animals[0].AnimalSize == AnimalSize.Big);
//Assert
Assert.True(animal1Added);
}