Restrict deprecated parts of audio API usage to whitelisted apps

extensions/browser/api/audio/audio_apitest.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 <stddef.h>
 #include <stdint.h>
 
+#include <memory>
+
+#include "base/auto_reset.h"
+#include "base/command_line.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "build/build_config.h"
+#include "extensions/common/features/feature_session_type.h"
+#include "extensions/common/switches.h"
 #include "extensions/shell/test/shell_apitest.h"
 #include "extensions/test/extension_test_message_listener.h"
 #include "extensions/test/result_catcher.h"
 
 #if defined(OS_CHROMEOS)
 #include "chromeos/audio/audio_devices_pref_handler_stub.h"
 #include "chromeos/audio/cras_audio_handler.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/dbus/fake_cras_audio_client.h"
 #endif
@@ skipping 59 matching lines @@
                    // stable_device_id_v2:
                    version == 2 ? info.stable_id ^ 0xFFFF : 0, info.device_name,
                    info.type, info.name, false, 0);
 }
 
 class AudioApiTest : public ShellApiTest {
  public:
   AudioApiTest() : cras_audio_handler_(NULL), fake_cras_audio_client_(NULL) {}
   ~AudioApiTest() override {}
 
+  void SetUp() override {
+    session_feature_type_ = extensions::ScopedCurrentFeatureSessionType(
+        extensions::FeatureSessionType::KIOSK);
+
+    ShellApiTest::SetUp();
+  }
+
   void SetUpCrasAudioHandlerWithTestingNodes(const AudioNodeList& audio_nodes) {
     chromeos::DBusThreadManager* dbus_manager =
         chromeos::DBusThreadManager::Get();
     DCHECK(dbus_manager);
     fake_cras_audio_client_ = static_cast<chromeos::FakeCrasAudioClient*>(
         dbus_manager->GetCrasAudioClient());
     fake_cras_audio_client_->SetAudioNodesAndNotifyObserversForTesting(
         audio_nodes);
     cras_audio_handler_ = chromeos::CrasAudioHandler::Get();
     DCHECK(cras_audio_handler_);
     base::RunLoop().RunUntilIdle();
   }
 
   void ChangeAudioNodes(const AudioNodeList& audio_nodes) {
     DCHECK(fake_cras_audio_client_);
     fake_cras_audio_client_->SetAudioNodesAndNotifyObserversForTesting(
         audio_nodes);
     base::RunLoop().RunUntilIdle();
   }
 
  protected:
   base::MessageLoopForUI message_loop_;
+  std::unique_ptr<base::AutoReset<extensions::FeatureSessionType>>
+      session_feature_type_;
   chromeos::CrasAudioHandler* cras_audio_handler_;  // Not owned.
   chromeos::FakeCrasAudioClient* fake_cras_audio_client_;  // Not owned.
 };
 
 IN_PROC_BROWSER_TEST_F(AudioApiTest, Audio) {
   // Set up the audio nodes for testing.
   AudioNodeList audio_nodes = {
       CreateAudioNode(kJabraSpeaker1, 2), CreateAudioNode(kJabraSpeaker2, 2),
       CreateAudioNode(kHDMIOutput, 2),    CreateAudioNode(kJabraMic1, 2),
       CreateAudioNode(kJabraMic2, 2),     CreateAudioNode(kUSBCameraMic, 2)};
@@ skipping 140 matching lines @@
   audio_nodes.erase(audio_nodes.begin() + init_device_size - 1);
   ChangeAudioNodes(audio_nodes);
   cras_audio_handler_->GetAudioDevices(&audio_devices);
   EXPECT_EQ(init_device_size - 1, audio_devices.size());
 
   // Verify the background app got the onNodesChanged event
   // with the last node removed.
   EXPECT_TRUE(result_catcher.GetNextResult()) << result_catcher.message();
 }
 
+class WhitelistedAudioApiTest : public AudioApiTest {
+ public:
+  WhitelistedAudioApiTest() {}
+  ~WhitelistedAudioApiTest() override = default;
+
+  void SetUpCommandLine(base::CommandLine* command_line) override {
+    command_line->AppendSwitchASCII(
+        extensions::switches::kWhitelistedExtensionID,
+        "jlgnoeceollaejlkenecblnjmdcfhfgc");
+  }
+};
+
+IN_PROC_BROWSER_TEST_F(WhitelistedAudioApiTest, DeprecatedApi) {
+  // Set up the audio nodes for testing.
+  AudioNodeList audio_nodes = {
+      CreateAudioNode(kJabraSpeaker1, 2), CreateAudioNode(kJabraSpeaker2, 2),
+      CreateAudioNode(kHDMIOutput, 2),    CreateAudioNode(kJabraMic1, 2),
+      CreateAudioNode(kJabraMic2, 2),     CreateAudioNode(kUSBCameraMic, 2)};
+
+  SetUpCrasAudioHandlerWithTestingNodes(audio_nodes);
+
+  EXPECT_TRUE(RunAppTest("api_test/audio/deprecated_api")) << message_;
+}
+
 #endif  // OS_CHROMEOS
 
 }  // namespace extensions