device/bluetooth: Implement Unregister() of BlueotoothAudioSinkChromeOS and disconnection-related c…

device/bluetooth/bluetooth_audio_sink_chromeos_unittest.cc

 // Copyright 2015 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 <vector>
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/memory/ref_counted.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 63 matching lines @@
     properties_.uuid = BluetoothMediaClient::kBluetoothAudioSinkUUID;
     properties_.codec = FakeBluetoothMediaTransportClient::kTransportCodec;
     properties_.configuration =
         FakeBluetoothMediaTransportClient::kTransportConfiguration;
     properties_.state = BluetoothMediaTransportClient::kStateIdle;
     properties_.delay.reset(
         new uint16_t(FakeBluetoothMediaTransportClient::kTransportDelay));
     properties_.volume.reset(
         new uint16_t(FakeBluetoothMediaTransportClient::kTransportVolume));
 
+    media_ = static_cast<FakeBluetoothMediaClient*>(
+        DBusThreadManager::Get()->GetBluetoothMediaClient());
+    transport_ = static_cast<FakeBluetoothMediaTransportClient*>(
+        DBusThreadManager::Get()->GetBluetoothMediaTransportClient());
+
     GetAdapter();
   }
 
   void TearDown() override {
     callback_count_ = 0;
     error_callback_count_ = 0;
+    observer_.reset();
+
+    media_->SetVisible(true);
 
     // The adapter should outlive audio sink.
     audio_sink_ = nullptr;
     adapter_ = nullptr;
     DBusThreadManager::Shutdown();
   }
 
   // Gets the existing Bluetooth adapter.
   void GetAdapter() {
     BluetoothAdapterFactory::GetAdapter(
@@ skipping 31 matching lines @@
               std::vector<uint8_t>({0x3f, 0xff, 0x12, 0x35}));
 
     // Registers |audio_sink_| with valid codec and capabilities
     adapter_->RegisterAudioSink(
         options,
         base::Bind(&BluetoothAudioSinkChromeOSTest::RegisterCallback,
                    base::Unretained(this)),
         base::Bind(&BluetoothAudioSinkChromeOSTest::RegisterErrorCallback,
                    base::Unretained(this)));
 
+    observer_.reset(new TestAudioSinkObserver(audio_sink_));
     EXPECT_EQ(callback_count_, 1);
     EXPECT_EQ(error_callback_count_, 0);
+    EXPECT_EQ(observer_->state_changed_count_, 0);
+    EXPECT_EQ(observer_->volume_changed_count_, 0);
+  }
+
+  void GetFakeMediaEndpoint() {
+    BluetoothAudioSinkChromeOS* audio_sink_chromeos =
+        static_cast<BluetoothAudioSinkChromeOS*>(audio_sink_.get());
+    ASSERT_NE(audio_sink_chromeos, nullptr);
+
+    media_endpoint_ = static_cast<FakeBluetoothMediaEndpointServiceProvider*>(
+        audio_sink_chromeos->GetEndpointServiceProvider());
   }
 
   // Called whenever RegisterAudioSink is completed successfully.
   void RegisterCallback(
       scoped_refptr<BluetoothAudioSink> audio_sink) {
     ++callback_count_;
     audio_sink_ = audio_sink;
+
+    GetFakeMediaEndpoint();
+    ASSERT_NE(media_endpoint_, nullptr);
+    media_->SetEndpointRegistered(media_endpoint_, true);
+
     ASSERT_NE(audio_sink_.get(), nullptr);
     ASSERT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
   }
 
   // Called whenever RegisterAudioSink failed.
   void RegisterErrorCallback(BluetoothAudioSink::ErrorCode error_code) {
     ++error_callback_count_;
     EXPECT_EQ(error_code, BluetoothAudioSink::ERROR_NOT_REGISTERED);
   }
 
   // Called whenever there capabilities are returned from SelectConfiguration.
   void SelectConfigurationCallback(const std::vector<uint8_t>& capabilities) {
     ++callback_count_;
 
     // |capabilities| should be the same as the capabilities for registering an
     // audio sink in GetAudioSink().
     EXPECT_EQ(capabilities, std::vector<uint8_t>({0x3f, 0xff, 0x12, 0x35}));
   }
 
