| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #ifndef CONTENT_SHELL_RENDERER_TEST_RUNNER_WEBTASK_H_ | |
| 6 #define CONTENT_SHELL_RENDERER_TEST_RUNNER_WEBTASK_H_ | |
| 7 | |
| 8 #include <vector> | |
| 9 | |
| 10 namespace content { | |
| 11 | |
| 12 class WebTaskList; | |
| 13 | |
| 14 // WebTask represents a task which can run by WebTestDelegate::postTask() or | |
| 15 // WebTestDelegate::postDelayedTask(). | |
| 16 class WebTask { | |
| 17 public: | |
| 18 explicit WebTask(WebTaskList*); | |
| 19 virtual ~WebTask(); | |
| 20 | |
| 21 // The main code of this task. | |
| 22 // An implementation of run() should return immediately if cancel() was call
ed. | |
| 23 virtual void run() = 0; | |
| 24 virtual void cancel() = 0; | |
| 25 | |
| 26 protected: | |
| 27 WebTaskList* m_taskList; | |
| 28 }; | |
| 29 | |
| 30 class WebTaskList { | |
| 31 public: | |
| 32 WebTaskList(); | |
| 33 ~WebTaskList(); | |
| 34 void registerTask(WebTask*); | |
| 35 void unregisterTask(WebTask*); | |
| 36 void revokeAll(); | |
| 37 | |
| 38 private: | |
| 39 std::vector<WebTask*> m_tasks; | |
| 40 }; | |
| 41 | |
| 42 // A task containing an object pointer of class T. Derived classes should | |
| 43 // override runIfValid() which in turn can safely invoke methods on the | |
| 44 // m_object. The Class T must have "WebTaskList* mutable_task_list()". | |
| 45 template<class T> | |
| 46 class WebMethodTask : public WebTask { | |
| 47 public: | |
| 48 explicit WebMethodTask(T* object) | |
| 49 : WebTask(object->mutable_task_list()) | |
| 50 , m_object(object) | |
| 51 { | |
| 52 } | |
| 53 | |
| 54 virtual ~WebMethodTask() { } | |
| 55 | |
| 56 virtual void run() | |
| 57 { | |
| 58 if (m_object) | |
| 59 runIfValid(); | |
| 60 } | |
| 61 | |
| 62 virtual void cancel() | |
| 63 { | |
| 64 m_object = 0; | |
| 65 m_taskList->unregisterTask(this); | |
| 66 m_taskList = 0; | |
| 67 } | |
| 68 | |
| 69 virtual void runIfValid() = 0; | |
| 70 | |
| 71 protected: | |
| 72 T* m_object; | |
| 73 }; | |
| 74 | |
| 75 } // namespace content | |
| 76 | |
| 77 #endif // CONTENT_SHELL_RENDERER_TEST_RUNNER_WEBTASK_H_ | |
| OLD | NEW |