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

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

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 3d4517b2f44eeed6b9e48d46b4208a2d96ce34cd..493fa10408e25b86d56c93a50ac70ecd074972a3 100644
--- a/ui/ozone/platform/dri/test/mock_dri_wrapper.h
+++ b/ui/ozone/platform/dri/test/mock_dri_wrapper.h
@@ -33,6 +33,7 @@ class MockDriWrapper : public ui::DriWrapper {
   }
   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; }
@@ -49,6 +50,11 @@ class MockDriWrapper : public ui::DriWrapper {
     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,
@@ -103,6 +109,7 @@ class MockDriWrapper : public ui::DriWrapper {
   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_;