+  void UnregisterErrorCallback(BluetoothAudioSink::ErrorCode error_code) {
+    ++error_callback_count_;
+    EXPECT_EQ(error_code, BluetoothAudioSink::ERROR_NOT_UNREGISTERED);
+  }
+
   // Generic callbacks.
   void Callback() {
     ++callback_count_;
   }
 
   void ErrorCallback() {
     ++error_callback_count_;
   }
 
  protected:
   int callback_count_;
   int error_callback_count_;
 
   base::MessageLoop message_loop_;
 
+  FakeBluetoothMediaClient* media_;
+  FakeBluetoothMediaTransportClient* transport_;
+  FakeBluetoothMediaEndpointServiceProvider* media_endpoint_;
+  scoped_ptr<TestAudioSinkObserver> observer_;
   scoped_refptr<BluetoothAdapter> adapter_;
   scoped_refptr<BluetoothAudioSink> audio_sink_;
 
   // The default property set used while calling SetConfiguration on a media
   // endpoint object.
   BluetoothMediaEndpointServiceProvider::Delegate::TransportProperties
       properties_;
 };
 
 TEST_F(BluetoothAudioSinkChromeOSTest, RegisterSucceeded) {
   GetAudioSink();
-
-  // Adds an observer for |audio_sink_|.
-  TestAudioSinkObserver observer(audio_sink_);
-  EXPECT_EQ(observer.state_changed_count_, 0);
-  EXPECT_EQ(observer.volume_changed_count_, 0);
 }
 
 TEST_F(BluetoothAudioSinkChromeOSTest, RegisterFailedWithInvalidOptions) {
   // Sets options with an invalid codec and valid capabilities.
   BluetoothAudioSink::Options options;
   options.codec = 0xff;
   options.capabilities = std::vector<uint8_t>({0x3f, 0xff, 0x12, 0x35});
 
   adapter_->RegisterAudioSink(
       options,
@@ skipping 15 matching lines @@
       base::Bind(&BluetoothAudioSinkChromeOSTest::RegisterErrorCallback,
                  base::Unretained(this)));
 
   EXPECT_EQ(callback_count_, 0);
   EXPECT_EQ(error_callback_count_, 2);
 }
 
 TEST_F(BluetoothAudioSinkChromeOSTest, SelectConfiguration) {
   GetAudioSink();
 
-  // Adds an observer for |audio_sink_|.
-  TestAudioSinkObserver observer(audio_sink_);
-  EXPECT_EQ(observer.state_changed_count_, 0);
-  EXPECT_EQ(observer.volume_changed_count_, 0);
-
   // Simulates calling SelectConfiguration on the media endpoint object owned by
   // |audio_sink_| with some fake capabilities.
-  BluetoothAudioSinkChromeOS* audio_sink_chromeos =
-      static_cast<BluetoothAudioSinkChromeOS*>(audio_sink_.get());
-  FakeBluetoothMediaEndpointServiceProvider* media_endpoint =
-      static_cast<FakeBluetoothMediaEndpointServiceProvider*>(
-          audio_sink_chromeos->GetEndpointServiceProvider());
-  ASSERT_NE(media_endpoint, nullptr);
-
-  media_endpoint->SelectConfiguration(
+  media_endpoint_->SelectConfiguration(
       std::vector<uint8_t>({0x21, 0x15, 0x33, 0x2C}),
       base::Bind(&BluetoothAudioSinkChromeOSTest::SelectConfigurationCallback,
                  base::Unretained(this)));
 
   EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
   EXPECT_EQ(callback_count_, 2);
   EXPECT_EQ(error_callback_count_, 0);
-  EXPECT_EQ(observer.state_changed_count_, 0);
-  EXPECT_EQ(observer.volume_changed_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 0);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
 }
 
 TEST_F(BluetoothAudioSinkChromeOSTest, SetConfiguration) {
   GetAudioSink();
 
-  // Adds an observer for |audio_sink_|.
-  TestAudioSinkObserver observer(audio_sink_);
-  EXPECT_EQ(observer.state_changed_count_, 0);
-  EXPECT_EQ(observer.volume_changed_count_, 0);
+  media_endpoint_->SelectConfiguration(
+      std::vector<uint8_t>({0x21, 0x15, 0x33, 0x2C}),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::SelectConfigurationCallback,
+                 base::Unretained(this)));
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 0);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
 
   // Simulates calling SetConfiguration on the media endpoint object owned by
   // |audio_sink_| with a fake transport path and a
   // Delegate::TransportProperties structure.
