public void Display_PrintsException_GivenExceptionIsPassed()
{
using (StringWriter sw = new StringWriter())
{
Console.SetOut(sw);
DisplayStrategy displayStrategy = new DisplayStrategy();
displayStrategy.Display(new Exception());
string expected = string.Format("ERROR: Exception of type 'System.Exception' was thrown.", Environment.NewLine);
Assert.AreEqual <string>(expected, sw.ToString());
}
}
public void Display_PrintsResult_GivenNumberIsPassed()
{
using (StringWriter sw = new StringWriter())
{
Console.SetOut(sw);
DisplayStrategy displayStrategy = new DisplayStrategy();
int answer = 5;
displayStrategy.Display(answer);
string expected = string.Format("The difference is 5", Environment.NewLine);
Assert.AreEqual <string>(expected, sw.ToString());
}
}
