public void Test_3MediumC_3BigH_1BigC_1MediumH_1SmallC()
{
//Arrange
trainController trainController = new trainController();
List <Animal> animals = new List <Animal>();
animals.Add(new Carnivore(3, "3C"));
animals.Add(new Carnivore(3, "3C"));
animals.Add(new Carnivore(3, "3C"));
animals.Add(new Herbivore(5, "5H"));
animals.Add(new Herbivore(5, "5H"));
animals.Add(new Herbivore(5, "5H"));
animals.Add(new Carnivore(5, "5C"));
animals.Add(new Herbivore(3, "3H"));
animals.Add(new Carnivore(1, "1C"));
//Action
trainController.AddToTrainFromTest(animals);
//Assert
Assert.AreEqual(trainController.GetWagonsInTrain(), 6);
Assert.AreEqual(trainController.GetAnimalsInTrain(), 9);
}
public void Test_2SmallH()
{
//Arrange
trainController trainController = new trainController();
List <Animal> animals = new List <Animal>();
animals.Add(new Herbivore(1, "1H"));
animals.Add(new Herbivore(1, "1H"));
//Action
trainController.AddToTrainFromTest(animals);
//Assert
Assert.AreEqual(trainController.GetWagonsInTrain(), 1);
Assert.AreEqual(trainController.GetAnimalsInTrain(), 2);
}
