| 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_CHROME_SEND_BROWSERTEST_H_ | 4 #ifndef CHROME_TEST_DATA_WEBUI_CHROME_SEND_BROWSERTEST_H_ |
| 5 #define CHROME_TEST_DATA_WEBUI_CHROME_SEND_BROWSERTEST_H_ | 5 #define CHROME_TEST_DATA_WEBUI_CHROME_SEND_BROWSERTEST_H_ |
| 6 #pragma once | 6 #pragma once |
| 7 | 7 |
| 8 #include "chrome/browser/ui/webui/web_ui_browsertest.h" | 8 #include "chrome/browser/ui/webui/web_ui_browsertest.h" |
| 9 #include "testing/gmock/include/gmock/gmock.h" | 9 #include "testing/gmock/include/gmock/gmock.h" |
| 10 | 10 |
| 11 // Test fixture for testing chrome.send. This class registers the "checkSend" | 11 // Test fixture for testing chrome.send. This class registers the "checkSend" |
| 12 // message, but should NOT receive it. | 12 // message, but should NOT receive it. |
| 13 class ChromeSendWebUITest : public WebUIBrowserTest { | 13 class ChromeSendWebUITest : public WebUIBrowserTest { |
| 14 public: | 14 public: |
| 15 ChromeSendWebUITest(); | 15 ChromeSendWebUITest(); |
| 16 virtual ~ChromeSendWebUITest(); | 16 virtual ~ChromeSendWebUITest(); |
| 17 | 17 |
| 18 // Mocked message handler class to register expects using gmock framework. | 18 // Mocked message handler class to register expects using gmock framework. |
| 19 class ChromeSendWebUIMessageHandler : public WebUIMessageHandler { | 19 class ChromeSendWebUIMessageHandler : public content::WebUIMessageHandler { |
| 20 public: | 20 public: |
| 21 ChromeSendWebUIMessageHandler(); | 21 ChromeSendWebUIMessageHandler(); |
| 22 ~ChromeSendWebUIMessageHandler(); | 22 ~ChromeSendWebUIMessageHandler(); |
| 23 | 23 |
| 24 MOCK_METHOD1(HandleCheckSend, void(const base::ListValue*)); | 24 MOCK_METHOD1(HandleCheckSend, void(const base::ListValue*)); |
| 25 | 25 |
| 26 private: | 26 private: |
| 27 virtual void RegisterMessages() OVERRIDE; | 27 virtual void RegisterMessages() OVERRIDE; |
| 28 }; | 28 }; |
| 29 | 29 |
| 30 | 30 |
| 31 protected: | 31 protected: |
| 32 // Strict mock will fail when unexpected chrome.send messages are received. | 32 // Strict mock will fail when unexpected chrome.send messages are received. |
| 33 ::testing::StrictMock<ChromeSendWebUIMessageHandler> message_handler_; | 33 ::testing::StrictMock<ChromeSendWebUIMessageHandler> message_handler_; |
| 34 | 34 |
| 35 private: | 35 private: |
| 36 virtual WebUIMessageHandler* GetMockMessageHandler() OVERRIDE; | 36 virtual content::WebUIMessageHandler* GetMockMessageHandler() OVERRIDE; |
| 37 | 37 |
| 38 DISALLOW_COPY_AND_ASSIGN(ChromeSendWebUITest); | 38 DISALLOW_COPY_AND_ASSIGN(ChromeSendWebUITest); |
| 39 }; | 39 }; |
| 40 | 40 |
| 41 // Test fixture for verifying chrome.send messages are passed through. This | 41 // Test fixture for verifying chrome.send messages are passed through. This |
| 42 // class DOES expect to receive the "checkSend" message. | 42 // class DOES expect to receive the "checkSend" message. |
| 43 class ChromeSendPassthroughWebUITest : public ChromeSendWebUITest { | 43 class ChromeSendPassthroughWebUITest : public ChromeSendWebUITest { |
| 44 public: | 44 public: |
| 45 ChromeSendPassthroughWebUITest(); | 45 ChromeSendPassthroughWebUITest(); |
| 46 virtual ~ChromeSendPassthroughWebUITest(); | 46 virtual ~ChromeSendPassthroughWebUITest(); |
| 47 | 47 |
| 48 private: | 48 private: |
| 49 virtual void SetUpOnMainThread() OVERRIDE; | 49 virtual void SetUpOnMainThread() OVERRIDE; |
| 50 | 50 |
| 51 DISALLOW_COPY_AND_ASSIGN(ChromeSendPassthroughWebUITest); | 51 DISALLOW_COPY_AND_ASSIGN(ChromeSendPassthroughWebUITest); |
| 52 }; | 52 }; |
| 53 | 53 |
| 54 #endif // CHROME_TEST_DATA_WEBUI_CHROME_SEND_BROWSERTEST_H_ | 54 #endif // CHROME_TEST_DATA_WEBUI_CHROME_SEND_BROWSERTEST_H_ |
| OLD | NEW |