Nocilla support for NSURLSessionDataTask about nocilla HOT 9 CLOSED

#import <XCTest/XCTest.h>

#define TestNeedsToWaitForBlock() __block BOOL blockFinished = NO
#define BlockFinished() blockFinished = YES
#define WaitForBlock() while (CFRunLoopRunInMode(kCFRunLoopDefaultMode, 0, true) && !blockFinished)

@interface NSURLProtocolImpl : NSURLProtocol
@end

@implementation NSURLProtocolImpl

+ (BOOL)canInitWithRequest:(NSURLRequest *)request
{
    NSLog(@"NSURLProtocolImpl canInitWithRequest: %@", request);
    return NO;
}

@end


@interface NSUrlProtocolTests : XCTestCase

@end

@implementation NSUrlProtocolTests

- (void)setUp
{
    [super setUp];
    [NSURLProtocol registerClass:[NSURLProtocolImpl class]];
}

- (void)tearDown
{
    // Put teardown code here. This method is called after the invocation of each test method in the class.
    [super tearDown];
}

- (void)testExample
{

    TestNeedsToWaitForBlock();

    NSURL *url = [NSURL URLWithString:@"http://www.example.com"];

    NSURLSession *urlsession = [NSURLSession sessionWithConfiguration:[NSURLSessionConfiguration defaultSessionConfiguration]];
    NSURLSessionDataTask *datatask = [urlsession dataTaskWithURL:url completionHandler:^(NSData *data, NSURLResponse *response, NSError *error) {
        if(error) {
            NSLog(@"error: %@", error);
        }
        else {
            NSLog(@"Success");
        }
        BlockFinished();
    }];
    [datatask resume];

    WaitForBlock();
}

@end

...
NSURLSessionConfiguration    configuration = [NSURLSessionConfiguration defaultSessionConfiguration];

configuration.protocolClasses=@[[LSHTTPStubURLProtocol class]];
...

@interface HttpSessionManager : AFHTTPSessionManager
@end

@implementation HttpSessionManager
    -(instancetype)init{
         //this retrieval of base url is specific to my implementation the main thing is you call the super constructor that you're wrapping in the category below
         NSString *baseUrl = [[Properties instance]config][@"baseUrl"];
         self = [super initWithBaseURL:[NSURL URLWithString:baseUrl]];
         if(!self){
            return nil;
         }
         return self;
    }
@end

@interface  HttpSessionManager (AddTestProtocols)
@end

@implementation  HttpSessionManager (AddTestProtocols)
- (instancetype)wrap_initWithBaseURL:(NSURL *)url  sessionConfiguration:(NSURLSessionConfiguration *)configuration{
    if (!configuration) {
        configuration = [NSURLSessionConfiguration defaultSessionConfiguration];
    }
    //set the protocol class here!
    configuration.protocolClasses=@[[LSHTTPStubURLProtocol class]];

    return [self wrap_initWithBaseURL:url sessionConfiguration:configuration];
}

//swizzle the initWithBaseURL and wrap_initWithBaseURL 
+ (void)load
{
    Method original, swizzled;

    SEL originalSelector=@selector(initWithBaseURL:sessionConfiguration:);
    SEL swizzledSelector=@selector(wrap_initWithBaseURL:sessionConfiguration:);
    original = class_getInstanceMethod(self, originalSelector);
    swizzled = class_getInstanceMethod(self, swizzledSelector);
    method_exchangeImplementations(original, swizzled);
}
@end

+ (AFHTTPSessionManager *)api
{
    return [Network sharedInstance].apiSession;
}

// Used with testing
- (void)hijackSessionWithProtocolClasses:(NSArray *)protocolClasses
{
    NSURLSessionConfiguration *hijackedConfig = self.apiSession.session.configuration;
    hijackedConfig.protocolClasses = protocolClasses;
    [self.apiSession invalidateSessionCancelingTasks:YES];
    self.apiSession = [[AFHTTPSessionManager alloc] initWithBaseURL:self.apiURL sessionConfiguration:hijackedConfig];
}

- (void)setUp
{
    [[Network sharedInstance] hijackSessionWithProtocolClasses:@[[LSHTTPStubURLProtocol class]]];
}

describe(@"MyClass", ^{
    beforeAll(^{
        [[LSNocilla sharedInstance] start];
    });
    afterEach(^{
        [[LSNocilla sharedInstance] clearStubs];
    });
    afterAll(^{
        [[LSNocilla sharedInstance] stop];
    });
    it(@"does the thing", ^AsyncBlock{
        [MyClass makeThatAPICallWithCompletion:^(NSURLSessionDataTask *task, id data, NSError *error) {
            NSInteger statusCode = [(NSHTTPURLResponse *)task.response statusCode];
            expect(data).willNot.beNil();
            expect(statusCode).will.equal(200);
            done();
        }];
    });
});

[[LSNocilla sharedInstance] registerHook:[[LSASIHTTPRequestHook alloc] init]];