public void CurrentColor_When4_ThrowArgumentException()
{
var currentColor = (Traffic_Color)4;
mock = new Mock <IBattery>();
mock.Setup((bat) => bat.SetPowerState(Power.On));
IBattery battery = mock.Object;
trafficLight = new TrafficThreeLight(battery);
Assert.Throws <ArgumentOutOfRangeException>(() => trafficLight.ChangeColor(currentColor));
}
public void Test_Change_Color_To_Green()
{
var currentColor = Traffic_Color.Red;
mock = new Mock <IBattery>();
mock.Setup((bat) => bat.SetPowerState(Power.On));
IBattery battery = mock.Object;
trafficLight = new TrafficThreeLight(battery);
Traffic_Color newColor = trafficLight.ChangeColor(currentColor);
newColor.Should().Be(Traffic_Color.Green);
}
