public void ShouldRemoveDataObject()
{
// arrange
ControllerContext context = new ControllerContext();
MockController c = new MockController();
context.Controller = c;
context.Controller.TempData["Test"] = "DataValue";
MockAttribute attr = new MockAttribute("Test");
// act
attr.Remove(context);
// assert
Assert.IsFalse(c.TempData.ContainsKey("Test"));
}
public void ShouldSetDataObject()
{
// arrange
ControllerContext context = new ControllerContext();
MockController c = new MockController();
context.Controller = c;
MockAttribute attr = new MockAttribute("Test");
// act
attr.Set(context, "DataValue");
// assert
Assert.IsTrue(c.TempData.ContainsKey("Test"));
Assert.AreEqual("DataValue", (string)c.TempData["Test"]);
}
public void ShouldGetDataObject()
{
// arrange
ControllerContext context = new ControllerContext();
context.Controller = new MockController();
context.Controller.TempData["Test"] = "DataValue";
MockAttribute attr = new MockAttribute("Test");
// act
string s = attr.Get(context) as string;
// assert
Assert.AreNotEqual(null, s);
Assert.AreEqual("DataValue", s);
}
