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 CHROME_BROWSER_PREVIEWS_PREVIEWS_INFOBAR_TAB_HELPER_H_ | 5 #ifndef CHROME_BROWSER_PREVIEWS_PREVIEWS_INFOBAR_TAB_HELPER_H_ |
6 #define CHROME_BROWSER_PREVIEWS_PREVIEWS_INFOBAR_TAB_HELPER_H_ | 6 #define CHROME_BROWSER_PREVIEWS_PREVIEWS_INFOBAR_TAB_HELPER_H_ |
7 | 7 |
8 #include "base/macros.h" | 8 #include "base/macros.h" |
9 #include "content/public/browser/web_contents_observer.h" | 9 #include "content/public/browser/web_contents_observer.h" |
10 #include "content/public/browser/web_contents_user_data.h" | 10 #include "content/public/browser/web_contents_user_data.h" |
(...skipping 19 matching lines...) Expand all Loading... |
30 | 30 |
31 // Sets whether the InfoBar for a preview has been shown for the page. | 31 // Sets whether the InfoBar for a preview has been shown for the page. |
32 // |displayed_preview_infobar_| is reset to false on | 32 // |displayed_preview_infobar_| is reset to false on |
33 // DidStartProvisionalLoadForFrame for the main frame. | 33 // DidStartProvisionalLoadForFrame for the main frame. |
34 void set_displayed_preview_infobar(bool displayed) { | 34 void set_displayed_preview_infobar(bool displayed) { |
35 displayed_preview_infobar_ = displayed; | 35 displayed_preview_infobar_ = displayed; |
36 } | 36 } |
37 | 37 |
38 private: | 38 private: |
39 friend class content::WebContentsUserData<PreviewsInfoBarTabHelper>; | 39 friend class content::WebContentsUserData<PreviewsInfoBarTabHelper>; |
| 40 friend class PreviewsInfoBarTabHelperUnitTest; |
40 | 41 |
41 explicit PreviewsInfoBarTabHelper(content::WebContents* web_contents); | 42 explicit PreviewsInfoBarTabHelper(content::WebContents* web_contents); |
42 | 43 |
43 // Overridden from content::WebContentsObserver: | 44 // Overridden from content::WebContentsObserver: |
44 void DidStartProvisionalLoadForFrame( | 45 void DidStartProvisionalLoadForFrame( |
45 content::RenderFrameHost* render_frame_host, | 46 content::RenderFrameHost* render_frame_host, |
46 const GURL& validated_url, | 47 const GURL& validated_url, |
47 bool is_error_page, | 48 bool is_error_page, |
48 bool is_iframe_srcdoc) override; | 49 bool is_iframe_srcdoc) override; |
49 void DidFinishNavigation( | 50 void DidFinishNavigation( |
50 content::NavigationHandle* navigation_handle) override; | 51 content::NavigationHandle* navigation_handle) override; |
51 | 52 |
52 // True if the InfoBar for a preview has been shown for the page. | 53 // True if the InfoBar for a preview has been shown for the page. |
53 bool displayed_preview_infobar_; | 54 bool displayed_preview_infobar_; |
54 | 55 |
55 DISALLOW_COPY_AND_ASSIGN(PreviewsInfoBarTabHelper); | 56 DISALLOW_COPY_AND_ASSIGN(PreviewsInfoBarTabHelper); |
56 }; | 57 }; |
57 | 58 |
58 #endif // CHROME_BROWSER_PREVIEWS_PREVIEWS_INFOBAR_TAB_HELPER_H_ | 59 #endif // CHROME_BROWSER_PREVIEWS_PREVIEWS_INFOBAR_TAB_HELPER_H_ |
OLD | NEW |