Index: ios/chrome/browser/ui/ntp/new_tab_page_controller_unittest.mm |
diff --git a/ios/chrome/browser/ui/ntp/new_tab_page_controller_unittest.mm b/ios/chrome/browser/ui/ntp/new_tab_page_controller_unittest.mm |
index 2314916466bf0a6f944a1ac9db346b815501ddf0..4ff7717acc7815c429040b85051b8d314bfedd51 100644 |
--- a/ios/chrome/browser/ui/ntp/new_tab_page_controller_unittest.mm |
+++ b/ios/chrome/browser/ui/ntp/new_tab_page_controller_unittest.mm |
@@ -19,6 +19,7 @@ |
#include "ios/chrome/browser/sessions/ios_chrome_tab_restore_service_factory.h" |
#import "ios/chrome/browser/sessions/test_session_service.h" |
#import "ios/chrome/browser/tabs/tab_model.h" |
+#import "ios/chrome/browser/ui/ntp/new_tab_page_controller_testing.h" |
#import "ios/chrome/browser/ui/ntp/new_tab_page_view.h" |
#include "ios/chrome/browser/ui/ui_util.h" |
#include "ios/chrome/test/block_cleanup_test.h" |
@@ -34,32 +35,6 @@ |
#error "This file requires ARC support." |
#endif |
-@interface NewTabPageController (TestSupport) |
-- (id<NewTabPagePanelProtocol>)currentController; |
-- (BookmarkHomeTabletNTPController*)bookmarkController; |
-- (id<NewTabPagePanelProtocol>)incognitoController; |
-@end |
- |
-@interface NewTabPageController (PrivateMethods) |
-@property(nonatomic, retain) NewTabPageView* ntpView; |
-@end |
- |
-@implementation NewTabPageController (TestSupport) |
- |
-- (id<NewTabPagePanelProtocol>)currentController { |
- return currentController_; |
-} |
- |
-- (BookmarkHomeTabletNTPController*)bookmarkController { |
- return bookmarkController_.get(); |
-} |
- |
-- (id<NewTabPagePanelProtocol>)incognitoController { |
- return incognitoController_; |
-} |
- |
-@end |
- |
namespace { |
class NewTabPageControllerTest : public BlockCleanupTest { |