00001 /* 00002 * Copyright 2011 Google Inc. 00003 * 00004 * Licensed under the Apache License, Version 2.0 (the "License"); 00005 * you may not use this file except in compliance with the License. 00006 * You may obtain a copy of the License at 00007 * 00008 * http:///www.apache.org/licenses/LICENSE-2.0 00009 * 00010 * Unless required by applicable law or agreed to in writing, software 00011 * distributed under the License is distributed on an "AS IS" BASIS, 00012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 00013 * See the License for the specific language governing permissions and 00014 * limitations under the License. 00015 */ 00016 00018 00019 #ifndef NET_INSTAWEB_UTIL_PUBLIC_MOCK_SCHEDULER_H_ 00020 #define NET_INSTAWEB_UTIL_PUBLIC_MOCK_SCHEDULER_H_ 00021 00022 #include "net/instaweb/util/public/basictypes.h" 00023 #include "net/instaweb/util/public/queued_worker_pool.h" 00024 #include "net/instaweb/util/public/scheduler.h" 00025 00026 namespace net_instaweb { 00027 00028 class MockTimer; 00029 class ThreadSystem; 00030 00037 class MockScheduler : public Scheduler { 00038 public: 00039 MockScheduler(ThreadSystem* thread_system, 00040 MockTimer* timer); 00041 virtual ~MockScheduler(); 00042 00043 virtual void RegisterWorker(QueuedWorkerPool::Sequence* w); 00044 virtual void UnregisterWorker(QueuedWorkerPool::Sequence* w); 00045 00047 void AwaitQuiescence(); 00048 00049 protected: 00050 virtual void AwaitWakeupUntilUs(int64 wakeup_time_us); 00051 00052 private: 00053 MockTimer* timer_; 00054 QueuedWorkerPool::SequenceSet workers_; 00055 00056 DISALLOW_COPY_AND_ASSIGN(MockScheduler); 00057 }; 00058 00059 } 00060 00061 #endif ///< NET_INSTAWEB_UTIL_PUBLIC_MOCK_SCHEDULER_H_