public void Can_Retrieve_Image_Data()
{
// Arrange - create a Product with image data
Lot prod = new Lot
{
LotID = 2,
Name = "Test",
ImageData = new byte[] { },
Preview = "image/png"
};
// Arrange - create the mock repository
Mock <ILotRepository> mock = new Mock <ILotRepository>();
mock.Setup(m => m.Lots).Returns(new Lot[] {
new Lot {
LotID = 1, Name = "P1"
},
prod,
new Lot {
LotID = 3, Name = "P3"
}
}.AsQueryable());
// Arrange - create the controller
LotController target = new LotController(mock.Object);
// Act - call the GetImage action method
ActionResult result = target.GetImage(2);
// Assert
Assert.IsNotNull(result);
Assert.IsInstanceOfType(result, typeof(FileResult));
Assert.AreEqual(prod.Preview, ((FileResult)result).ContentType);
}
public void Cannot_Retrieve_Image_Data_For_Invalid_ID()
{
// Arrange - create the mock repository
Mock <ILotRepository> mock = new Mock <ILotRepository>();
mock.Setup(m => m.Lots).Returns(new Lot[] {
new Lot {
LotID = 1, Name = "P1"
},
new Lot {
LotID = 2, Name = "P2"
}
}.AsQueryable());
// Arrange - create the controller
LotController target = new LotController(mock.Object);
// Act - call the GetImage action method
ActionResult result = target.GetImage(100);
// Assert
Assert.IsNull(result);
}
