public void Can_Retrieve_Image_Data()
{
Employee prod = new Employee
{
EmployeeId = 2,
Name = "Test",
ImageData = new byte[] { },
ImageMimeType = "image/png"
};
Mock<IEmployeeRepository> mock = new Mock<IEmployeeRepository>();
mock.Setup(m => m.Employees).Returns(new Employee[]
{
new Employee {EmployeeId = 1, Name = "P1"},
prod,
new Employee {EmployeeId = 3, Name = "P3"},
}.AsQueryable);
EmployeeController target = new EmployeeController(mock.Object);
ActionResult result = target.GetImage(2);
Assert.IsNotNull(result);
Assert.IsInstanceOfType(result, typeof(FileResult));
Assert.AreEqual(prod.ImageMimeType, ((FileResult)result).ContentType);
}
public void Cannot_Retrieve_Image_Data_For_Invalid_ID()
{
Mock<IEmployeeRepository> mock = new Mock<IEmployeeRepository>();
mock.Setup(m => m.Employees).Returns(new Employee[]
{
new Employee {EmployeeId = 1, Name = "P1"},
new Employee {EmployeeId = 2, Name = "P2"},
}.AsQueryable);
EmployeeController target = new EmployeeController(mock.Object);
ActionResult result = target.GetImage(100);
Assert.IsNull(result);
}