Update the test about download operation, which using protocol instead of class

2018-04-05 15:25:14 +08:00 · 2018-04-05 15:25:14 +08:00 · 1d7dba33f5
parent 73c811dc23
commit 1d7dba33f5
2 changed files with 6 additions and 1 deletions
--- a/SDWebImage/SDWebImageOperation.h
+++ b/SDWebImage/SDWebImageOperation.h
@ -13,3 +13,8 @@
 - (void)cancel;

@end
+
+// NSOperation conform to `SDWebImageOperation`
+@interface NSOperation (SDWebImageOperation) <SDWebImageOperation>
+
+@end
--- a/Tests/Tests/SDWebImageDownloaderTests.m
+++ b/Tests/Tests/SDWebImageDownloaderTests.m
@ -27,7 +27,7 @@
 - (nullable SDWebImageDownloadToken *)addProgressCallback:(SDWebImageDownloaderProgressBlock)progressBlock
                                           completedBlock:(SDWebImageDownloaderCompletedBlock)completedBlock
                                                   forURL:(nullable NSURL *)url
-                                           createCallback:(SDWebImageDownloaderOperation *(^)(void))createCallback;
+                                           createCallback:(NSOperation<SDWebImageDownloaderOperation> *(^)(void))createCallback;
@end