OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #ifndef CHROME_TEST_DATA_WEBUI_ASYNC_GEN_H_ | 4 #ifndef CHROME_TEST_DATA_WEBUI_ASYNC_GEN_H_ |
5 #define CHROME_TEST_DATA_WEBUI_ASYNC_GEN_H_ | 5 #define CHROME_TEST_DATA_WEBUI_ASYNC_GEN_H_ |
6 | 6 |
7 #include "chrome/test/base/web_ui_browser_test.h" | 7 #include "chrome/test/base/web_ui_browser_test.h" |
8 #include "content/public/browser/web_ui_message_handler.h" | 8 #include "content/public/browser/web_ui_message_handler.h" |
9 #include "testing/gmock/include/gmock/gmock.h" | 9 #include "testing/gmock/include/gmock/gmock.h" |
10 | 10 |
11 namespace base { | 11 namespace base { |
12 class ListValue; | 12 class ListValue; |
13 } // namespace base | 13 } // namespace base |
14 | 14 |
15 // C++ support class for javascript-generated asynchronous tests. | 15 // C++ support class for javascript-generated asynchronous tests. |
16 class WebUIBrowserAsyncGenTest : public WebUIBrowserTest { | 16 class WebUIBrowserAsyncGenTest : public WebUIBrowserTest { |
17 public: | 17 public: |
18 WebUIBrowserAsyncGenTest(); | 18 WebUIBrowserAsyncGenTest(); |
19 virtual ~WebUIBrowserAsyncGenTest(); | 19 ~WebUIBrowserAsyncGenTest() override; |
20 | 20 |
21 protected: | 21 protected: |
22 class AsyncWebUIMessageHandler : public content::WebUIMessageHandler { | 22 class AsyncWebUIMessageHandler : public content::WebUIMessageHandler { |
23 public: | 23 public: |
24 AsyncWebUIMessageHandler(); | 24 AsyncWebUIMessageHandler(); |
25 ~AsyncWebUIMessageHandler(); | 25 ~AsyncWebUIMessageHandler(); |
26 | 26 |
27 MOCK_METHOD1(HandleTearDown, void(const base::ListValue*)); | 27 MOCK_METHOD1(HandleTearDown, void(const base::ListValue*)); |
28 | 28 |
29 private: | 29 private: |
30 void HandleCallJS(const base::ListValue* list_value); | 30 void HandleCallJS(const base::ListValue* list_value); |
31 | 31 |
32 // WebUIMessageHandler implementation. | 32 // WebUIMessageHandler implementation. |
33 virtual void RegisterMessages() override; | 33 void RegisterMessages() override; |
34 }; | 34 }; |
35 | 35 |
36 // Handler for this test fixture. | 36 // Handler for this test fixture. |
37 ::testing::StrictMock<AsyncWebUIMessageHandler> message_handler_; | 37 ::testing::StrictMock<AsyncWebUIMessageHandler> message_handler_; |
38 | 38 |
39 private: | 39 private: |
40 // Provide this object's handler. | 40 // Provide this object's handler. |
41 virtual content::WebUIMessageHandler* GetMockMessageHandler() override { | 41 content::WebUIMessageHandler* GetMockMessageHandler() override { |
42 return &message_handler_; | 42 return &message_handler_; |
43 } | 43 } |
44 | 44 |
45 virtual void SetUpOnMainThread() override { | 45 void SetUpOnMainThread() override { |
46 WebUIBrowserTest::SetUpOnMainThread(); | 46 WebUIBrowserTest::SetUpOnMainThread(); |
47 EXPECT_CALL(message_handler_, HandleTearDown(::testing::_)); | 47 EXPECT_CALL(message_handler_, HandleTearDown(::testing::_)); |
48 } | 48 } |
49 | 49 |
50 DISALLOW_COPY_AND_ASSIGN(WebUIBrowserAsyncGenTest); | 50 DISALLOW_COPY_AND_ASSIGN(WebUIBrowserAsyncGenTest); |
51 }; | 51 }; |
52 | 52 |
53 #endif // CHROME_TEST_DATA_WEBUI_ASYNC_GEN_H_ | 53 #endif // CHROME_TEST_DATA_WEBUI_ASYNC_GEN_H_ |
OLD | NEW |