Merge pull request #62 from hpique/master

Don't override Content-Length in initializer if already set
AliSoftware · May 18, 2014 · 9877c7f · 9877c7f
2 parents 233f747 + 2cdaf85
commit 9877c7f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/OHHTTPStubs/Sources/OHHTTPStubsResponse.m b/OHHTTPStubs/Sources/OHHTTPStubsResponse.m
@@ -114,7 +114,11 @@ -(instancetype)initWithInputStream:(NSInputStream*)inputStream
         self.dataSize = dataSize;
         self.statusCode = statusCode;
         NSMutableDictionary * headers = [NSMutableDictionary dictionaryWithDictionary:httpHeaders];
-        headers[@"Content-Length"] = [NSString stringWithFormat:@"%llu",self.dataSize];
+        static NSString *const ContentLengthHeader = @"Content-Length";
+        if (!headers[ContentLengthHeader])
+        {
+            headers[ContentLengthHeader] = [NSString stringWithFormat:@"%llu",self.dataSize];
+        }
         self.httpHeaders = [NSDictionary dictionaryWithDictionary:headers];
     }
     return self;