-  BluetoothAudioSinkChromeOS* audio_sink_chromeos =
-      static_cast<BluetoothAudioSinkChromeOS*>(audio_sink_.get());
-  FakeBluetoothMediaEndpointServiceProvider* media_endpoint =
-      static_cast<FakeBluetoothMediaEndpointServiceProvider*>(
-          audio_sink_chromeos->GetEndpointServiceProvider());
-  ASSERT_NE(media_endpoint, nullptr);
-
-  media_endpoint->SetConfiguration(
-      ObjectPath(FakeBluetoothMediaTransportClient::kTransportPath),
-      properties_);
-
-  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_IDLE);
-  EXPECT_EQ(callback_count_, 1);
-  EXPECT_EQ(error_callback_count_, 0);
-  EXPECT_EQ(observer.state_changed_count_, 1);
-  EXPECT_EQ(observer.volume_changed_count_, 1);
+  media_endpoint_->SetConfiguration(
+      transport_->GetTransportPath(media_endpoint_->object_path()),
+      properties_);
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_IDLE);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 1);
+  EXPECT_EQ(observer_->volume_changed_count_, 1);
 }
 
 TEST_F(BluetoothAudioSinkChromeOSTest, SetConfigurationWithUnexpectedState) {
   GetAudioSink();
 
-  // Adds an observer for |audio_sink_|.
-  TestAudioSinkObserver observer(audio_sink_);
-  EXPECT_EQ(observer.state_changed_count_, 0);
-  EXPECT_EQ(observer.volume_changed_count_, 0);
-
-  // Simulates calling SetConfiguration on the media endpoint object owned by
-  // |audio_sink_| with a fake transport path and a
-  // Delegate::TransportProperties structure.
-  BluetoothAudioSinkChromeOS* audio_sink_chromeos =
-      static_cast<BluetoothAudioSinkChromeOS*>(audio_sink_.get());
-  FakeBluetoothMediaEndpointServiceProvider* media_endpoint =
-      static_cast<FakeBluetoothMediaEndpointServiceProvider*>(
-          audio_sink_chromeos->GetEndpointServiceProvider());
-  ASSERT_NE(media_endpoint, nullptr);
+  media_endpoint_->SelectConfiguration(
+      std::vector<uint8_t>({0x21, 0x15, 0x33, 0x2C}),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::SelectConfigurationCallback,
+                 base::Unretained(this)));
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 0);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
 
   // Set state of Delegate::TransportProperties with an unexpected value.
