| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROME_TEST_BASE_THREAD_OBSERVER_HELPER_H_ | 5 #ifndef CHROME_TEST_BASE_THREAD_OBSERVER_HELPER_H_ |
| 6 #define CHROME_TEST_BASE_THREAD_OBSERVER_HELPER_H_ | 6 #define CHROME_TEST_BASE_THREAD_OBSERVER_HELPER_H_ |
| 7 #pragma once | 7 #pragma once |
| 8 | 8 |
| 9 #include "base/bind.h" |
| 9 #include "base/memory/ref_counted.h" | 10 #include "base/memory/ref_counted.h" |
| 10 #include "base/synchronization/waitable_event.h" | 11 #include "base/synchronization/waitable_event.h" |
| 11 #include "content/public/browser/browser_thread.h" | 12 #include "content/public/browser/browser_thread.h" |
| 12 #include "content/public/browser/notification_registrar.h" | 13 #include "content/public/browser/notification_registrar.h" |
| 13 #include "content/test/notification_observer_mock.h" | 14 #include "content/test/notification_observer_mock.h" |
| 14 | 15 |
| 15 // Helper class to add and remove observers on a non-UI thread from | 16 // Helper class to add and remove observers on a non-UI thread from |
| 16 // the UI thread. | 17 // the UI thread. |
| 17 template <class T, typename Traits> | 18 template <class T, typename Traits> |
| 18 class ThreadObserverHelper : public base::RefCountedThreadSafe<T, Traits> { | 19 class ThreadObserverHelper : public base::RefCountedThreadSafe<T, Traits> { |
| 19 public: | 20 public: |
| 20 explicit ThreadObserverHelper(content::BrowserThread::ID id) | 21 explicit ThreadObserverHelper(content::BrowserThread::ID id) |
| 21 : id_(id), done_event_(false, false) {} | 22 : id_(id), done_event_(false, false) {} |
| 22 | 23 |
| 23 void Init() { | 24 void Init() { |
| 24 using content::BrowserThread; | 25 using content::BrowserThread; |
| 25 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 26 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| 26 BrowserThread::PostTask( | 27 BrowserThread::PostTask( |
| 27 id_, | 28 id_, |
| 28 FROM_HERE, | 29 FROM_HERE, |
| 29 NewRunnableMethod(this, &ThreadObserverHelper::RegisterObserversTask)); | 30 base::Bind(&ThreadObserverHelper::RegisterObserversTask, this)); |
| 30 done_event_.Wait(); | 31 done_event_.Wait(); |
| 31 } | 32 } |
| 32 | 33 |
| 33 virtual ~ThreadObserverHelper() { | 34 virtual ~ThreadObserverHelper() { |
| 34 DCHECK(content::BrowserThread::CurrentlyOn(id_)); | 35 DCHECK(content::BrowserThread::CurrentlyOn(id_)); |
| 35 registrar_.RemoveAll(); | 36 registrar_.RemoveAll(); |
| 36 } | 37 } |
| 37 | 38 |
| 38 content::NotificationObserverMock* observer() { | 39 content::NotificationObserverMock* observer() { |
| 39 return &observer_; | 40 return &observer_; |
| (...skipping 23 matching lines...) Expand all Loading... |
| 63 DBThreadObserverHelper, | 64 DBThreadObserverHelper, |
| 64 content::BrowserThread::DeleteOnDBThread> DBThreadObserverHelperBase; | 65 content::BrowserThread::DeleteOnDBThread> DBThreadObserverHelperBase; |
| 65 | 66 |
| 66 class DBThreadObserverHelper : public DBThreadObserverHelperBase { | 67 class DBThreadObserverHelper : public DBThreadObserverHelperBase { |
| 67 public: | 68 public: |
| 68 DBThreadObserverHelper() : | 69 DBThreadObserverHelper() : |
| 69 DBThreadObserverHelperBase(content::BrowserThread::DB) {} | 70 DBThreadObserverHelperBase(content::BrowserThread::DB) {} |
| 70 }; | 71 }; |
| 71 | 72 |
| 72 #endif // CHROME_TEST_BASE_THREAD_OBSERVER_HELPER_H_ | 73 #endif // CHROME_TEST_BASE_THREAD_OBSERVER_HELPER_H_ |
| OLD | NEW |