| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 #include "components/test_runner/mock_color_chooser.h" | 5 #include "components/test_runner/mock_color_chooser.h" |
| 6 | 6 |
| 7 #include "components/test_runner/test_runner.h" | |
| 8 #include "components/test_runner/web_test_delegate.h" | 7 #include "components/test_runner/web_test_delegate.h" |
| 9 #include "components/test_runner/web_test_proxy.h" | 8 #include "components/test_runner/web_test_proxy.h" |
| 10 | 9 |
| 11 namespace test_runner { | 10 namespace test_runner { |
| 12 | 11 |
| 13 namespace { | 12 namespace { |
| 14 class HostMethodTask : public WebMethodTask<MockColorChooser> { | 13 class HostMethodTask : public WebMethodTask<MockColorChooser> { |
| 15 public: | 14 public: |
| 16 typedef void (MockColorChooser::*CallbackMethodType)(); | 15 typedef void (MockColorChooser::*CallbackMethodType)(); |
| 17 HostMethodTask(MockColorChooser* object, CallbackMethodType callback) | 16 HostMethodTask(MockColorChooser* object, CallbackMethodType callback) |
| 18 : WebMethodTask<MockColorChooser>(object), | 17 : WebMethodTask<MockColorChooser>(object), |
| 19 callback_(callback) {} | 18 callback_(callback) {} |
| 20 | 19 |
| 21 void RunIfValid() override { (object_->*callback_)(); } | 20 void RunIfValid() override { (object_->*callback_)(); } |
| 22 | 21 |
| 23 private: | 22 private: |
| 24 CallbackMethodType callback_; | 23 CallbackMethodType callback_; |
| 25 }; | 24 }; |
| 26 | 25 |
| 27 } // namespace | 26 } // namespace |
| 28 | 27 |
| 29 MockColorChooser::MockColorChooser(blink::WebColorChooserClient* client, | 28 MockColorChooser::MockColorChooser(blink::WebColorChooserClient* client, |
| 30 WebTestDelegate* delegate, | 29 WebTestDelegate* delegate, |
| 31 TestRunner* test_runner) | 30 WebTestProxyBase* proxy) |
| 32 : client_(client), delegate_(delegate), test_runner_(test_runner) { | 31 : client_(client), |
| 33 test_runner_->DidOpenChooser(); | 32 delegate_(delegate), |
| 33 proxy_(proxy) { |
| 34 proxy_->DidOpenChooser(); |
| 34 } | 35 } |
| 35 | 36 |
| 36 MockColorChooser::~MockColorChooser() { | 37 MockColorChooser::~MockColorChooser() { |
| 37 test_runner_->DidCloseChooser(); | 38 proxy_->DidCloseChooser(); |
| 38 } | 39 } |
| 39 | 40 |
| 40 void MockColorChooser::setSelectedColor(const blink::WebColor color) {} | 41 void MockColorChooser::setSelectedColor(const blink::WebColor color) {} |
| 41 | 42 |
| 42 void MockColorChooser::endChooser() { | 43 void MockColorChooser::endChooser() { |
| 43 DCHECK(client_); | |
| 44 DCHECK(delegate_); | |
| 45 delegate_->PostDelayedTask( | 44 delegate_->PostDelayedTask( |
| 46 new HostMethodTask(this, &MockColorChooser::InvokeDidEndChooser), 0); | 45 new HostMethodTask(this, &MockColorChooser::InvokeDidEndChooser), 0); |
| 47 delegate_ = nullptr; | |
| 48 } | 46 } |
| 49 | 47 |
| 50 void MockColorChooser::InvokeDidEndChooser() { | 48 void MockColorChooser::InvokeDidEndChooser() { |
| 51 client_->didEndChooser(); | 49 client_->didEndChooser(); |
| 52 client_ = nullptr; | |
| 53 } | 50 } |
| 54 | 51 |
| 55 } // namespace test_runner | 52 } // namespace test_runner |
| OLD | NEW |