Index: content/public/test/repeated_notification_observer.cc |
diff --git a/content/public/test/repeated_notification_observer.cc b/content/public/test/repeated_notification_observer.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..f9e6a99904806bb90d4b3fea9efdf9c5e6eb1361 |
--- /dev/null |
+++ b/content/public/test/repeated_notification_observer.cc |
@@ -0,0 +1,38 @@ |
+// Copyright (c) 2017 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 "content/public/test/repeated_notification_observer.h" |
+ |
+#include "base/macros.h" |
+#include "content/public/browser/browser_thread.h" |
+#include "content/public/browser/notification_service.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace content { |
+ |
+RepeatedNotificationObserver::RepeatedNotificationObserver(int type, int count) |
+ : num_outstanding_(count), running_(false) { |
+ registrar_.Add(this, type, NotificationService::AllSources()); |
+} |
+ |
+void RepeatedNotificationObserver::Observe(int type, |
+ const NotificationSource& source, |
+ const NotificationDetails& details) { |
+ ASSERT_GT(num_outstanding_, 0); |
+ if (!--num_outstanding_ && running_) { |
+ BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, |
+ run_loop_.QuitClosure()); |
+ } |
+} |
+ |
+void RepeatedNotificationObserver::Wait() { |
+ if (num_outstanding_ <= 0) |
+ return; |
+ |
+ running_ = true; |
+ run_loop_.Run(); |
+ running_ = false; |
+} |
+ |
+} // namespace content |