Adding support for unit tests using ARC.

DELTA=11  (9 added, 0 deleted, 2 changed)

1 files changed, 11 insertions, 2 deletions

diff --git a/UnitTesting/GTMIPhoneUnitTestMain.m b/UnitTesting/GTMIPhoneUnitTestMain.m
index b7af4ba..6bfc711 100644
--- a/UnitTesting/GTMIPhoneUnitTestMain.m
+++ b/UnitTesting/GTMIPhoneUnitTestMain.m
@@ -26,14 +26,23 @@
 // SenTestCase, outputs results and test run time, and terminates right
 // afterwards.
 int main(int argc, char *argv[]) {
+  int retVal;
+#if __has_feature(objc_arc)
+  @autoreleasepool {
+#else
   NSAutoreleasePool * pool = [[NSAutoreleasePool alloc] init];
+#endif
 #if GTM_IPHONE_USE_SENTEST
   // Is using SenTest, just create a dummy app that can be used as the
   // TEST_HOST.
-  int retVal = UIApplicationMain(argc, argv, nil, nil);
+  retVal = UIApplicationMain(argc, argv, nil, nil);
 #else
-  int retVal = UIApplicationMain(argc, argv, nil, @"GTMIPhoneUnitTestDelegate");
+  retVal = UIApplicationMain(argc, argv, nil, @"GTMIPhoneUnitTestDelegate");
 #endif
+#if __has_feature(objc_arc)
+  }
+#else
   [pool release];
+#endif
   return retVal;
 }