Awake Mac host display when remote session starts.

remoting/host/capturer_mac_unittest.cc

 // Copyright (c) 2011 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 "remoting/host/capturer.h"
 
 #include <ApplicationServices/ApplicationServices.h>
 
 #include <ostream>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/message_loop.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace remoting {
 
 class CapturerMacTest : public testing::Test {
  protected:
   virtual void SetUp() {
     capturer_.reset(Capturer::Create());
   }
 
   void AddDirtyRect() {
     SkIRect rect = SkIRect::MakeXYWH(0, 0, 10, 10);
     region_.op(rect, SkRegion::kUnion_Op);
   }
 
   scoped_ptr<Capturer> capturer_;
   SkRegion region_;
+  MessageLoop message_loop_;
 };
 
 // CapturerCallback1 verifies that the whole screen is initially dirty.
 class CapturerCallback1 {
  public:
   CapturerCallback1() { }
   void CaptureDoneCallback(scoped_refptr<CaptureData> capture_data);
 
  private:
   DISALLOW_COPY_AND_ASSIGN(CapturerCallback1);
@@ skipping 49 matching lines @@
   // Check that we get an initial full-screen updated.
   CapturerCallback1 callback1;
   capturer_->CaptureInvalidRegion(base::Bind(
       &CapturerCallback1::CaptureDoneCallback, base::Unretained(&callback1)));
   // Check that subsequent dirty rects are propagated correctly.
   AddDirtyRect();
   CapturerCallback2 callback2(region_);
   capturer_->InvalidateRegion(region_);
   capturer_->CaptureInvalidRegion(base::Bind(
       &CapturerCallback2::CaptureDoneCallback, base::Unretained(&callback2)));
+  message_loop_.RunAllPending();
 }
 
 }  // namespace remoting
 
 namespace gfx {
 
 std::ostream& operator<<(std::ostream& out, const SkRegion& region) {
   out << "SkRegion(";
   for (SkRegion::Iterator i(region); !i.done(); i.next()) {
     const SkIRect& r = i.rect();
     out << "(" << r.fLeft << ","  << r.fTop << ","
         << r.fRight  << ","  << r.fBottom << ")";
   }
   out << ")";
   return out;
 }
 
 }  // namespace gfx