Merge pull request #177 from najeeb-ur-rehman/fix/DownloadHandler

Use the Instance DownloadHandler
JanGorman · Mar 23, 2020 · 007c697 · 007c697
2 parents edd90f8 + 4104ad7
commit 007c697
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Agrume/Agrume.swift b/Agrume/Agrume.swift
@@ -401,7 +401,8 @@ extension Agrume: AgrumeDataSource {
   }
 
   public func image(forIndex index: Int, completion: @escaping (UIImage?) -> Void) {
-    if let handler = AgrumeServiceLocator.shared.downloadHandler, let url = images[index].url {
+    let downloadHandler = download ?? AgrumeServiceLocator.shared.downloadHandler
+    if let handler = downloadHandler, let url = images[index].url {
       handler(url, completion)
     } else if let url = images[index].url {
       downloadTask = ImageDownloader.downloadImage(url, completion: completion)