| // Copyright 2019 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 "chrome/test/base/chrome_test_utils.h" |
| |
| #include "build/build_config.h" |
| |
| #if defined(OS_ANDROID) |
| #include "chrome/browser/ui/android/tab_model/tab_model.h" |
| #include "chrome/browser/ui/android/tab_model/tab_model_list.h" |
| #else |
| #include "chrome/browser/ui/browser.h" |
| #include "chrome/browser/ui/tabs/tab_strip_model.h" |
| #endif |
| |
| namespace chrome_test_utils { |
| |
| content::WebContents* GetActiveWebContents(PlatformBrowserTest* browser_test) { |
| #if defined(OS_ANDROID) |
| for (size_t i = 0; i < TabModelList::size(); ++i) { |
| if (TabModelList::get(i)->IsCurrentModel()) |
| return TabModelList::get(i)->GetActiveWebContents(); |
| } |
| NOTREACHED() << "No active TabModel??"; |
| return nullptr; |
| #else |
| return browser_test->browser()->tab_strip_model()->GetActiveWebContents(); |
| #endif |
| } |
| |
| } // namespace chrome_test_utils |