-  properties_.state = "active";
-
-  media_endpoint->SetConfiguration(
-      ObjectPath(FakeBluetoothMediaTransportClient::kTransportPath),
-      properties_);
-
-  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
-  EXPECT_EQ(callback_count_, 1);
-  EXPECT_EQ(error_callback_count_, 0);
-  EXPECT_EQ(observer.state_changed_count_, 0);
-  EXPECT_EQ(observer.volume_changed_count_, 0);
-}
-
-// Checks if the observer is notified while the media object is
-// removed(invisible). Once the media object is removed, the audio sink is no
-// longer valid.
-TEST_F(BluetoothAudioSinkChromeOSTest, ObserverNotifiedWhenMediaRemoved) {
-  GetAudioSink();
-
-  // Adds an observer for |audio_sink_|.
-  TestAudioSinkObserver observer(audio_sink_);
-  EXPECT_EQ(observer.state_changed_count_, 0);
-  EXPECT_EQ(observer.volume_changed_count_, 0);
+  properties_.state = "pending";
+
+  media_endpoint_->SetConfiguration(
+      transport_->GetTransportPath(media_endpoint_->object_path()),
+      properties_);
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 0);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
+}
+
+// TODO(mcchou): Adds test on media-removed events for STATE_PENDING and
+// STATE_ACTIVE.
+// Checks if the observer is notified on media-removed event when the state of
+// |audio_sink_| is STATE_DISCONNECTED. Once the media object is removed, the
+// audio sink is no longer valid.
+TEST_F(BluetoothAudioSinkChromeOSTest, MediaRemovedDuringDisconnectedState) {
+  GetAudioSink();
 
   // Gets the media object and makes it invisible to see if the state of the
   // audio sink changes accordingly.
-  FakeBluetoothMediaClient* media = static_cast<FakeBluetoothMediaClient*>(
-      DBusThreadManager::Get()->GetBluetoothMediaClient());
-  media->SetVisible(false);
-
-  BluetoothAudioSinkChromeOS* audio_sink_chromeos =
-      static_cast<BluetoothAudioSinkChromeOS*>(audio_sink_.get());
-  FakeBluetoothMediaEndpointServiceProvider* media_endpoint =
-      static_cast<FakeBluetoothMediaEndpointServiceProvider*>(
-          audio_sink_chromeos->GetEndpointServiceProvider());
+  media_->SetVisible(false);
+
+  GetFakeMediaEndpoint();
 
   EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_INVALID);
