public void GetFlashMessages_returns_a_collection()
{
TempDataDictionary tempData = new TempDataDictionary();
var original = new FlashMessageCollection(tempData);
original.Add(new FlashMessage("info", "Hello, World!"));
var flashMessages = tempData.GetFlashMessages();
Assert.AreEqual(1, flashMessages.Count());
}
/// <summary>
/// Add a new message to the flash with the given key.
/// </summary>
/// <param name="controller">Controller.</param>
/// <param name="key">Flash key.</param>
/// <param name="message">Flash message.</param>
public static void Flash(this Controller controller, string key, string message)
{
Contract.Requires<ArgumentNullException>(controller != null);
Contract.Requires<NullReferenceException>(controller.TempData != null);
Contract.Requires<ArgumentNullException>(key != null);
Contract.Requires<ArgumentNullException>(message != null);
Contract.Requires<ArgumentException>(!string.IsNullOrEmpty(key));
FlashMessageCollection flashMessageCollection = new FlashMessageCollection(controller.TempData);
FlashMessage flashMessage = new FlashMessage(key, message);
flashMessageCollection.Add(flashMessage);
}
