[Ozone-DRI] Fix waiting for page flips in mirror mode

ui/ozone/platform/dri/test/mock_dri_wrapper.h

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef UI_OZONE_PLATFORM_DRI_TEST_MOCK_DRI_WRAPPER_H_
 #define UI_OZONE_PLATFORM_DRI_TEST_MOCK_DRI_WRAPPER_H_
 
 #include <queue>
 #include <vector>
 
@@ skipping 15 matching lines @@
   int get_set_crtc_call_count() const { return set_crtc_call_count_; }
   int get_restore_crtc_call_count() const { return restore_crtc_call_count_; }
   int get_add_framebuffer_call_count() const {
     return add_framebuffer_call_count_;
   }
   int get_remove_framebuffer_call_count() const {
     return remove_framebuffer_call_count_;
   }
   int get_page_flip_call_count() const { return page_flip_call_count_; }
   int get_overlay_flip_call_count() const { return overlay_flip_call_count_; }
+  int get_handle_events_count() const { return handle_events_count_; }
   void fail_init() { fd_ = -1; }
   void set_set_crtc_expectation(bool state) { set_crtc_expectation_ = state; }
   void set_page_flip_expectation(bool state) { page_flip_expectation_ = state; }
   void set_add_framebuffer_expectation(bool state) {
     add_framebuffer_expectation_ = state;
   }
   void set_create_dumb_buffer_expectation(bool state) {
     create_dumb_buffer_expectation_ = state;
   }
 
   uint32_t current_framebuffer() const { return current_framebuffer_; }
 
   const std::vector<skia::RefPtr<SkSurface> > buffers() const {
     return buffers_;
   }
 
+  // Overwrite the list of controllers used when serving the PageFlip requests.
+  void set_controllers(const std::queue<CrtcController*>& controllers) {
+    controllers_ = controllers;
+  }
+
   // DriWrapper:
   virtual ScopedDrmCrtcPtr GetCrtc(uint32_t crtc_id) OVERRIDE;
   virtual bool SetCrtc(uint32_t crtc_id,
                        uint32_t framebuffer,
                        std::vector<uint32_t> connectors,
                        drmModeModeInfo* mode) OVERRIDE;
   virtual bool SetCrtc(drmModeCrtc* crtc,
                        std::vector<uint32_t> connectors) OVERRIDE;
   virtual ScopedDrmConnectorPtr GetConnector(uint32_t connector_id) OVERRIDE;
   virtual bool AddFramebuffer(uint32_t width,
@@ skipping 34 matching lines @@
                                  void* pixels) OVERRIDE;
 
  private:
   int get_crtc_call_count_;
   int set_crtc_call_count_;
   int restore_crtc_call_count_;
   int add_framebuffer_call_count_;
   int remove_framebuffer_call_count_;
   int page_flip_call_count_;
   int overlay_flip_call_count_;
+  int handle_events_count_;
 
   bool set_crtc_expectation_;
   bool add_framebuffer_expectation_;
   bool page_flip_expectation_;
   bool create_dumb_buffer_expectation_;
 
   uint32_t current_framebuffer_;
 
   std::vector<skia::RefPtr<SkSurface> > buffers_;
 
   std::queue<CrtcController*> controllers_;
 
   DISALLOW_COPY_AND_ASSIGN(MockDriWrapper);
 };
 
 }  // namespace ui
 
 #endif  // UI_OZONE_PLATFORM_DRI_TEST_MOCK_DRI_WRAPPER_H_