Add unit test for WebRTCInternalsMessageHandler.

content/browser/media/webrtc/webrtc_internals_message_handler.cc

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/browser/media/webrtc/webrtc_internals_message_handler.h"
 
 #include "content/browser/media/webrtc/webrtc_internals.h"
 #include "content/common/media/peer_connection_tracker_messages.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_ui.h"
 #include "content/public/common/url_constants.h"
 
 namespace content {
 
-WebRTCInternalsMessageHandler::WebRTCInternalsMessageHandler() {
-  WebRTCInternals::GetInstance()->AddObserver(this);
+WebRTCInternalsMessageHandler::WebRTCInternalsMessageHandler()
+    : WebRTCInternalsMessageHandler(WebRTCInternals::GetInstance()) {}
+
+WebRTCInternalsMessageHandler::WebRTCInternalsMessageHandler(
+    WebRTCInternals* webrtc_internals)
+    : webrtc_internals_(webrtc_internals) {
+  webrtc_internals_->AddObserver(this);
 }
 
 WebRTCInternalsMessageHandler::~WebRTCInternalsMessageHandler() {
-  WebRTCInternals::GetInstance()->RemoveObserver(this);
+  webrtc_internals_->RemoveObserver(this);
 }
 
 void WebRTCInternalsMessageHandler::RegisterMessages() {
   web_ui()->RegisterMessageCallback("getAllStats",
       base::Bind(&WebRTCInternalsMessageHandler::OnGetAllStats,
                  base::Unretained(this)));
 
   web_ui()->RegisterMessageCallback("enableAudioDebugRecordings",
       base::Bind(
           &WebRTCInternalsMessageHandler::OnSetAudioDebugRecordingsEnabled,
@@ skipping 43 matching lines @@
   for (RenderProcessHost::iterator i(
        content::RenderProcessHost::AllHostsIterator());
        !i.IsAtEnd(); i.Advance()) {
     i.GetCurrentValue()->Send(new PeerConnectionTracker_GetAllStats());
   }
 }
 
 void WebRTCInternalsMessageHandler::OnSetAudioDebugRecordingsEnabled(
     bool enable, const base::ListValue* /* unused_list */) {
   if (enable) {
-    WebRTCInternals::GetInstance()->EnableAudioDebugRecordings(
-        web_ui()->GetWebContents());
+    webrtc_internals_->EnableAudioDebugRecordings(web_ui()->GetWebContents());
   } else {
-    WebRTCInternals::GetInstance()->DisableAudioDebugRecordings();
+    webrtc_internals_->DisableAudioDebugRecordings();
   }
 }
 
 void WebRTCInternalsMessageHandler::OnSetEventLogRecordingsEnabled(
     bool enable,
     const base::ListValue* /* unused_list */) {
   if (enable) {
-    WebRTCInternals::GetInstance()->EnableEventLogRecordings(
-        web_ui()->GetWebContents());
+    webrtc_internals_->EnableEventLogRecordings(web_ui()->GetWebContents());
   } else {
-    WebRTCInternals::GetInstance()->DisableEventLogRecordings();
+    webrtc_internals_->DisableEventLogRecordings();
   }
 }
 
 void WebRTCInternalsMessageHandler::OnDOMLoadDone(
     const base::ListValue* /* unused_list */) {
-  WebRTCInternals::GetInstance()->UpdateObserver(this);
+  webrtc_internals_->UpdateObserver(this);
 
-  if (WebRTCInternals::GetInstance()->IsAudioDebugRecordingsEnabled()) {
+  if (webrtc_internals_->IsAudioDebugRecordingsEnabled()) {
     RenderFrameHost* host = GetWebRTCInternalsHost();
     if (!host)
       return;
 
     std::vector<const base::Value*> args_vector;
     base::string16 script =
         WebUI::GetJavascriptCall("setAudioDebugRecordingsEnabled", args_vector);
     host->ExecuteJavaScript(script);
   }
 }
 
 void WebRTCInternalsMessageHandler::OnUpdate(const std::string& command,
                                              const base::Value* args) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 
   RenderFrameHost* host = GetWebRTCInternalsHost();
   if (!host)
     return;
 
   std::vector<const base::Value*> args_vector;
   if (args)
     args_vector.push_back(args);
 
   base::string16 update = WebUI::GetJavascriptCall(command, args_vector);
   host->ExecuteJavaScript(update);
 }
 
 }  // namespace content