First draft of delay()/go()

luisobo · Sep 30, 2015 · e978c8a · e978c8a
1 parent 590eedb
commit e978c8a
Show file tree

Hide file tree

Showing 5 changed files with 59 additions and 1 deletion.
diff --git a/Nocilla/DSL/LSStubResponseDSL.h b/Nocilla/DSL/LSStubResponseDSL.h
@@ -8,6 +8,7 @@
 typedef LSStubResponseDSL *(^ResponseWithBodyMethod)(id<LSHTTPBody>);
 typedef LSStubResponseDSL *(^ResponseWithHeaderMethod)(NSString *, NSString *);
 typedef LSStubResponseDSL *(^ResponseWithHeadersMethod)(NSDictionary *);
+typedef LSStubResponseDSL *(^ResponseVoidMethod)();
 
 @interface LSStubResponseDSL : NSObject
 - (id)initWithResponse:(LSStubResponse *)response;
@@ -16,4 +17,7 @@ typedef LSStubResponseDSL *(^ResponseWithHeadersMethod)(NSDictionary *);
 @property (nonatomic, strong, readonly) ResponseWithHeadersMethod withHeaders;
 @property (nonatomic, strong, readonly) ResponseWithBodyMethod withBody;
 
+@property (nonatomic, strong, readonly) ResponseVoidMethod delay;
+@property (nonatomic, strong, readonly) ResponseVoidMethod go;
+
 @end
diff --git a/Nocilla/DSL/LSStubResponseDSL.m b/Nocilla/DSL/LSStubResponseDSL.m
@@ -37,4 +37,19 @@ - (ResponseWithBodyMethod)withBody {
         return self;
     };
 }
+
+- (ResponseVoidMethod)delay {
+    return ^{
+        [self.response delay];
+        return self;
+    };
+}
+
+- (ResponseVoidMethod)go {
+    return ^{
+        [self.response go];
+        return self;
+    };
+}
+
 @end
diff --git a/Nocilla/Hooks/NSURLRequest/LSHTTPStubURLProtocol.m b/Nocilla/Hooks/NSURLRequest/LSHTTPStubURLProtocol.m
@@ -31,6 +31,8 @@ - (void)startLoading {
     [cookieStorage setCookies:[NSHTTPCookie cookiesWithResponseHeaderFields:stubbedResponse.headers forURL:request.url]
                        forURL:request.URL mainDocumentURL:request.URL];
 
+    [stubbedResponse waitForGo];
+
     if (stubbedResponse.shouldFail) {
         [client URLProtocol:self didFailWithError:stubbedResponse.error];
     } else {

diff --git a/Nocilla/Stubs/LSStubResponse.h b/Nocilla/Stubs/LSStubResponse.h
@@ -15,4 +15,8 @@
 - (id)initWithRawResponse:(NSData *)rawResponseData;
 - (id)initDefaultResponse;
 - (void)setHeader:(NSString *)header value:(NSString *)value;
+
+- (void)delay;
+- (void)go;
+- (void)waitForGo;
 @end
diff --git a/Nocilla/Stubs/LSStubResponse.m b/Nocilla/Stubs/LSStubResponse.m
@@ -1,6 +1,8 @@
 #import "LSStubResponse.h"
 
-@interface LSStubResponse ()
+@interface LSStubResponse () {
+    NSCondition *_delayLock;
+}
 @property (nonatomic, assign, readwrite) NSInteger statusCode;
 @property (nonatomic, strong) NSMutableDictionary *mutableHeaders;
 @property (nonatomic, assign) UInt64 offset;
@@ -78,4 +80,35 @@ - (NSString *)description {
             self.mutableHeaders,
             self.body];
 }
+
+- (NSCondition*)delayLock {
+    @synchronized(self) {
+        return _delayLock;
+    }
+}
+
+- (void)delay {
+    @synchronized(self) {
+        if(!_delayLock)
+            _delayLock = [[NSCondition alloc] init];
+    }
+}
+
+- (void)go {
+    NSCondition *condition = self.delayLock;
+    @synchronized(self) {
+        _delayLock = nil;
+    }
+    [condition lock];
+    [condition broadcast];
+    [condition unlock];
+}
+
+- (void)waitForGo {
+    NSCondition *condition = self.delayLock;
+    [condition lock];
+    [condition wait];
+    [condition unlock];
+}
+
 @end