Add an extensions keep-alive client.

extensions/renderer/api_test_base.h

 // Copyright 2014 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 EXTENSIONS_RENDERER_API_TEST_BASE_H_
 #define EXTENSIONS_RENDERER_API_TEST_BASE_H_
 
 #include <map>
 #include <string>
 #include <utility>
 
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
+#include "extensions/common/mojo/keep_alive.mojom.h"
 #include "extensions/renderer/module_system_test.h"
 #include "extensions/renderer/v8_schema_registry.h"
 #include "gin/handle.h"
 #include "gin/modules/module_registry.h"
 #include "gin/object_template_builder.h"
 #include "gin/wrappable.h"
 #include "mojo/bindings/js/handle.h"
 #include "mojo/public/cpp/bindings/interface_request.h"
 #include "mojo/public/cpp/system/core.h"
 
@@ skipping 30 matching lines @@
   static void ForwardToServiceFactory(
       const base::Callback<void(mojo::InterfaceRequest<Interface>)>
           service_factory,
       mojo::ScopedMessagePipeHandle handle) {
     service_factory.Run(mojo::MakeRequest<Interface>(handle.Pass()));
   }
   std::map<std::string, base::Callback<void(mojo::ScopedMessagePipeHandle)> >
       service_factories_;
 };
 
+// A counter for the number of keep alives requested.
+class KeepAliveCounter {

[Ken Rockot(use gerrit already), 2014/10/24 18:29:54]

nit: What do you think about moving KeepAliveCounter and maybe
TestServiceProvider into their own source files? Doing so might also warrant
introducing a new //extensions/renderer/test/ home to keep them separated from
everything else. 

[OR we could invert this and like //content have an //extensions/test/ (and
maybe even subpaths for {browser, common, renderer}), but I don't want to fret
too much over this detail for your CL]

[Sam McNally, 2014/10/27 05:07:24]

TestServiceProvider is fairly tightly coupled to ApiTestBase so I think it's
reasonable for it to live here. Removed KeepAliveCounter.

+ public:
+  KeepAliveCounter();
+
+  void CreateKeepAlive(mojo::InterfaceRequest<KeepAlive> request);
+  void WaitUntilKeepAlivesFinish();
+
+  int keep_alives() { return keep_alives_; }
+  int keep_alive_ends() { return keep_alive_ends_; }
+
+ private:
+  class Instance;
+
+  void PollKeepAlives(const base::Closure& quit_closure);
+
+  int keep_alives_;
+  int keep_alive_ends_;
+
+  DISALLOW_COPY_AND_ASSIGN(KeepAliveCounter);
+};
+
 // A base class for unit testing apps/extensions API custom bindings implemented
 // on Mojo services. To use:
 // 1. Register test Mojo service implementations on service_provider().
 // 2. Write JS tests in extensions/test/data/test_file.js.
 // 3. Write one C++ test function for each JS test containing
 //    RunTest("test_file.js", "testFunctionName").
 // See extensions/renderer/api_test_base_unittest.cc and
 // extensions/test/data/api_test_base_unittest.js for sample usage.
 class ApiTestBase : public ModuleSystemTest {
  protected:
   ApiTestBase();
   virtual ~ApiTestBase();
   virtual void SetUp() override;
   void RunTest(const std::string& file_name, const std::string& test_name);
   TestServiceProvider* service_provider() { return service_provider_; }
+  KeepAliveCounter& keep_alive_counter() { return keep_alive_counter_; }
 
  private:
   void RegisterModules();
   void InitializeEnvironment();
   void RunTestInner(const std::string& test_name,
                     const base::Closure& quit_closure);
   void RunPromisesAgain();
 
   base::MessageLoop message_loop_;
   TestServiceProvider* service_provider_;
   scoped_ptr<V8SchemaRegistry> v8_schema_registry_;
+  KeepAliveCounter keep_alive_counter_;
 };
 
 }  // namespace extensions
 
 #endif  // EXTENSIONS_RENDERER_API_TEST_BASE_H_