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); }