WebSocket Pepper API: allow to release in completion callbacks

ppapi/tests/test_utils.h

 // Copyright (c) 2012 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 PPAPI_TESTS_TEST_UTILS_H_
 #define PPAPI_TESTS_TEST_UTILS_H_
 
 #include <string>
 
 #include "ppapi/c/dev/ppb_testing_dev.h"
@@ skipping 44 matching lines @@
   bool waiting_;
   bool signalled_;
   // Disable copy and assign.
   NestedEvent(const NestedEvent&);
   NestedEvent& operator=(const NestedEvent&);
 };
 
 enum CallbackType { PP_REQUIRED, PP_OPTIONAL, PP_BLOCKING };
 class TestCompletionCallback {
  public:
+  class Delegate {
+   public:
+    virtual ~Delegate() {}
+    virtual void OnCallback(void* user_data, int32_t result) = 0;
+  };
   explicit TestCompletionCallback(PP_Instance instance);
   // TODO(dmichael): Remove this constructor.
   TestCompletionCallback(PP_Instance instance, bool force_async);
 
   TestCompletionCallback(PP_Instance instance, CallbackType callback_type);
 
+  // Sets a Delegate instance. OnCallback() of this instance will be invoked
+  // when the completion callback is invoked.
+  // The delegate will be reset when Reset() or GetCallback() is called.
+  void SetDelegate(Delegate* delegate) { delegate_ = delegate; }
+
   // Waits for the callback to be called and returns the
   // result. Returns immediately if the callback was previously called
   // and the result wasn't returned (i.e. each result value received
   // by the callback is returned by WaitForResult() once and only
   // once). DEPRECATED: Please use the one below.
   // TODO(dmichael): Remove this one when all the tests are updated.
   int32_t WaitForResult();
 
   // Wait for a result, given the return from the call which took this callback
   // as a parameter. If |result| is PP_OK_COMPLETIONPENDING, WaitForResult will
@@ skipping 65 matching lines @@
   bool wait_for_result_called_;
   // Indicates whether we have already been invoked.
   bool have_result_;
   // The last result received (or PP_OK_COMPLETIONCALLBACK if none).
   int32_t result_;
   CallbackType callback_type_;
   bool post_quit_task_;
   std::string errors_;
   unsigned run_count_;
   PP_Instance instance_;
+  Delegate* delegate_;
 };
 
 // Verifies that the callback didn't record any errors. If the callback is run
 // in an unexpected way (e.g., if it's invoked asynchronously when the call
 // should have blocked), this returns an appropriate error string.
 #define CHECK_CALLBACK_BEHAVIOR(callback) \
 do { \
   if ((callback).failed()) \
     return (callback).errors(); \
 } while (false)
@@ skipping 22 matching lines @@
 #error Please add support for your platform in ppapi/tests/test_utils.h
 #endif
 
 /* These are used to determine POSIX-like implementations vs Windows. */
 #if defined(__linux__) || defined(__APPLE__) || defined(__FreeBSD__) || \
     defined(__OpenBSD__) || defined(__sun) || defined(__native_client__)
 #define PPAPI_POSIX 1
 #endif
 
 #endif  // PPAPI_TESTS_TEST_UTILS_H_