CCPackageDownload unit tests for normal request and resuming request(rang request) addded.

Author: NickyWeber

UnitTests/CCPackageDownloadTests.m

@@ -7,8 +7,119 @@
 //
 
 #import <XCTest/XCTest.h>
+#import <Foundation/Foundation.h>
+#import "CCPackageDownload.h"
+#import "CCPackage.h"
+#import "CCPackageDownloadDelegate.h"
 
-@interface CCPackageDownloadTests : XCTestCase
+static NSUInteger __fileDownloadSize = 0;
+static BOOL __support_range_request = YES;
+
+@interface CCPackageDownloadTestURLProtocol : NSURLProtocol @end
+
+@implementation CCPackageDownloadTestURLProtocol
+
++ (BOOL)canInitWithRequest:(NSURLRequest*)theRequest
+{
+    return [theRequest.URL.scheme caseInsensitiveCompare:@"http"] == NSOrderedSame;
+}
+
++ (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)theRequest
+{
+    return theRequest;
+}
+
+- (NSUInteger)parseRangeHeaderValue:(NSString *)string
+{
+    if (!string)
+    {
+        return 0;
+    }
+
+    NSError *error;
+    NSRegularExpression *regex = [NSRegularExpression regularExpressionWithPattern:@"bytes=(\\d+)-"
+                                                          options:NSRegularExpressionCaseInsensitive
+                                                            error:&error];
+
+    NSTextCheckingResult *match = [regex firstMatchInString:string
+                                                    options:NSMatchingAnchored
+                                                      range:NSMakeRange(0, string.length)];
+
+    if (match.numberOfRanges == 2)
+    {
+        NSString *byteStr = [string substringWithRange:[match rangeAtIndex:1]];
+        return (NSUInteger) [byteStr integerValue];
+    }
+
+    return 0;
+}
+
+- (void)startLoading
+{
+    NSMutableDictionary *headers = [NSMutableDictionary dictionary];
+
+
+    NSFileManager *fileManager = [NSFileManager defaultManager];
+    NSString *fileName = [self.request.URL lastPathComponent];
+    NSString *pathToPackage = [[NSBundle mainBundle] pathForResource:[NSString stringWithFormat:@"Resources-shared/Packages/%@", fileName] ofType:nil];
+    NSDictionary *attribs = [fileManager attributesOfItemAtPath:pathToPackage error:nil];
+    NSUInteger fileSize = [attribs[NSFileSize] unsignedIntegerValue];
+
+    NSUInteger byteRangeStart = 0;
+    if (__support_range_request)
+    {
+        byteRangeStart = [self parseRangeHeaderValue:self.request.allHTTPHeaderFields[@"Range"]];
+        headers[@"Accept-Ranges"] = @"bytes";
+        headers[@"Content-Range"] = [NSString stringWithFormat:@"bytes %u-%u/%u", byteRangeStart, fileSize - 1, fileSize];
+    }
+
+    NSData *data = [[NSData dataWithContentsOfFile:pathToPackage] subdataWithRange:NSMakeRange(byteRangeStart, fileSize - byteRangeStart)];
+
+    __fileDownloadSize = fileSize;
+
+    NSHTTPURLResponse *response;
+    if (pathToPackage)
+    {
+        headers[@"Content-Length"] = [NSString stringWithFormat:@"%u", [data length]];
+        response = [[NSHTTPURLResponse alloc] initWithURL:self.request.URL
+                                                              statusCode:200
+                                                             HTTPVersion:@"HTTP/1.1"
+                                                            headerFields:headers];
+    }
+    else
+    {
+        response = [[NSHTTPURLResponse alloc] initWithURL:self.request.URL
+                                                                  statusCode:404
+                                                                 HTTPVersion:@"HTTP/1.1"
+                                                                headerFields:nil];
+    }
+
+    id<NSURLProtocolClient> client = [self client];
+    [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
+    [client URLProtocol:self didLoadData:data];
+    [client URLProtocolDidFinishLoading:self];
+}
+
+- (void)stopLoading
+{
+    // Nothing to do
+}
+
+@end
+
+
+#pragma mark - test
+
+
+@interface CCPackageDownloadTests : XCTestCase <CCPackageDownloadDelegate>
+
+@property (nonatomic, strong) CCPackageDownload *download;
+@property (nonatomic, strong) CCPackage *package;
+@property (nonatomic, copy) NSString *downloadPath;
+@property (nonatomic) BOOL downloadReturned;
+@property (nonatomic) BOOL downloadSuccessful;
+@property (nonatomic, strong) NSError *downloadError;
+@property (nonatomic, copy) NSURL *localURL;
 
 @end
 
@@ -17,18 +128,137 @@ @implementation CCPackageDownloadTests
 - (void)setUp
 {
     [super setUp];
-    // Put setup code here. This method is called before the invocation of each test method in the class.
+
+    [NSURLProtocol registerClass:[CCPackageDownloadTestURLProtocol class]];
+
+    self.downloadPath = [NSTemporaryDirectory() stringByAppendingPathComponent:@"Downloads"];
+
+    [self deleteDownloadFolder];
+    [self createDownloadFolder];
+
+    self.downloadReturned = NO;
+    self.downloadError = nil;
+    self.downloadSuccessful = NO;
+
+
+    self.package = [[CCPackage alloc] initWithName:@"testpackage"
+                                        resolution:@"phonehd"
+                                                os:@"iOS"
+                                         remoteURL:[NSURL URLWithString:@"http://package.request.fake/testpackage-iOS-phonehd.zip"]];
+
+    self.localURL = [[NSURL fileURLWithPath:_downloadPath] URLByAppendingPathComponent:@"testdownload.zip"];
+    self.download = [[CCPackageDownload alloc] initWithPackage:_package localURL:_localURL];
+    _download.delegate = self;
 }
 
 - (void)tearDown
 {
-    // Put teardown code here. This method is called after the invocation of each test method in the class.
+    [NSURLProtocol unregisterClass:[CCPackageDownloadTestURLProtocol class]];
+
     [super tearDown];
 }
 
-- (void)testExample
+- (void)deleteDownloadFolder
+{
+    NSFileManager *fileManager = [NSFileManager defaultManager];
+    NSError *error;
+    if (![fileManager removeItemAtPath:_downloadPath error:&error])
+    {
+        NSLog(@"%@",error);
+    }
+}
+
+- (void)createDownloadFolder
+{
+    NSError *error;
+    NSFileManager *fileManager = [NSFileManager defaultManager];
+    if (![fileManager createDirectoryAtURL:[NSURL fileURLWithPath:_downloadPath]
+               withIntermediateDirectories:YES
+                                attributes:nil
+                                     error:&error])
+    {
+        NSLog(@"%@", error);
+    }
+}
+
+
+#pragma mark - tests
+
+- (void)testDownloadPackage
+{
+    [self startDownloadAndWaitForDelegateToReturn];
+
+    NSFileManager *fileManager = [NSFileManager defaultManager];
+    NSDictionary *attribs = [fileManager attributesOfItemAtPath:_localURL.path error:nil];
+
+    XCTAssertTrue([fileManager fileExistsAtPath:_localURL.path]);
+    XCTAssertEqual([attribs[NSFileSize] unsignedIntegerValue], __fileDownloadSize);
+}
+
+- (void)startDownloadAndWaitForDelegateToReturn
+{
+    [_download start];
+
+    while (!_downloadReturned)
+    {
+        [[NSRunLoop currentRunLoop] runMode:NSDefaultRunLoopMode beforeDate:[NSDate distantFuture]];
+    }
+}
+
+- (void)testRangeRequest
+{
+    // Setting up partial download on disk
+    NSString *fileName = [_package.remoteURL lastPathComponent];
+    NSString *pathToPackage = [[NSBundle mainBundle] pathForResource:[NSString stringWithFormat:@"Resources-shared/Packages/%@", fileName] ofType:nil];
+    NSData *data = [[NSData dataWithContentsOfFile:pathToPackage] subdataWithRange:NSMakeRange(0, 5000)];
+    NSString *tempName = [_download performSelector:@selector(createTempName)];
+    [data writeToFile:[[_localURL.path stringByDeletingLastPathComponent] stringByAppendingPathComponent:tempName] atomically:YES];
+
+    [self startDownloadAndWaitForDelegateToReturn];
+
+    NSFileManager *fileManager = [NSFileManager defaultManager];
+    NSDictionary *attribs = [fileManager attributesOfItemAtPath:_localURL.path error:nil];
+    XCTAssertTrue([fileManager fileExistsAtPath:_localURL.path]);
+    XCTAssertEqual([attribs[NSFileSize] unsignedIntegerValue], __fileDownloadSize);
+}
+
+/*
+- (void)testResumeDownloadedPackage
+{
+
+}
+
+- (void)testPauseDownload
+{
+
+}
+
+
+- (void)testDownloadWith404Response
+{
+
+}
+*/
+
+
+#pragma mark - CCPackageDownloadDelegate
+
+- (void)downloadFinished:(CCPackageDownload *)download
+{
+    self.downloadReturned = YES;
+    self.downloadSuccessful = YES;
+}
+
+- (void)downloadFailed:(CCPackageDownload *)download error:(NSError *)error
+{
+    self.downloadReturned = YES;
+    self.downloadError = error;
+    self.downloadSuccessful = NO;
+}
+
+- (BOOL)shouldResumeDownload:(CCPackageDownload *)download
 {
-    XCTFail(@"No implementation for \"%s\"", __PRETTY_FUNCTION__);
+    return YES;
 }
 
 @end