1 // Copyright 2014 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 UI_OZONE_PLATFORM_DRI_TEST_MOCK_DRI_SURFACE_H_ 6 #define UI_OZONE_PLATFORM_DRI_TEST_MOCK_DRI_SURFACE_H_ 7 8 #include <vector> 9 10 #include "ui/ozone/platform/dri/dri_surface.h" 11 12 namespace gfx { 13 class Size; 14 } // namespace gfx 15 16 namespace ui { 17 18 class DriBuffer; 19 class DriWrapper; 20 21 class MockDriSurface : public DriSurface { 22 public: 23 MockDriSurface(DriWrapper* dri, const gfx::Size& size); 24 virtual ~MockDriSurface(); 25 26 const std::vector<ui::DriBuffer*>& bitmaps() const { return bitmaps_; } 27 void set_initialize_expectation(bool state) { 28 initialize_expectation_ = state; 29 } 30 31 private: 32 // DriSurface: 33 virtual ui::DriBuffer* CreateBuffer() OVERRIDE; 34 35 DriWrapper* dri_; // Not owned. 36 std::vector<DriBuffer*> bitmaps_; // Not owned. 37 38 bool initialize_expectation_; 39 40 DISALLOW_COPY_AND_ASSIGN(MockDriSurface); 41 }; 42 43 } // namespace ui 44 45 #endif // UI_OZONE_PLATFORM_DRI_TEST_MOCK_DRI_SURFACE_H_ 46