1 // Copyright (c) 2011 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 GPU_COMMAND_BUFFER_SERVICE_GPU_SCHEDULER_MOCK_H_ 6 #define GPU_COMMAND_BUFFER_SERVICE_GPU_SCHEDULER_MOCK_H_ 7 8 #include "gpu/command_buffer/service/gpu_scheduler.h" 9 #include "testing/gmock/include/gmock/gmock.h" 10 11 namespace gpu { 12 13 class MockGpuScheduler : public GpuScheduler { 14 public: 15 explicit MockGpuScheduler(CommandBuffer* command_buffer) 16 : GpuScheduler(command_buffer) { 17 } 18 19 MOCK_METHOD1(GetSharedMemoryBuffer, Buffer(int32 shm_id)); 20 MOCK_METHOD1(set_token, void(int32 token)); 21 22 private: 23 DISALLOW_COPY_AND_ASSIGN(MockGpuScheduler); 24 }; 25 26 } // namespace gpu 27 28 #endif // GPU_COMMAND_BUFFER_SERVICE_GPU_SCHEDULER_MOCK_H_ 29