Mojo EDK: Introduce MojoQueryHandleSignalsState API

mojo/edk/test/mojo_test_base.h

 // Copyright 2016 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.
 
 #ifndef MOJO_EDK_TEST_MOJO_TEST_BASE_H_
 #define MOJO_EDK_TEST_MOJO_TEST_BASE_H_
 
 #include <memory>
 #include <string>
 #include <utility>
@@ skipping 12 matching lines @@
 namespace mojo {
 namespace edk {
 namespace test {
 
 class MojoTestBase : public testing::Test {
  public:
   MojoTestBase();
   ~MojoTestBase() override;
 
   using LaunchType = MultiprocessTestHelper::LaunchType;
-
- protected:
   using HandlerCallback = base::Callback<void(ScopedMessagePipeHandle)>;
 
   class ClientController {
    public:
     ClientController(const std::string& client_name,
                      MojoTestBase* test,
                      const ProcessErrorCallback& process_error_callback,
                      LaunchType launch_type);
     ~ClientController();
 
@@ skipping 100 matching lines @@
   static void CreateDataPipe(MojoHandle* producer,
                              MojoHandle* consumer,
                              size_t capacity);
 
   // Writes data to a data pipe.
   static void WriteData(MojoHandle producer, const std::string& data);
 
   // Reads data from a data pipe.
   static std::string ReadData(MojoHandle consumer, size_t size);
 
+  // Queries the signals state of |handle|.
+  static MojoHandleSignalsState GetSignalsState(MojoHandle handle);
+
+  // Helper to block the calling thread waiting for signals to be raised.
+  static MojoResult WaitForSignals(MojoHandle handle,
+                                   MojoHandleSignals signals,
+                                   MojoHandleSignalsState* state = nullptr);
+
   void set_launch_type(LaunchType launch_type) { launch_type_ = launch_type; }
 
  private:
   friend class ClientController;
 
   std::vector<std::unique_ptr<ClientController>> clients_;
 
   ProcessErrorCallback process_error_callback_;
 
   LaunchType launch_type_ = LaunchType::CHILD;
@@ skipping 69 matching lines @@
 #else  // !defined(OS_IOS)
 #define DEFINE_TEST_CLIENT_WITH_PIPE(client_name, test_base, pipe_name)
 #define DEFINE_TEST_CLIENT_TEST_WITH_PIPE(client_name, test_base, pipe_name)
 #endif  // !defined(OS_IOS)
 
 }  // namespace test
 }  // namespace edk
 }  // namespace mojo
 
 #endif  // MOJO_EDK_TEST_MOJO_TEST_BASE_H_