Index: mojo/edk/system/core_test_base.h |
diff --git a/mojo/edk/system/core_test_base.h b/mojo/edk/system/core_test_base.h |
index 80a926b6ddc18e2acdc9fd513de8572a0bc16a27..c2b5ee2eede3da14b0e1f9b7e74d35d52f63841d 100644 |
--- a/mojo/edk/system/core_test_base.h |
+++ b/mojo/edk/system/core_test_base.h |
@@ -15,6 +15,7 @@ namespace mojo { |
namespace system { |
class Core; |
+class Awakable; |
namespace test { |
@@ -62,6 +63,9 @@ class CoreTestBase_MockHandleInfo { |
unsigned GetRemoveAwakableCallCount() const; |
unsigned GetCancelAllAwakablesCallCount() const; |
+ size_t GetAddedAwakableSize() const; |
+ Awakable* GetAddedAwakableAt(unsigned i) const; |
+ |
// For use by |MockDispatcher|: |
void IncrementCtorCallCount(); |
void IncrementDtorCallCount(); |
@@ -78,6 +82,10 @@ class CoreTestBase_MockHandleInfo { |
void IncrementRemoveAwakableCallCount(); |
void IncrementCancelAllAwakablesCallCount(); |
+ void AllowAddAwakable(bool alllow); |
+ bool IsAddAwakableAllowed() const; |
+ void AwakableWasAdded(Awakable*); |
+ |
private: |
mutable base::Lock lock_; // Protects the following members. |
unsigned ctor_call_count_; |
@@ -95,6 +103,9 @@ class CoreTestBase_MockHandleInfo { |
unsigned remove_awakable_call_count_; |
unsigned cancel_all_awakables_call_count_; |
+ bool add_awakable_allowed_; |
+ std::vector<Awakable*> added_awakables_; |
+ |
DISALLOW_COPY_AND_ASSIGN(CoreTestBase_MockHandleInfo); |
}; |