React to primary display change instead of removal

chrome/browser/ui/webui/chromeos/login/oobe_display_chooser_unittest.cc

Index: chrome/browser/ui/webui/chromeos/login/oobe_display_chooser_unittest.cc
diff --git a/chrome/browser/ui/webui/chromeos/login/oobe_display_chooser_unittest.cc b/chrome/browser/ui/webui/chromeos/login/oobe_display_chooser_unittest.cc
index 517ec783ae834b9125449de84635e146ed66bfe4..acf074ae225fd6af55f93ded65c3882bc3763420 100644
--- a/chrome/browser/ui/webui/chromeos/login/oobe_display_chooser_unittest.cc
+++ b/chrome/browser/ui/webui/chromeos/login/oobe_display_chooser_unittest.cc
@@ -9,6 +9,7 @@
 #include "ash/display/display_configuration_controller.h"
 #include "ash/shell.h"
 #include "ash/test/ash_test_base.h"
+#include "base/run_loop.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/display/display.h"
 #include "ui/display/display_observer.h"
@@ -63,6 +64,7 @@ TEST_F(OobeDisplayChooserTest, PreferTouchAsPrimary) {
 
   EXPECT_EQ(ids[0], GetPrimaryDisplay());
   display_chooser.TryToPlaceUiOnTouchDisplay();
+  base::RunLoop().RunUntilIdle();
 
   EXPECT_EQ(ids[1], GetPrimaryDisplay());
 }
@@ -77,6 +79,7 @@ TEST_F(OobeDisplayChooserTest, AddingSecondTouchDisplayShouldbeNOP) {
 
   EXPECT_EQ(ids[0], GetPrimaryDisplay());
   display_chooser.TryToPlaceUiOnTouchDisplay();
+  base::RunLoop().RunUntilIdle();
 
   EXPECT_EQ(ids[0], GetPrimaryDisplay());
 }