Index: ui/ozone/platform/dri/test/mock_dri_wrapper.h |
diff --git a/ui/ozone/platform/dri/test/mock_dri_wrapper.h b/ui/ozone/platform/dri/test/mock_dri_wrapper.h |
index 4c5fd6aef358a87b0d26d067626a6eb1546c2a67..3dc09f43d55c701a7f7053594fa341e4a854ee93 100644 |
--- a/ui/ozone/platform/dri/test/mock_dri_wrapper.h |
+++ b/ui/ozone/platform/dri/test/mock_dri_wrapper.h |
@@ -13,6 +13,8 @@ |
namespace ui { |
+class HardwareDisplayController; |
+ |
// The real DriWrapper makes actual DRM calls which we can't use in unit tests. |
class MockDriWrapper : public ui::DriWrapper { |
public: |
@@ -39,6 +41,8 @@ class MockDriWrapper : public ui::DriWrapper { |
create_dumb_buffer_expectation_ = state; |
} |
+ uint32_t current_framebuffer() const { return current_framebuffer_; } |
+ |
const std::vector<skia::RefPtr<SkSurface> > buffers() const { |
return buffers_; |
} |
@@ -100,8 +104,12 @@ class MockDriWrapper : public ui::DriWrapper { |
bool page_flip_expectation_; |
bool create_dumb_buffer_expectation_; |
+ uint32_t current_framebuffer_; |
+ |
std::vector<skia::RefPtr<SkSurface> > buffers_; |
+ HardwareDisplayController* controller_; |
+ |
DISALLOW_COPY_AND_ASSIGN(MockDriWrapper); |
}; |