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

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

 // 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.
 
 #include "ui/ozone/platform/dri/test/mock_dri_wrapper.h"
 
 #include <xf86drm.h>
 #include <xf86drmMode.h>
 
 #include "base/logging.h"
@@ skipping 12 matching lines @@
 
 MockDriWrapper::MockDriWrapper(int fd)
     : DriWrapper(""),
       get_crtc_call_count_(0),
       set_crtc_call_count_(0),
       restore_crtc_call_count_(0),
       add_framebuffer_call_count_(0),
       remove_framebuffer_call_count_(0),
       page_flip_call_count_(0),
       overlay_flip_call_count_(0),
+      handle_events_count_(0),
       set_crtc_expectation_(true),
       add_framebuffer_expectation_(true),
       page_flip_expectation_(true),
       create_dumb_buffer_expectation_(true),
       current_framebuffer_(0) {
   fd_ = fd;
 }
 
 MockDriWrapper::~MockDriWrapper() {
   fd_ = -1;
@@ skipping 82 matching lines @@
 }
 
 bool MockDriWrapper::MoveCursor(uint32_t crtc_id, const gfx::Point& point) {
   return true;
 }
 
 void MockDriWrapper::HandleEvent(drmEventContext& event) {
   CHECK(!controllers_.empty());
   controllers_.front()->OnPageFlipEvent(0, 0, 0);
   controllers_.pop();
+  handle_events_count_++;
 }
 
 bool MockDriWrapper::CreateDumbBuffer(const SkImageInfo& info,
                                       uint32_t* handle,
                                       uint32_t* stride,
                                       void** pixels) {
   if (!create_dumb_buffer_expectation_)
     return false;
 
   *handle = 0;
   *stride = info.minRowBytes();
   *pixels = new char[info.getSafeSize(*stride)];
   buffers_.push_back(
       skia::AdoptRef(SkSurface::NewRasterDirect(info, *pixels, *stride)));
   buffers_.back()->getCanvas()->clear(SK_ColorBLACK);
 
   return true;
 }
 
 void MockDriWrapper::DestroyDumbBuffer(const SkImageInfo& info,
                                        uint32_t handle,
                                        uint32_t stride,
                                        void* pixels) {
   delete[] static_cast<char*>(pixels);
 }
 
 }  // namespace ui