Allow audio debug recordings through a private API extension.

chrome/browser/extensions/api/webrtc_logging_private/webrtc_logging_private_apitest.cc

 // Copyright 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 "base/json/json_writer.h"
 #include "base/strings/string_split.h"
 #include "base/strings/stringprintf.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/extensions/api/webrtc_logging_private/webrtc_logging_private_api.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/extension_function_test_utils.h"
 #include "chrome/browser/extensions/extension_tab_util.h"
 #include "chrome/browser/media/webrtc_log_uploader.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
+#include "chrome/common/chrome_switches.h"
 #include "components/compression/compression_utils.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/test/test_utils.h"
 #include "extensions/common/test_util.h"
 
 using compression::GzipUncompress;
 using extensions::Extension;
 using extensions::WebrtcLoggingPrivateDiscardFunction;
 using extensions::WebrtcLoggingPrivateSetMetaDataFunction;
 using extensions::WebrtcLoggingPrivateStartFunction;
 using extensions::WebrtcLoggingPrivateStartRtpDumpFunction;
 using extensions::WebrtcLoggingPrivateStopFunction;
 using extensions::WebrtcLoggingPrivateStopRtpDumpFunction;
 using extensions::WebrtcLoggingPrivateStoreFunction;
 using extensions::WebrtcLoggingPrivateUploadFunction;
 using extensions::WebrtcLoggingPrivateUploadStoredFunction;
+using extensions::WebrtcLoggingPrivateStartAudioDebugRecordingsFunction;
+using extensions::WebrtcLoggingPrivateStopAudioDebugRecordingsFunction;
 
 namespace utils = extension_function_test_utils;
 
 namespace {
 
 static const char kTestLoggingSessionIdKey[] = "app_session_id";
 static const char kTestLoggingSessionIdValue[] = "0123456789abcdef";
 static const char kTestLoggingUrl[] = "dummy url string";
 
 std::string ParamsToString(const base::ListValue& parameters) {
@@ skipping 10 matching lines @@
   meta_data->Append(meta_data_entry);
   meta_data_entry = new base::DictionaryValue();
   meta_data_entry->SetString("key", "url");
   meta_data_entry->SetString("value", kTestLoggingUrl);
   meta_data->Append(meta_data_entry);
   parameters->Append(meta_data);
 }
 
 class WebrtcLoggingPrivateApiTest : public ExtensionApiTest {
  protected:
-
   void SetUp() override {
     ExtensionApiTest::SetUp();
     extension_ = extensions::test_util::CreateEmptyExtension();
   }
 
   template<typename T>
   scoped_refptr<T> CreateFunction() {
     scoped_refptr<T> function(new T());
     function->set_extension(extension_.get());
     function->set_has_callback(true);
@@ skipping 83 matching lines @@
     return RunFunction<WebrtcLoggingPrivateStoreFunction>(params, false);
   }
 
   bool UploadStoredLog(const std::string& log_id) {
     base::ListValue params;
     AppendTabIdAndUrl(&params);
     params.AppendString(log_id);
     return RunFunction<WebrtcLoggingPrivateUploadStoredFunction>(params, true);
   }
 
+  bool StartAudioDebugRecordings(int seconds) {
+    base::ListValue params;
+    AppendTabIdAndUrl(&params);
+    params.AppendInteger(seconds);
+    return RunFunction<WebrtcLoggingPrivateStartAudioDebugRecordingsFunction>(
+        params, true);
+  }
+
+  bool StopAudioDebugRecordings() {
+    base::ListValue params;
+    AppendTabIdAndUrl(&params);
+    return RunFunction<WebrtcLoggingPrivateStopAudioDebugRecordingsFunction>(
+        params, true);
+  }
+
  private:
   scoped_refptr<Extension> extension_;
 };
 
 // Helper class to temporarily tell the uploader to save the multipart buffer to
 // a test string instead of uploading.
 class ScopedOverrideUploadBuffer {
  public:
   ScopedOverrideUploadBuffer() {
     g_browser_process->webrtc_log_uploader()->
@@ skipping 225 matching lines @@
   ASSERT_TRUE(StoreLog(kLogId));
 
   ScopedOverrideUploadBuffer buffer_override;
   EXPECT_TRUE(UploadStoredLog(kLogId));
   EXPECT_NE(std::string::npos,
             buffer_override.multipart().find("filename=\"webrtc_log.gz\""));
   EXPECT_NE(std::string::npos,
             buffer_override.multipart().find(kTestLoggingUrl));
 }
 
+IN_PROC_BROWSER_TEST_F(WebrtcLoggingPrivateApiTest,
+                       TestStartStopAudioDebugRecordings) {
+  // TODO(guidou): These tests are missing verification of the actual AEC dump
+  // data. This will be fixed with a separate browser test.
+  // See crbug.com/569957.
+  base::CommandLine::ForCurrentProcess()->AppendSwitch(
+      switches::kEnableAudioDebugRecordingsFromExtension);
+  ASSERT_TRUE(StartAudioDebugRecordings(0));
+  ASSERT_TRUE(StopAudioDebugRecordings());
+}
+
+IN_PROC_BROWSER_TEST_F(WebrtcLoggingPrivateApiTest,
+                       TestStartTimedAudioDebugRecordings) {
+  base::CommandLine::ForCurrentProcess()->AppendSwitch(
+      switches::kEnableAudioDebugRecordingsFromExtension);
+  ASSERT_TRUE(StartAudioDebugRecordings(1));
+}