Page Speed Optimization Libraries  1.5.27.2
net/instaweb/util/public/mock_scheduler.h
Go to the documentation of this file.
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_
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines