authenticator-ios-v2/unit-tests/FRAMockURLProtocol.m

b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni/*
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni * The contents of this file are subject to the terms of the Common Development and
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni * Distribution License (the License). You may not use this file except in compliance with the
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni * License.
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni *
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni * You can obtain a copy of the License at legal/CDDLv1.0.txt. See the License for the
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni * specific language governing permission and limitations under the License.
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni *
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni * When distributing Covered Software, include this CDDL Header Notice in each file and include
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni * the License file at legal/CDDLv1.0.txt. If applicable, add the following below the CDDL
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni * Header, with the fields enclosed by brackets [] replaced by your own identifying
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni * information: "Portions copyright [year] [name of copyright owner]".
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni *
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni * Copyright 2016 ForgeRock AS.
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni */
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni#import "FRAMockURLProtocol.h"
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantonistatic NSURLRequest *initialRequest;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantonistatic NSData *mockResponseData = nil;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantonistatic NSDictionary *mockResponseHeaders = nil;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantonistatic NSInteger mockStatusCode = 200;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantonistatic NSError *mockError = nil;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni@implementation FRAMockURLProtocol
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni+ (BOOL)canInitWithRequest:(NSURLRequest *)request {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    if(request != initialRequest) {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni        initialRequest = request;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    }
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    return YES;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni}
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni+ (NSURLRequest *)getRequest {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    return initialRequest;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni}
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni+ (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    return request;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni}
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni+ (void)setResponseData:(NSData*)data {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    if(data != mockResponseData) {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni        mockResponseData = data;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    }
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni}
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni+ (void)setResponseHeaders:(NSDictionary*)headers {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    if(headers != mockResponseHeaders) {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni        mockResponseHeaders = headers;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    }
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni}
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni+ (void)setStatusCode:(NSInteger)statusCode {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    mockStatusCode = statusCode;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni}
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni+ (void)setError:(NSError*)error {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    if(error != mockError) {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni        mockError = error;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    }
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni}
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni- (NSCachedURLResponse *)cachedResponse {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    return nil;
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni}
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni- (void)startLoading {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    NSURLRequest *request = [self request];
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    id<NSURLProtocolClient> client = [self client];
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    NSHTTPURLResponse *response = [[NSHTTPURLResponse alloc] initWithURL:[request URL]
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni                                                              statusCode:mockStatusCode
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni                                                             HTTPVersion:@"1.1"
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni                                                            headerFields:mockResponseHeaders];
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    [client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    [client URLProtocol:self didLoadData:mockResponseData];
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni    [client URLProtocolDidFinishLoading:self];
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni}
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni- (void)stopLoading {
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni}
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni
b3a21de0a0e0a5dea71bece7e5c0356700136fbcDiego Colantoni@end