1 files changed, 166 insertions, 0 deletions
diff --git a/UnitTesting/GTMTestHTTPServer.m b/UnitTesting/GTMTestHTTPServer.m
new file mode 100644
index 0000000..7744b5e
--- /dev/null
+++ b/UnitTesting/GTMTestHTTPServer.m
@@ -0,0 +1,166 @@
+//
+//  GTMTestHTTPServer.m
+//
+//  Copyright 2007-2008 Google Inc.
+//
+//  Licensed under the Apache License, Version 2.0 (the "License"); you may not
+//  use this file except in compliance with the License.  You may obtain a copy
+//  of the License at
+// 
+//  http://www.apache.org/licenses/LICENSE-2.0
+// 
+//  Unless required by applicable law or agreed to in writing, software
+//  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+//  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+//  License for the specific language governing permissions and limitations under
+//  the License.
+//
+
+#import "GTMTestHTTPServer.h"
+#import "GTMHTTPServer.h"
+#import "GTMRegex.h"
+
+static NSArray *GetSubPatternsOfFirstStringMatchedByPattern(NSString *stringToSearch,
+                                                            NSString *pattern) {
+  GTMRegex *regex = [GTMRegex regexWithPattern:pattern];
+  NSString *firstMatch = [regex firstSubStringMatchedInString:stringToSearch];
+  NSArray *subPatterns = [regex subPatternsOfString:firstMatch];
+  return subPatterns;
+}
+
+@implementation GTMTestHTTPServer
+
+- (id)initWithDocRoot:(NSString *)docRoot {
+  self = [super init];
+  if (self) {
+    docRoot_ = [docRoot copy];
+    server_ = [[GTMHTTPServer alloc] initWithDelegate:self];
+    NSError *error = nil;
+    if ((docRoot == nil) || (![server_ start:&error])) {
+      _GTMDevLog(@"Failed to start up the webserver (docRoot='%@', error=%@)",
+                 docRoot_, error);
+      [self release];
+      return nil;
+    }
+  }
+  return self;
+}
+
+- (void)dealloc {
+  [docRoot_ release];
+  [server_ release];
+  [super dealloc];
+}
+
+- (uint16_t)port {
+  return [server_ port];
+}
+
+- (GTMHTTPResponseMessage *)httpServer:(GTMHTTPServer *)server
+                         handleRequest:(GTMHTTPRequestMessage *)request {
+  _GTMDevAssert(server == server_, @"how'd we get a different server?!");
+  UInt32 resultStatus = 0;
+  NSData *data = nil;
+  // clients should treat dates as opaque, generally
+  NSString *modifiedDate = @"thursday";
+  
+  NSString *postString = @"";
+  NSData *postData = [request body];
+  if ([postData length] > 0) {
+    postString = [[[NSString alloc] initWithData:postData
+                                        encoding:NSUTF8StringEncoding] autorelease];
+  }
+  
+  NSDictionary *allHeaders = [request allHeaderFieldValues];
+  NSString *ifModifiedSince = [allHeaders objectForKey:@"If-Modified-Since"];
+  NSString *authorization = [allHeaders objectForKey:@"Authorization"];
+  NSString *path = [[request URL] absoluteString];
+  
+  if ([path hasSuffix:@".auth"]) {
+    if (![authorization isEqualToString:@"GoogleLogin auth=GoodAuthToken"]) {
+      GTMHTTPResponseMessage *response =
+        [GTMHTTPResponseMessage emptyResponseWithCode:401];
+      return response;
+    } else {
+      path = [path substringToIndex:[path length] - 5];
+    }
+  }
+  
+  NSString *overrideHeader = [allHeaders objectForKey:@"X-HTTP-Method-Override"];
+  NSString *httpCommand = [request method];
+  if ([httpCommand isEqualToString:@"POST"] && 
+      [overrideHeader length] > 1) {
+    httpCommand = overrideHeader;
+  }
+  NSArray *searchResult = nil;
+  if ([path hasSuffix:@"/accounts/ClientLogin"]) {
+    // it's a sign-in attempt; it's good unless the password is "bad" or
+    // "captcha"
+    
+    // use regular expression to find the password
+    NSString *password = @"";
+    searchResult = GetSubPatternsOfFirstStringMatchedByPattern(path, @"Passwd=([^&\n]*)");
+    if ([searchResult count] == 2) {
+      password = [searchResult objectAtIndex:1];
+    }
+    
+    if ([password isEqualToString:@"bad"]) {
+      resultStatus = 403;
+    } else if ([password isEqualToString:@"captcha"]) {
+      NSString *loginToken = @"";
+      NSString *loginCaptcha = @"";
+      
+      searchResult = GetSubPatternsOfFirstStringMatchedByPattern(postString, @"logintoken=([^&\n]*)");
+      if ([searchResult count] == 2) {
+        loginToken = [searchResult objectAtIndex:1];
+      }
+      
+      searchResult = GetSubPatternsOfFirstStringMatchedByPattern(postString, @"logincaptcha=([^&\n]*)");
+      if ([searchResult count] == 2) {
+        loginCaptcha = [searchResult objectAtIndex:1];
+      }
+      
+      if ([loginToken isEqualToString:@"CapToken"] &&
+          [loginCaptcha isEqualToString:@"good"]) {
+        resultStatus = 200;
+      } else {
+        // incorrect captcha token or answer provided
+        resultStatus = 403;
+      }
+    } else {
+      // valid username/password
+      resultStatus = 200;
+    }
+  } else if ([httpCommand isEqualToString:@"DELETE"]) {
+    // it's an object delete; read and return empty data
+    resultStatus = 200;
+  } else {
+    // queries that have something like "?status=456" should fail with the
+    // status code
+    searchResult = GetSubPatternsOfFirstStringMatchedByPattern(path, @"status=([0-9]+)");
+    if ([searchResult count] == 2) {
+      resultStatus = [[searchResult objectAtIndex:1] intValue];
+    } else if ([ifModifiedSince isEqualToString:modifiedDate]) {
+      resultStatus = 304;
+    } else {
+      NSString *docPath = [docRoot_ stringByAppendingPathComponent:path];
+      data = [NSData dataWithContentsOfFile:docPath];
+      if (data) {
+        resultStatus = 200;
+      } else {
+        resultStatus = 404; 
+      }
+    }
+  }
+
+  GTMHTTPResponseMessage *response =
+    [GTMHTTPResponseMessage responseWithBody:data
+                                 contentType:@"text/plain"
+                                  statusCode:resultStatus];
+  [response setValue:modifiedDate forHeaderField:@"Last-Modified"];
+  [response setValue:[NSString stringWithFormat:@"TestCookie=%@", [path lastPathComponent]]
+      forHeaderField:@"Set-Cookie"];
+  return response;
+}
+
+@end