Add an extensions keep-alive client.

extensions/renderer/api_test_base.cc

 // 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.
 
 #include "extensions/renderer/api_test_base.h"
 
 #include <vector>
 
 #include "base/run_loop.h"
 #include "extensions/common/extension_urls.h"
@@ skipping 83 matching lines @@
 }
 
 TestServiceProvider::TestServiceProvider() {
 }
 
 ApiTestBase::ApiTestBase() {
 }
 ApiTestBase::~ApiTestBase() {
 }
 
+class KeepAliveCounter::Instance : public mojo::InterfaceImpl<KeepAlive> {
+ public:
+  Instance(KeepAliveCounter* counter);
+  ~Instance() override;
+
+ private:
+  KeepAliveCounter* counter_;
+
+  DISALLOW_COPY_AND_ASSIGN(Instance);
+};
+
+KeepAliveCounter::KeepAliveCounter() : keep_alives_(0), keep_alive_ends_(0) {
+}
+
+void KeepAliveCounter::CreateKeepAlive(
+    mojo::InterfaceRequest<KeepAlive> request) {
+  keep_alives_++;
+  mojo::BindToRequest(new Instance(this), &request);
+}
+
+void KeepAliveCounter::WaitUntilKeepAlivesFinish() {
+  base::RunLoop run_loop;
+  base::MessageLoop::current()->PostTask(
+      FROM_HERE, base::Bind(&KeepAliveCounter::PollKeepAlives,
+                            base::Unretained(this), run_loop.QuitClosure()));
+  run_loop.Run();
+}
+
+void KeepAliveCounter::PollKeepAlives(const base::Closure& quit_closure) {
+  if (keep_alives_ == keep_alive_ends_) {
+    quit_closure.Run();
+    return;
+  }
+  base::MessageLoop::current()->PostTask(
+      FROM_HERE, base::Bind(&KeepAliveCounter::PollKeepAlives,
+                            base::Unretained(this), quit_closure));
+}
+
+KeepAliveCounter::Instance::Instance(KeepAliveCounter* counter)
+    : counter_(counter) {
+}
+
+KeepAliveCounter::Instance::~Instance() {
+  counter_->keep_alive_ends_++;
+}
+
 void ApiTestBase::SetUp() {
   ModuleSystemTest::SetUp();
   InitializeEnvironment();
   RegisterModules();
 }
 
 void ApiTestBase::RegisterModules() {
   v8_schema_registry_.reset(new V8SchemaRegistry);
   const std::vector<std::pair<std::string, int> > resources =
       Dispatcher::GetJsResources();
@@ skipping 42 matching lines @@
       ->AddBuiltinModule(env()->isolate(),
                          mojo::js::Support::kModuleName,
                          mojo::js::Support::GetModule(env()->isolate()));
   gin::Handle<TestServiceProvider> service_provider =
       TestServiceProvider::Create(env()->isolate());
   service_provider_ = service_provider.get();
   gin::ModuleRegistry::From(env()->context()->v8_context())
       ->AddBuiltinModule(env()->isolate(),
                          "content/public/renderer/service_provider",
                          service_provider.ToV8());
+  service_provider_->AddService(
+      base::Bind(&KeepAliveCounter::CreateKeepAlive,
+                 base::Unretained(&keep_alive_counter_)));
 }
 
 void ApiTestBase::InitializeEnvironment() {
   gin::Dictionary global(env()->isolate(),
                          env()->context()->v8_context()->Global());
   gin::Dictionary navigator(gin::Dictionary::CreateEmpty(env()->isolate()));
   navigator.Set("appVersion", base::StringPiece(""));
   global.Set("navigator", navigator);
   gin::Dictionary chrome(gin::Dictionary::CreateEmpty(env()->isolate()));
   global.Set("chrome", chrome);
@@ skipping 14 matching lines @@
       TestNatives::Create(env()->isolate(), run_loop.QuitClosure()).ToV8());
   base::MessageLoop::current()->PostTask(FROM_HERE,
                                          base::Bind(&ApiTestBase::RunTestInner,
                                                     base::Unretained(this),
                                                     test_name,
                                                     run_loop.QuitClosure()));
   base::MessageLoop::current()->PostTask(
       FROM_HERE,
       base::Bind(&ApiTestBase::RunPromisesAgain, base::Unretained(this)));
   run_loop.Run();
+  keep_alive_counter_.WaitUntilKeepAlivesFinish();
 }
 
 void ApiTestBase::RunTestInner(const std::string& test_name,
                                const base::Closure& quit_closure) {
   v8::HandleScope scope(env()->isolate());
   ModuleSystem::NativesEnabledScope natives_enabled(env()->module_system());
   v8::Handle<v8::Value> result =
       env()->module_system()->CallModuleMethod("testBody", test_name);
   if (!result->IsTrue()) {
     base::MessageLoop::current()->PostTask(FROM_HERE, quit_closure);
     FAIL() << "Failed to run test \"" << test_name << "\"";
   }
 }
 
 void ApiTestBase::RunPromisesAgain() {
   RunResolvedPromises();
   base::MessageLoop::current()->PostTask(
       FROM_HERE,
       base::Bind(&ApiTestBase::RunPromisesAgain, base::Unretained(this)));
 }
 
 }  // namespace extensions