Page Speed Optimization Libraries
1.3.25.1
|
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 #include "net/instaweb/util/public/timer.h" 00026 00027 namespace net_instaweb { 00028 00029 class MockTimer; 00030 class ThreadSystem; 00031 00038 class MockScheduler : public Scheduler { 00039 public: 00040 MockScheduler(ThreadSystem* thread_system, 00041 MockTimer* timer); 00042 virtual ~MockScheduler(); 00043 00044 virtual void RegisterWorker(QueuedWorkerPool::Sequence* w); 00045 virtual void UnregisterWorker(QueuedWorkerPool::Sequence* w); 00046 00048 void AwaitQuiescence(); 00049 00052 void AdvanceTimeMs(int64 timeout_ms) { 00053 AdvanceTimeUs(timeout_ms * Timer::kMsUs); 00054 } 00055 void AdvanceTimeUs(int64 timeout_us); 00056 00059 void SetTimeUs(int64 time_us); 00060 00061 protected: 00062 virtual void AwaitWakeupUntilUs(int64 wakeup_time_us); 00063 00064 private: 00065 inline void SetTimeUsMutexHeld(int64 time_us); 00066 00067 MockTimer* timer_; 00068 QueuedWorkerPool::SequenceSet workers_; 00069 00070 DISALLOW_COPY_AND_ASSIGN(MockScheduler); 00071 }; 00072 00073 } 00074 00075 #endif ///< NET_INSTAWEB_UTIL_PUBLIC_MOCK_SCHEDULER_H_