-  EXPECT_EQ(media_endpoint, nullptr);
-}
-
-// Checks if the observer is notified while the media transport is
-// removed(invisible). Once the media transport object is removed, the audio
-// sink is disconnected.
-TEST_F(BluetoothAudioSinkChromeOSTest, ObserverNotifiedWhenTransportRemoved) {
-  GetAudioSink();
-
-  // Adds an observer for |audio_sink_|.
-  TestAudioSinkObserver observer(audio_sink_);
-  EXPECT_EQ(observer.state_changed_count_, 0);
-  EXPECT_EQ(observer.volume_changed_count_, 0);
-
-  // Simulates calling SetConfiguration on the media endpoint object owned by
-  // |audio_sink_| with a fake transport path and a
-  // Delegate::TransportProperties structure.
-  BluetoothAudioSinkChromeOS* audio_sink_chromeos =
-      static_cast<BluetoothAudioSinkChromeOS*>(audio_sink_.get());
-  FakeBluetoothMediaEndpointServiceProvider* media_endpoint =
-      static_cast<FakeBluetoothMediaEndpointServiceProvider*>(
-          audio_sink_chromeos->GetEndpointServiceProvider());
-  ASSERT_NE(media_endpoint, nullptr);
-
-  media_endpoint->SetConfiguration(
-      ObjectPath(FakeBluetoothMediaTransportClient::kTransportPath),
-      properties_);
-
-  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_IDLE);
-  EXPECT_EQ(callback_count_, 1);
-  EXPECT_EQ(error_callback_count_, 0);
-  EXPECT_EQ(observer.state_changed_count_, 1);
-  EXPECT_EQ(observer.volume_changed_count_, 1);
-
-  // Gets the media transport object and makes it invalid to see if the state
-  // of the audio sink changes accordingly.
-  FakeBluetoothMediaTransportClient* transport =
-      static_cast<FakeBluetoothMediaTransportClient*>(
-          DBusThreadManager::Get()->GetBluetoothMediaTransportClient());
-
-  transport->SetValid(media_endpoint->object_path(), false);
-  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
-  EXPECT_NE(media_endpoint, nullptr);
+  EXPECT_EQ(media_endpoint_, nullptr);
+  EXPECT_EQ(observer_->state_changed_count_, 1);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
+}
+
+// Checks if the observer is  notified on media-removed event when the state of
+// |audio_sink_| is STATE_IDLE. Once the media object is removed, the audio sink
+// is no longer valid.
+TEST_F(BluetoothAudioSinkChromeOSTest, MediaRemovedDuringIdleState) {
+  GetAudioSink();
+
+  media_endpoint_->SelectConfiguration(
+      std::vector<uint8_t>({0x21, 0x15, 0x33, 0x2C}),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::SelectConfigurationCallback,
+                 base::Unretained(this)));
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 0);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
+
+  media_endpoint_->SetConfiguration(
+      transport_->GetTransportPath(media_endpoint_->object_path()),
+      properties_);
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_IDLE);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 1);
+  EXPECT_EQ(observer_->volume_changed_count_, 1);
+
+  // Gets the media object and makes it invisible to see if the state of the
+  // audio sink changes accordingly.
+  media_->SetVisible(false);
+
+  GetFakeMediaEndpoint();
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_INVALID);
+  EXPECT_EQ(media_endpoint_, nullptr);
+
+  // The state becomes disconnted and then invalid, since the removal of
+  // transport object should happend before media becomes invisible.
+  // State: STATE_IDLE -> STATE_DISCONNECTED -> STATE_INVALID
+  EXPECT_EQ(observer_->state_changed_count_, 3);
+  EXPECT_EQ(observer_->volume_changed_count_, 2);
+}
+
+// TODO(mcchou): Add tests on transport-removed event for STATE_PENDING and
+// STATE_ACTIVE.
+// Checks if the observer is notified on transport-removed event when the state
+// of |audio_sink_| is STATE_IDEL. Once the media transport object is removed,
+// the audio sink is disconnected.
+TEST_F(BluetoothAudioSinkChromeOSTest, TransportRemovedDuringIdleState) {
+  GetAudioSink();
+
+  media_endpoint_->SelectConfiguration(
+      std::vector<uint8_t>({0x21, 0x15, 0x33, 0x2C}),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::SelectConfigurationCallback,
+                 base::Unretained(this)));
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 0);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
+
+  media_endpoint_->SetConfiguration(
+      transport_->GetTransportPath(media_endpoint_->object_path()),
+      properties_);
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_IDLE);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 1);
+  EXPECT_EQ(observer_->volume_changed_count_, 1);
+
+  // Makes the transport object invalid to see if the state of the audio sink
+  // changes accordingly.
+  transport_->SetValid(media_endpoint_, false);
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_NE(media_endpoint_, nullptr);
+  EXPECT_EQ(observer_->state_changed_count_, 2);
+  EXPECT_EQ(observer_->volume_changed_count_, 2);
+}
+
+TEST_F(BluetoothAudioSinkChromeOSTest,
+       AdapterPoweredChangedDuringDisconnectedState) {
+  GetAudioSink();
+
+  adapter_->SetPowered(
+      false,
+      base::Bind(&BluetoothAudioSinkChromeOSTest::Callback,
+                 base::Unretained(this)),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::ErrorCallback,
+                 base::Unretained(this)));
+
+  EXPECT_TRUE(adapter_->IsPresent());
+  EXPECT_FALSE(adapter_->IsPowered());
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 0);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
+
+  adapter_->SetPowered(
+      true,
+      base::Bind(&BluetoothAudioSinkChromeOSTest::Callback,
+                 base::Unretained(this)),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::ErrorCallback,
+                 base::Unretained(this)));
+
+  EXPECT_TRUE(adapter_->IsPresent());
+  EXPECT_TRUE(adapter_->IsPowered());
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_EQ(callback_count_, 3);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 0);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
+}
+
+TEST_F(BluetoothAudioSinkChromeOSTest, AdapterPoweredChangedDuringIdleState) {
+  GetAudioSink();
+
+  media_endpoint_->SelectConfiguration(
+      std::vector<uint8_t>({0x21, 0x15, 0x33, 0x2C}),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::SelectConfigurationCallback,
+                 base::Unretained(this)));
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 0);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
+
+  media_endpoint_->SetConfiguration(
+      transport_->GetTransportPath(media_endpoint_->object_path()),
+      properties_);
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_IDLE);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 1);
+  EXPECT_EQ(observer_->volume_changed_count_, 1);
+
+  adapter_->SetPowered(
+      false,
+      base::Bind(&BluetoothAudioSinkChromeOSTest::Callback,
+                 base::Unretained(this)),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::ErrorCallback,
+                 base::Unretained(this)));
+  GetFakeMediaEndpoint();
+
+  EXPECT_TRUE(adapter_->IsPresent());
+  EXPECT_FALSE(adapter_->IsPowered());
+  EXPECT_NE(media_endpoint_, nullptr);
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_EQ(callback_count_, 3);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 2);
+  EXPECT_EQ(observer_->volume_changed_count_, 2);
+}
+
+// TODO(mcchou): Add tests on UnregisterAudioSink for STATE_PENDING and
+// STATE_ACTIVE.
+
+TEST_F(BluetoothAudioSinkChromeOSTest,
+       UnregisterAudioSinkDuringDisconnectedState) {
+  GetAudioSink();
+
+  audio_sink_->Unregister(
+      base::Bind(&BluetoothAudioSinkChromeOSTest::Callback,
+                 base::Unretained(this)),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::UnregisterErrorCallback,
+                 base::Unretained(this)));
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_INVALID);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 1);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
+}
+
+TEST_F(BluetoothAudioSinkChromeOSTest, UnregisterAudioSinkDuringIdleState) {
+  GetAudioSink();
+
+  media_endpoint_->SelectConfiguration(
+      std::vector<uint8_t>({0x21, 0x15, 0x33, 0x2C}),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::SelectConfigurationCallback,
+                 base::Unretained(this)));
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_DISCONNECTED);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 0);
+  EXPECT_EQ(observer_->volume_changed_count_, 0);
+
+  media_endpoint_->SetConfiguration(
+      transport_->GetTransportPath(media_endpoint_->object_path()),
+      properties_);
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_IDLE);
+  EXPECT_EQ(callback_count_, 2);
+  EXPECT_EQ(error_callback_count_, 0);
+  EXPECT_EQ(observer_->state_changed_count_, 1);
+  EXPECT_EQ(observer_->volume_changed_count_, 1);
+
+  audio_sink_->Unregister(
+      base::Bind(&BluetoothAudioSinkChromeOSTest::Callback,
+                 base::Unretained(this)),
+      base::Bind(&BluetoothAudioSinkChromeOSTest::UnregisterErrorCallback,
+                 base::Unretained(this)));
+
+  EXPECT_EQ(audio_sink_->GetState(), BluetoothAudioSink::STATE_INVALID);
+  EXPECT_EQ(callback_count_, 3);
+  EXPECT_EQ(error_callback_count_, 0);
+
+  // The state becomes disconnted and then invalid, since the removal of
+  // transport object should happend before the unregistration of endpoint.
+  // State: STATE_IDLE -> STATE_DISCONNECTED -> STATE_INVALID
+  EXPECT_EQ(observer_->state_changed_count_, 3);
+  EXPECT_EQ(observer_->volume_changed_count_, 2);
 }
 
 }  // namespace chromeos