1 files changed, 6 insertions, 0 deletions
diff --git a/system/gd/module.h b/system/gd/module.h
index 816d455482..57e4db7d16 100644
--- a/system/gd/module.h
+++ b/system/gd/module.h
@@ -32,6 +32,9 @@
 #include "os/thread.h"
 
 namespace bluetooth {
+// Timeout for waiting for a handler to stop, used in Handler::WaitUntilStopped()
+constexpr std::chrono::milliseconds kHandlerStopTimeout = std::chrono::milliseconds(2000);
+
 namespace shim {
 class Stack;
 }  // namespace shim
@@ -191,6 +194,9 @@ public:
   os::Thread& GetTestThread() { return test_thread; }
   os::Handler* GetTestHandler() { return test_handler_; }
 
+  // Override the StopAll method to use the test thread and handler.
+  void StopAll();
+
   bool SynchronizeModuleHandler(const ModuleFactory* module,
                                 std::chrono::milliseconds timeout) const {
     return SynchronizeHandler(GetTestModuleHandler(module), timeout);