| OLD | NEW |
| (Empty) |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #import "ios/web/public/test/response_providers/delayed_response_provider.h" | |
| 6 | |
| 7 #import <Foundation/Foundation.h> | |
| 8 | |
| 9 #import "base/ios/weak_nsobject.h" | |
| 10 #import "base/mac/scoped_nsobject.h" | |
| 11 #import "base/mac/foundation_util.h" | |
| 12 #import "ios/third_party/gcdwebserver/src/GCDWebServer/Responses/GCDWebServerDat
aResponse.h" | |
| 13 | |
| 14 // A proxy class that will forward all messages to the |_response|. | |
| 15 // It will delay read operation |_delay| seconds. | |
| 16 @interface DelayedGCDWebServerResponse : NSProxy { | |
| 17 base::scoped_nsobject<GCDWebServerResponse> _response; | |
| 18 NSTimeInterval _delay; | |
| 19 } | |
| 20 | |
| 21 + (instancetype)responseWithServerResponse:(GCDWebServerResponse*)response | |
| 22 delay:(NSTimeInterval)delay; | |
| 23 - (instancetype)initWithServerResponse:(GCDWebServerResponse*)response | |
| 24 delay:(NSTimeInterval)delay; | |
| 25 @end | |
| 26 | |
| 27 @implementation DelayedGCDWebServerResponse | |
| 28 + (instancetype)responseWithServerResponse:(GCDWebServerResponse*)response | |
| 29 delay:(NSTimeInterval)delay { | |
| 30 return [[[DelayedGCDWebServerResponse alloc] initWithServerResponse:response | |
| 31 delay:delay] | |
| 32 autorelease]; | |
| 33 } | |
| 34 | |
| 35 - (instancetype)initWithServerResponse:(GCDWebServerResponse*)response | |
| 36 delay:(NSTimeInterval)delay { | |
| 37 _response.reset([response retain]); | |
| 38 _delay = delay; | |
| 39 return self; | |
| 40 } | |
| 41 | |
| 42 #pragma mark - GCDWebServerResponse | |
| 43 | |
| 44 - (void)performReadDataWithCompletion: | |
| 45 (GCDWebServerBodyReaderCompletionBlock)block { | |
| 46 [self asyncReadDataWithCompletion:block]; | |
| 47 } | |
| 48 | |
| 49 - (void)asyncReadDataWithCompletion: | |
| 50 (GCDWebServerBodyReaderCompletionBlock)block { | |
| 51 dispatch_async(dispatch_get_main_queue(), ^{ | |
| 52 base::WeakNSObject<GCDWebServerResponse> weakResponse(_response); | |
| 53 if ([_response | |
| 54 respondsToSelector:@selector(asyncReadDataWithCompletion:)]) { | |
| 55 dispatch_after(dispatch_time(DISPATCH_TIME_NOW, | |
| 56 static_cast<int64_t>(_delay * NSEC_PER_SEC)), | |
| 57 dispatch_get_main_queue(), ^{ | |
| 58 base::scoped_nsobject<GCDWebServerResponse> response( | |
| 59 [weakResponse retain]); | |
| 60 [response asyncReadDataWithCompletion:block]; | |
| 61 }); | |
| 62 } else { | |
| 63 dispatch_after(dispatch_time(DISPATCH_TIME_NOW, | |
| 64 static_cast<int64_t>(_delay * NSEC_PER_SEC)), | |
| 65 dispatch_get_main_queue(), ^{ | |
| 66 base::scoped_nsobject<GCDWebServerResponse> response( | |
| 67 [weakResponse retain]); | |
| 68 if (!response) { | |
| 69 return; | |
| 70 } | |
| 71 NSError* error = nil; | |
| 72 NSData* data = [response readData:&error]; | |
| 73 block(data, error); | |
| 74 }); | |
| 75 } | |
| 76 }); | |
| 77 } | |
| 78 | |
| 79 #pragma mark - NSProxy | |
| 80 | |
| 81 - (BOOL)conformsToProtocol:(Protocol*)protocol { | |
| 82 return [[_response class] conformsToProtocol:protocol]; | |
| 83 } | |
| 84 | |
| 85 - (BOOL)respondsToSelector:(SEL)selector { | |
| 86 return [[_response class] instancesRespondToSelector:selector]; | |
| 87 } | |
| 88 | |
| 89 - (NSMethodSignature*)methodSignatureForSelector:(SEL)selector { | |
| 90 return [[_response class] instanceMethodSignatureForSelector:selector]; | |
| 91 } | |
| 92 | |
| 93 - (void)forwardInvocation:(NSInvocation*)invocation { | |
| 94 SEL selector = [invocation selector]; | |
| 95 if ([_response respondsToSelector:selector]) | |
| 96 [invocation invokeWithTarget:_response]; | |
| 97 } | |
| 98 | |
| 99 @end | |
| 100 | |
| 101 namespace web { | |
| 102 | |
| 103 DelayedResponseProvider::DelayedResponseProvider( | |
| 104 std::unique_ptr<web::ResponseProvider> delayed_provider, | |
| 105 double delay) | |
| 106 : web::ResponseProvider(), | |
| 107 delayed_provider_(std::move(delayed_provider)), | |
| 108 delay_(delay) {} | |
| 109 | |
| 110 DelayedResponseProvider::~DelayedResponseProvider() {} | |
| 111 | |
| 112 bool DelayedResponseProvider::CanHandleRequest(const Request& request) { | |
| 113 return delayed_provider_->CanHandleRequest(request); | |
| 114 } | |
| 115 | |
| 116 GCDWebServerResponse* DelayedResponseProvider::GetGCDWebServerResponse( | |
| 117 const Request& request) { | |
| 118 DelayedGCDWebServerResponse* response = [DelayedGCDWebServerResponse | |
| 119 responseWithServerResponse:delayed_provider_->GetGCDWebServerResponse( | |
| 120 request) | |
| 121 delay:delay_]; | |
| 122 return base::mac::ObjCCastStrict<GCDWebServerResponse>(response); | |
| 123 } | |
| 124 | |
| 125 } // namespace web | |
| OLD | NEW |