public void testUpgraderDoesntAddBedroomToStudios()
{
Studio studio = new Studio();
UnitUpgrader upgrader = new UnitUpgrader();
upgrader.upgrade(studio);
Assert.Equal(0, studio.numberOfBedrooms);
}
public void testUpgraderAddsBedroomToApartment()
{
PenthouseSuite penthouse = new PenthouseSuite();
UnitUpgrader upgrader = new UnitUpgrader();
upgrader.upgrade(penthouse);
Assert.Equal(5, penthouse.numberOfBedrooms);
}
public void testUpgraderIncreasesSquareFootageOfPenthouse()
{
PenthouseSuite penthouse = new PenthouseSuite();
penthouse.setSquareFootage(1500);
UnitUpgrader upgrader = new UnitUpgrader();
upgrader.upgrade(penthouse);
Assert.Equal(1540, penthouse.squareFootage);
}
public void testUpgraderIncreasesSquareFootageOfStudio()
{
Studio studio = new Studio();
studio.setSquareFootage(550);
UnitUpgrader upgrader = new UnitUpgrader();
upgrader.upgrade(studio);
Assert.Equal(590, studio.squareFootage);
}