extensions/browser/mojo/keep_alive_impl_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "extensions/browser/mojo/keep_alive_impl.h"
   6
   7 #include "base/message_loop/message_loop.h"
   8 #include "base/run_loop.h"
   9 #include "content/public/browser/notification_service.h"
  10 #include "extensions/browser/extension_registry.h"
  11 #include "extensions/browser/extension_system.h"
  12 #include "extensions/browser/extensions_test.h"
  13 #include "extensions/browser/mock_extension_system.h"
  14 #include "extensions/browser/process_manager.h"
  15 #include "extensions/browser/test_extensions_browser_client.h"
  16 #include "extensions/common/extension_builder.h"
  17
  18 namespace extensions {
  19
  20 class KeepAliveTest : public ExtensionsTest {
  21  public:
  22   KeepAliveTest()
  23       : notification_service_(content::NotificationService::Create()) {}
  24   ~KeepAliveTest() override {}
  25
  26   void SetUp() override {
  27     ExtensionsTest::SetUp();
  28     message_loop_.reset(new base::MessageLoop);
  29     browser_client_.reset(new TestExtensionsBrowserClient(browser_context()));
  30     browser_client_->set_extension_system_factory(&extension_system_factory_);
  31     ExtensionsBrowserClient::Set(browser_client_.get());
  32     extension_ =
  33         ExtensionBuilder()
  34             .SetManifest(
  35                  DictionaryBuilder()
  36                      .Set("name", "app")
  37                      .Set("version", "1")
  38                      .Set("manifest_version", 2)
  39                      .Set("app",
  40                           DictionaryBuilder().Set(
  41                               "background",
  42                               DictionaryBuilder().Set(
  43                                   "scripts",
  44                                   ListBuilder().Append("background.js")))))
  45             .SetID("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa")
  46             .Build();
  47   }
  48
  49   void TearDown() override {
  50     message_loop_.reset();
  51     ExtensionsTest::TearDown();
  52   }
  53
  54   void WaitUntilLazyKeepAliveChanges() {
  55     int initial_keep_alive_count = GetKeepAliveCount();
  56     while (GetKeepAliveCount() == initial_keep_alive_count) {
  57       base::RunLoop().RunUntilIdle();
  58     }
  59   }
  60
  61   void CreateKeepAlive(mojo::InterfaceRequest<KeepAlive> request) {
  62     KeepAliveImpl::Create(browser_context(), extension_.get(), request.Pass());
  63   }
  64
  65   const Extension* extension() { return extension_.get(); }
  66
  67   int GetKeepAliveCount() {
  68     return ProcessManager::Get(browser_context())
  69         ->GetLazyKeepaliveCount(extension());
  70   }
  71
  72  private:
  73   scoped_ptr<base::MessageLoop> message_loop_;
  74   MockExtensionSystemFactory<MockExtensionSystem> extension_system_factory_;
  75   scoped_ptr<content::NotificationService> notification_service_;
  76   scoped_refptr<const Extension> extension_;
  77   scoped_ptr<TestExtensionsBrowserClient> browser_client_;
  78
  79   DISALLOW_COPY_AND_ASSIGN(KeepAliveTest);
  80 };
  81
  82 TEST_F(KeepAliveTest, Basic) {
  83   mojo::InterfacePtr<KeepAlive> keep_alive;
  84   CreateKeepAlive(mojo::GetProxy(&keep_alive));
  85   EXPECT_EQ(1, GetKeepAliveCount());
  86
  87   keep_alive.reset();
  88   WaitUntilLazyKeepAliveChanges();
  89   EXPECT_EQ(0, GetKeepAliveCount());
  90 }
  91
  92 TEST_F(KeepAliveTest, TwoKeepAlives) {
  93   mojo::InterfacePtr<KeepAlive> keep_alive;
  94   CreateKeepAlive(mojo::GetProxy(&keep_alive));
  95   EXPECT_EQ(1, GetKeepAliveCount());
  96
  97   mojo::InterfacePtr<KeepAlive> other_keep_alive;
  98   CreateKeepAlive(mojo::GetProxy(&other_keep_alive));
  99   EXPECT_EQ(2, GetKeepAliveCount());
 100
 101   keep_alive.reset();
 102   WaitUntilLazyKeepAliveChanges();
 103   EXPECT_EQ(1, GetKeepAliveCount());
 104
 105   other_keep_alive.reset();
 106   WaitUntilLazyKeepAliveChanges();
 107   EXPECT_EQ(0, GetKeepAliveCount());
 108 }
 109
 110 }  // namespace extensions