Fix that YYImage plugin isAllFramesLoaded not work by workaround, update tests

Author: dreampiggy

Example/Tests/SDYYCacheTests.m

@@ -65,7 +65,7 @@ - (void)testCustomImageCache {
 - (void)testYYCacheQueryOp {
     XCTestExpectation *expectation = [self expectationWithDescription:@"SDImageCache query op works"];
     NSData *imageData = [NSData dataWithContentsOfFile:[self testJPEGPath]];
-    [SDYYCacheTests.sharedCache.diskCache setObject:imageData forKey:kTestImageKeyJPEG];
+    [SDYYCacheTests.sharedCache.diskCache setData:imageData forKey:kTestImageKeyJPEG];
     [SDYYCacheTests.sharedCache queryImageForKey:kTestImageKeyJPEG options:0 context:nil completion:^(UIImage * _Nullable image, NSData * _Nullable data, SDImageCacheType cacheType) {
         expect(image).notTo.beNil();
         [expectation fulfill];

Example/Tests/SDYYImageTests.m

@@ -35,7 +35,31 @@ - (void)testYYAnimatedImageViewSetImageWithURL {
                             expect(error).to.beNil();
                             expect(originalImageURL).to.equal(imageURL);
 
-                            expect(imageView.image.class).to.equal(YYImage.class);
+                            expect(image.class).to.equal(YYImage.class);
+                            [expectation fulfill];
+                        }];
+    [self waitForExpectationsWithCommonTimeout];
+}
+
+- (void)testYYAnimatedImageViewSetImageWithURLPreloadAllFrames {
+    XCTestExpectation *expectation = [self expectationWithDescription:@"YYAnimatedImageView setImageWithURL preloadAllFrames"];
+    
+    YYAnimatedImageView *imageView = [[YYAnimatedImageView alloc] init];
+    NSURL *originalImageURL = [NSURL URLWithString:kTestAPNGPURL];
+    
+    [imageView sd_setImageWithURL:originalImageURL
+                 placeholderImage:nil
+                          options:SDWebImagePreloadAllFrames
+                        completed:^(UIImage * _Nullable image, NSError * _Nullable error, SDImageCacheType cacheType, NSURL * _Nullable imageURL) {
+                            expect(image).toNot.beNil();
+                            expect(error).to.beNil();
+                            expect(originalImageURL).to.equal(imageURL);
+                            
+                            YYImage *animatedImage = (YYImage *)image;
+                            expect(animatedImage.class).to.equal(YYImage.class);
+                            expect(animatedImage.isAllFramesLoaded).beTruthy();
+                            [animatedImage unloadAllFrames];
+                            expect(animatedImage.isAllFramesLoaded).beFalsy();
                             [expectation fulfill];
                         }];
     [self waitForExpectationsWithCommonTimeout];
@@ -95,7 +119,7 @@ - (void)testSDImageYYCoderProgressiveJPEGWorks {
     [SDImageCodersManager.sharedManager addCoder:SDImageYYCoder.sharedCoder];
 
     NSURL *imageURL = [NSURL URLWithString:kTestProgressiveJPEGURL];
-    [[SDWebImageDownloader sharedDownloader] downloadImageWithURL:imageURL options:SDWebImageDownloaderProgressiveLoad progress:nil completed:^(UIImage * _Nullable image, NSData * _Nullable data, NSError * _Nullable error, BOOL finished) {
+    [[SDWebImageDownloader sharedDownloader] downloadImageWithURL:imageURL options:SDWebImageDownloaderProgressiveLoad context:@{SDWebImageContextAnimatedImageClass : YYImage.class} progress:nil completed:^(UIImage * _Nullable image, NSData * _Nullable data, NSError * _Nullable error, BOOL finished) {
         if (image && data && !error && finished) {
             [expectation fulfill];
         } else if (finished) {

SDWebImageYYPlugin/Classes/YYImage/YYImageBridge/YYImage+SDAdditions.m

@@ -6,6 +6,7 @@
  */
 
 #import "YYImage+SDAdditions.h"
+#import <objc/runtime.h>
 
 @implementation YYImage (SDAdditions)
 
@@ -22,10 +23,16 @@ - (instancetype)initWithAnimatedCoder:(id<SDAnimatedImageCoder>)animatedCoder sc
 
 - (void)preloadAllFrames {
     [self setPreloadAllAnimatedImageFrames:YES];
+    // YYImage contains bug that don't update ivar value, simple workaround
+    Ivar ivar = class_getInstanceVariable(self.class, "_preloadAllAnimatedImageFrames");
+    ((void (*)(id, Ivar, BOOL))object_setIvar)(self, ivar, YES);
 }
 
 - (void)unloadAllFrames {
     [self setPreloadAllAnimatedImageFrames:NO];
+    // YYImage contains bug that don't update ivar value, simple workaround
+    Ivar ivar = class_getInstanceVariable(self.class, "_preloadAllAnimatedImageFrames");
+    ((void (*)(id, Ivar, BOOL))object_setIvar)(self, ivar, NO);
 }
 
 - (BOOL)isAllFramesLoaded {