Unregister Bluetooth profile when the extension that added it is unloaded.

chrome/browser/extensions/api/bluetooth/bluetooth_event_router_unittest.cc

 // Copyright (c) 2012 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 <string>
 
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
+#include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/extensions/api/bluetooth/bluetooth_event_router.h"
 #include "chrome/browser/extensions/event_names.h"
 #include "chrome/browser/extensions/extension_system_factory.h"
 #include "chrome/browser/extensions/test_extension_system.h"
 #include "chrome/common/extensions/api/bluetooth.h"
 #include "chrome/test/base/testing_profile.h"
+#include "content/public/browser/notification_service.h"
 #include "content/public/test/test_browser_thread.h"
 #include "device/bluetooth/test/mock_bluetooth_adapter.h"
 #include "device/bluetooth/test/mock_bluetooth_device.h"
 #include "device/bluetooth/test/mock_bluetooth_profile.h"
 #include "device/bluetooth/test/mock_bluetooth_socket.h"
 #include "extensions/browser/event_router.h"
+#include "extensions/common/extension_builder.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
+const char kTestExtensionId[] = "test extension id";
 const char kAudioProfileUuid[] = "audio profile uuid";
 const char kHealthProfileUuid[] = "health profile uuid";
 
 class FakeEventRouter : public extensions::EventRouter {
  public:
   explicit FakeEventRouter(Profile* profile) : EventRouter(profile, NULL) {}
 
   virtual void DispatchEventToExtension(
       const std::string& extension_id,
       scoped_ptr<extensions::Event> event) OVERRIDE {
@@ skipping 86 matching lines @@
   router_.OnListenerRemoved();
   router_.OnListenerRemoved();
   EXPECT_CALL(*mock_adapter_, RemoveObserver(testing::_)).Times(1);
   router_.OnListenerRemoved();
 }
 
 TEST_F(ExtensionBluetoothEventRouterTest, Profiles) {
   EXPECT_FALSE(router_.HasProfile(kAudioProfileUuid));
   EXPECT_FALSE(router_.HasProfile(kHealthProfileUuid));
 
-  router_.AddProfile(kAudioProfileUuid, &mock_audio_profile_);
-  router_.AddProfile(kHealthProfileUuid, &mock_health_profile_);
+  router_.AddProfile(
+      kAudioProfileUuid, kTestExtensionId, &mock_audio_profile_);
+  router_.AddProfile(
+      kHealthProfileUuid, kTestExtensionId, &mock_health_profile_);
   EXPECT_TRUE(router_.HasProfile(kAudioProfileUuid));
   EXPECT_TRUE(router_.HasProfile(kHealthProfileUuid));
 
   EXPECT_CALL(mock_audio_profile_, Unregister()).Times(1);
   router_.RemoveProfile(kAudioProfileUuid);
   EXPECT_FALSE(router_.HasProfile(kAudioProfileUuid));
   EXPECT_TRUE(router_.HasProfile(kHealthProfileUuid));
 
   // Make sure remaining profiles are unregistered in destructor.
   EXPECT_CALL(mock_health_profile_, Unregister()).Times(1);
   EXPECT_CALL(*mock_adapter_, RemoveObserver(testing::_)).Times(1);
 }
 
+TEST_F(ExtensionBluetoothEventRouterTest, UnloadExtension) {
+  scoped_refptr<const extensions::Extension> extension =
+      extensions::ExtensionBuilder()
+          .SetManifest(extensions::DictionaryBuilder()
+              .Set("name", "BT event router test")
+              .Set("version", "1.0")
+              .Set("manifest_version", 2))
+          .SetID(kTestExtensionId)
+          .Build();
+
+  router_.AddProfile(
+      kAudioProfileUuid, kTestExtensionId, &mock_audio_profile_);
+  router_.AddProfile(
+      kHealthProfileUuid, kTestExtensionId, &mock_health_profile_);
+  EXPECT_TRUE(router_.HasProfile(kAudioProfileUuid));
+  EXPECT_TRUE(router_.HasProfile(kHealthProfileUuid));
+
+  // Unloading the extension should unregister all profiles added by it.
+  EXPECT_CALL(mock_audio_profile_, Unregister()).Times(1);
+  EXPECT_CALL(mock_health_profile_, Unregister()).Times(1);
+
+  content::NotificationService* notifier =
+      content::NotificationService::current();
+  UnloadedExtensionInfo details(
+      extension, UnloadedExtensionInfo::REASON_DISABLE);
+  notifier->Notify(chrome::NOTIFICATION_EXTENSION_UNLOADED,
+                   content::Source<Profile>(test_profile_.get()),
+                   content::Details<UnloadedExtensionInfo>(&details));
+
+  EXPECT_FALSE(router_.HasProfile(kAudioProfileUuid));
+  EXPECT_FALSE(router_.HasProfile(kHealthProfileUuid));
+  EXPECT_CALL(*mock_adapter_, RemoveObserver(testing::_)).Times(1);
+}
+
 TEST_F(ExtensionBluetoothEventRouterTest, DispatchConnectionEvent) {
-  router_.AddProfile(kAudioProfileUuid, &mock_audio_profile_);
+  router_.AddProfile(
+      kAudioProfileUuid, kTestExtensionId, &mock_audio_profile_);
 
   FakeExtensionSystem* fake_extension_system =
       static_cast<FakeExtensionSystem*>(ExtensionSystemFactory::GetInstance()->
           SetTestingFactoryAndUse(test_profile_.get(),
                                   &BuildFakeExtensionSystem));
 
-  const char test_extension_id[] = "test extension id";
   testing::NiceMock<device::MockBluetoothDevice> mock_device(
       mock_adapter_, 0, "device name", "device address", true, false);
   scoped_refptr<testing::NiceMock<device::MockBluetoothSocket> > mock_socket(
       new testing::NiceMock<device::MockBluetoothSocket>());
 
-  router_.DispatchConnectionEvent(test_extension_id,
+  router_.DispatchConnectionEvent(kTestExtensionId,
                                   kAudioProfileUuid,
                                   &mock_device,
                                   mock_socket);
 
   FakeEventRouter* fake_event_router =
       static_cast<FakeEventRouter*>(fake_extension_system->event_router());
 
-  EXPECT_STREQ(test_extension_id, fake_event_router->extension_id().c_str());
+  EXPECT_STREQ(kTestExtensionId, fake_event_router->extension_id().c_str());
   EXPECT_STREQ(bluetooth::OnConnection::kEventName,
                fake_event_router->event()->event_name.c_str());
 
   base::ListValue* event_args = fake_event_router->event()->event_args.get();
   base::DictionaryValue* socket_value = NULL;
   ASSERT_TRUE(event_args->GetDictionary(0, &socket_value));
   int socket_id;
   ASSERT_TRUE(socket_value->GetInteger("id", &socket_id));
   EXPECT_EQ(mock_socket.get(), router_.GetSocket(socket_id).get());
 
@@ skipping 23 matching lines @@
                                   &mock_device,
                                   mock_socket);
   FakeEventRouter* fake_event_router =
       static_cast<FakeEventRouter*>(fake_extension_system->event_router());
   EXPECT_TRUE(fake_event_router->event() == NULL);
 
   EXPECT_CALL(*mock_adapter_, RemoveObserver(testing::_)).Times(1);
 }
 
 }  // namespace extensions