Reflect the current state in Media Router Action icon.

chrome/browser/ui/views/media_router/media_router_ui_browsertest.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 "base/bind.h"
 #include "base/thread_task_runner_handle.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/toolbar/media_router_action.h"
@@ skipping 26 matching lines @@
     ASSERT_TRUE(browser_actions_container);
 
     media_router_action_.reset(new MediaRouterAction(browser()));
 
     // Sets delegate on |media_router_action_|.
     toolbar_action_view_.reset(
         new ToolbarActionView(media_router_action_.get(), browser()->profile(),
                               browser_actions_container));
   }
 
+  void TearDownOnMainThread() override {
+    toolbar_action_view_.reset();
+    media_router_action_.reset();
+    InProcessBrowserTest::TearDownOnMainThread();
+  }
+
   void OpenMediaRouterDialogAndWaitForNewWebContents() {
     content::TestNavigationObserver nav_observer(NULL);
     nav_observer.StartWatchingNewWebContents();
 
     ToolbarView* toolbar =
         BrowserView::GetBrowserViewForBrowser(browser())->toolbar();
 
     // When the Media Router Action executes, it opens a dialog with web
     // contents to chrome://media-router.
     base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE,
@@ skipping 42 matching lines @@
 
   // Navigate away.
   ui_test_utils::NavigateToURL(browser(), GURL("about:blank"));
 
   // The navigation should have removed the previously created dialog.
   // We expect a new dialog WebContents to be created by calling this.
   OpenMediaRouterDialogAndWaitForNewWebContents();
 }
 
 }  // namespace media_router