public void Can_Retrieve_Image_Data()
{
// Arrange
Dish dish = new Dish
{
DishId = 2,
Name = "dish2",
ImageData = new byte[] { },
ImageMimeType = "image/png"
};
Mock <IDishRepository> mock = new Mock <IDishRepository>();
mock.Setup(m => m.Dishes).Returns(new List <Dish>
{
new Dish {
DishId = 1, Name = "dish1"
},
dish,
new Dish {
DishId = 3, Name = "dish2"
}
}.AsQueryable());
DishController controller = new DishController(mock.Object);
// Act
ActionResult result = controller.GetImage(2);
// Asserts
Assert.IsNotNull(result);
Assert.IsInstanceOfType(result, typeof(FileResult));
Assert.AreEqual(dish.ImageMimeType, ((FileResult)result).ContentType);
}
public void Cannot_Retrieve_Image_Data_For_Invalid_ID()
{
// Arrange
Mock <IDishRepository> mock = new Mock <IDishRepository>();
mock.Setup(m => m.Dishes).Returns(new List <Dish> {
new Dish {
DishId = 1, Name = "dish1"
},
new Dish {
DishId = 2, Name = "dish2"
}
}.AsQueryable());
DishController controller = new DishController(mock.Object);
// Act
ActionResult result = controller.GetImage(10);
// Asserts
Assert.IsNull(result);
}
