Page Speed Optimization Libraries
1.2.24.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 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 00051 void AdvanceTimeMs(int64 timeout_ms) { 00052 AdvanceTimeUs(timeout_ms * Timer::kMsUs); 00053 } 00054 void AdvanceTimeUs(int64 timeout_us); 00055 00058 void SetTimeUs(int64 time_us); 00059 00060 protected: 00061 virtual void AwaitWakeupUntilUs(int64 wakeup_time_us); 00062 00063 private: 00064 inline void SetTimeUsMutexHeld(int64 time_us); 00065 00066 MockTimer* timer_; 00067 QueuedWorkerPool::SequenceSet workers_; 00068 00069 DISALLOW_COPY_AND_ASSIGN(MockScheduler); 00070 }; 00071 00072 } 00073 00074 #endif ///< NET_INSTAWEB_UTIL_PUBLIC_MOCK_SCHEDULER_H_