Add unit test for ImageController