From 98bc29cdce1ea12cb21833e53da49e701d301333 Mon Sep 17 00:00:00 2001
From: Corey Johnson <cj@github.com>
Date: Mon, 12 Mar 2012 14:24:06 -0700
Subject: [PATCH] Benchmarks are run when atom is run with --benchmark argument

---
 Atom/src/Atom.mm           | 7 +++++++
 Atom/src/BrowserDelegate.h | 1 +
 Atom/src/client_handler.mm | 6 ++++++
 3 files changed, 14 insertions(+)

diff --git a/Atom/src/Atom.mm b/Atom/src/Atom.mm
index 8d95a62c2..d3b849e3d 100755
--- a/Atom/src/Atom.mm
+++ b/Atom/src/Atom.mm
@@ -113,6 +113,13 @@
   context->Exit();
 }
 
+- (void)loadEnd {
+  if ([[[NSProcessInfo processInfo] arguments] containsObject:@"--benchmark"]) {
+    NSLog(@"Running Benchmarks");
+    [self runBenchmarks:self];
+  }
+}
+
 @end
   
 // Returns the application settings based on command line arguments.
diff --git a/Atom/src/BrowserDelegate.h b/Atom/src/BrowserDelegate.h
index 6d94f27be..b81f812f4 100644
--- a/Atom/src/BrowserDelegate.h
+++ b/Atom/src/BrowserDelegate.h
@@ -6,6 +6,7 @@
 @optional
 - (void)afterCreated;
 - (void)loadStart;
+- (void)loadEnd;
 - (bool)keyEventOfType:(cef_handler_keyevent_type_t)type code:(int)code modifiers:(int)modifiers isSystemKey:(bool)isSystemKey isAfterJavaScript:(bool)isAfterJavaScript;
 
 @end
diff --git a/Atom/src/client_handler.mm b/Atom/src/client_handler.mm
index 215ffdbc3..82a118e96 100755
--- a/Atom/src/client_handler.mm
+++ b/Atom/src/client_handler.mm
@@ -82,6 +82,12 @@ void ClientHandler::OnLoadEnd(CefRefPtr<CefBrowser> browser,
                               int httpStatusCode)
 {
   REQUIRE_UI_THREAD();
+  
+  
+  if ([m_delegate respondsToSelector:@selector(loadEnd)]) {
+    [m_delegate loadEnd];
+  }
+
 }
 
 bool ClientHandler::OnLoadError(CefRefPtr<CefBrowser> browser,