public void DoesAnimalHaveASpotTest4() { CircusTrein.Train train = new CircusTrein.Train(); List <CircusTrein.Animal> animals = new List <Animal>(); // Arrange train.AddAnimalToQueue(MockAnimals.testAnimals[0]); train.AddAnimalToQueue(MockAnimals.testAnimals[4]); train.AddAnimalToQueue(MockAnimals.testAnimals[6]); train.AddAnimalToQueue(MockAnimals.testAnimals[6]); animals.Add(MockAnimals.testAnimals[6]); animals.Add(MockAnimals.testAnimals[6]); animals.Add(MockAnimals.testAnimals[4]); animals.Add(MockAnimals.testAnimals[0]); List <Animal> actualAnimals = new List <Animal>(); train.InsertAnimalsToWagon(); foreach (var wagon in train.RetrieveWagons()) { foreach (var animal in wagon.RetrieveAnimals()) { actualAnimals.Add(animal); } } CollectionAssert.AreEqual(animals, actualAnimals); Assert.AreEqual(2, train.RetrieveWagons().Count); }
public void DoesAnimalHaveASpotTest3() { CircusTrein.Train train = new CircusTrein.Train(); List <CircusTrein.Animal> animals = new List <Animal>(); // Arrange train.AddAnimalToQueue(MockAnimals.testAnimals[0]); train.AddAnimalToQueue(MockAnimals.testAnimals[4]); train.AddAnimalToQueue(MockAnimals.testAnimals[6]); animals.Add(MockAnimals.testAnimals[6]); animals.Add(MockAnimals.testAnimals[4]); animals.Add(MockAnimals.testAnimals[0]); List <CircusTrein.Animal> actualList = new List <Animal>(); // Act train.InsertAnimalsToWagon(); var actualValue = train.RetrieveWagons(); foreach (var wagon in actualValue) { actualList = wagon.RetrieveAnimals(); } // Assert CollectionAssert.AreEqual(animals, actualList); Assert.AreEqual(1, train.RetrieveWagons().Count); }