From caaa8d110ee38cbbb9bb1ab5fbbfb9b6f7eeb00c Mon Sep 17 00:00:00 2001
From: Corey Johnson <cj@github.com>
Date: Wed, 14 Mar 2012 12:19:50 -0700
Subject: [PATCH] :lipstick:

---
 Atom/src/Atom.mm | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/Atom/src/Atom.mm b/Atom/src/Atom.mm
index 6e447661c..0fecdd337 100755
--- a/Atom/src/Atom.mm
+++ b/Atom/src/Atom.mm
@@ -84,6 +84,13 @@
 
 }
 
+- (void)exitAfterSpecs {
+  [self modifyJavaScript:^(CefRefPtr<CefV8Context> context, CefRefPtr<CefV8Value> global) {
+    CefRefPtr<CefV8Value> atom = context->GetGlobal()->GetValue("atom");    
+    atom->SetValue("exitAfterSpecs", CefV8Value::CreateBool(YES), V8_PROPERTY_ATTRIBUTE_NONE);
+  }]; 
+}
+
 - (CefRefPtr<CefV8Context>)atomContext {
   return _clientHandler->GetBrowser()->GetMainFrame()->GetV8Context();
 }
@@ -116,20 +123,12 @@
 
 - (void)loadEnd {
   if ([[[NSProcessInfo processInfo] arguments] containsObject:@"--benchmark"]) {
-    [self modifyJavaScript:^(CefRefPtr<CefV8Context> context, CefRefPtr<CefV8Value> global) {
-      CefRefPtr<CefV8Value> atom = context->GetGlobal()->GetValue("atom");    
-      atom->SetValue("exitAfterSpecs", CefV8Value::CreateBool(YES), V8_PROPERTY_ATTRIBUTE_NONE);
-    }];
-    
+    [self exitAfterSpecs];    
     [self runBenchmarks:self];
   }
   
   if ([[[NSProcessInfo processInfo] arguments] containsObject:@"--test"]) {
-    [self modifyJavaScript:^(CefRefPtr<CefV8Context> context, CefRefPtr<CefV8Value> global) {
-      CefRefPtr<CefV8Value> atom = context->GetGlobal()->GetValue("atom");    
-      atom->SetValue("exitAfterSpecs", CefV8Value::CreateBool(YES), V8_PROPERTY_ATTRIBUTE_NONE);
-    }];
-    
+    [self exitAfterSpecs];
     [self runSpecs:self];
   }
 }