public void AddAnimal_Add2Animals_True() { // Arrange CircusTrain train = new CircusTrain(); List <Animal> animals = new List <Animal>() { new Animal(Diet.Carnivore, Size.Large), new Animal(Diet.Carnivore, Size.Large), }; // Act animals = animals.OrderBy(animal => animal.Diet).ThenByDescending(animal => animal.Size).ToList(); animals.ForEach(animal => train.AddAnimal(animal)); // Assert Assert.AreEqual(2, train.Wagons.Count); }
public void AddAnimal_Add3Animals_True() { // Arrange CircusTrain train = new CircusTrain(); List <Animal> animals = new List <Animal>() { new Animal(Diet.Carnivore, Size.Large), new Animal(Diet.Herbivore, Size.Medium), new Animal(Diet.Carnivore, Size.Medium), }; // Act animals = animals.OrderBy(animal => animal.Diet).ThenByDescending(animal => animal.Size).ToList(); animals.ForEach(animal => train.AddAnimal(animal)); IReadOnlyList <Animal> animalsInTrain = train.Animals; // Assert Assert.IsTrue(animalsInTrain[0].Size == Size.Large & animalsInTrain[0].Diet == Diet.Carnivore); Assert.IsTrue(animalsInTrain[1].Size == Size.Medium & animalsInTrain[1].Diet == Diet.Carnivore); Assert.IsTrue(animalsInTrain[2].Size == Size.Medium & animalsInTrain[2].Diet == Diet.Herbivore); }