public void Can_Retrieve_Image_Data()
{
// Arrange - create a location with image data
Location loc = new Location
{
Id = 2,
Name = "Test"
};
// Arrange - create the mock repository
Mock<ILocationRepository> mock = new Mock<ILocationRepository>();
mock.Setup(m => m.Locations).Returns(new Location[]
{
new Location {Id=1, Name = "P1" },
loc,
new Location {Id=3, Name = "P3" }
}.AsQueryable());
// Arrange - create the controller
LocImageController target = new LocImageController(mock.Object);
// Act - call the GetImage action method
ActionResult result = target.GetImage(2);
// Assert
Assert.IsNotNull(result);
Assert.IsInstanceOfType(result, typeof(FileResult));
}
public void Cannot_Retrieve_Image_Data_For_Invalid_Id()
{
// Arrange - create a location with image data
// Arrange - create the mock repository
Mock<ILocationRepository> mock = new Mock<ILocationRepository>();
mock.Setup(m => m.Locations).Returns(new Location[]
{
new Location {Id=1, Name = "P1" },
new Location {Id=2, Name = "P2" }
}.AsQueryable());
// Arrange - create the controller
LocImageController target = new LocImageController(mock.Object);
// Act - call the GetImage action method
ActionResult result = target.GetImage(100);
// Assert
Assert.IsNull(result);
}