| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef IOS_WEB_SHELL_TEST_WEB_SHELL_TEST_UTIL_H_ | 5 #import "ios/web/shell/test/app/web_shell_test_util.h" |
| 6 #define IOS_WEB_SHELL_TEST_WEB_SHELL_TEST_UTIL_H_ | 6 |
| 7 #import <UIKit/UIKit.h> |
| 7 | 8 |
| 8 #import "ios/web/shell/view_controller.h" | 9 #import "ios/web/shell/view_controller.h" |
| 9 | 10 |
| 10 namespace web { | 11 namespace web { |
| 11 namespace web_shell_test_util { | 12 namespace web_shell_test_util { |
| 12 | 13 |
| 13 // Gets the current ViewController for the web shell. | 14 web::WebState* GetCurrentWebState() { |
| 14 ViewController* GetCurrentViewController(); | 15 ViewController* view_controller = static_cast<ViewController*>([[ |
| 16 [[UIApplication sharedApplication] delegate] window] rootViewController]); |
| 17 return view_controller.webState; |
| 18 } |
| 15 | 19 |
| 16 } // namespace web_shell_test_util | 20 } // namespace web_shell_test_util |
| 17 } // namespace web | 21 } // namespace web |
| 18 | |
| 19 #endif // IOS_WEB_SHELL_TEST_WEB_SHELL_TEST_UTIL_H_ | |
| OLD | NEW |