chrome/browser/extensions/extension_migrator_unittest.cc

   1 // Copyright 2015 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 "chrome/browser/extensions/extension_migrator.h"
   6
   7 #include "base/files/file_util.h"
   8 #include "base/macros.h"
   9 #include "base/run_loop.h"
  10 #include "chrome/browser/extensions/extension_service.h"
  11 #include "chrome/browser/extensions/extension_service_test_base.h"
  12 #include "chrome/browser/extensions/external_provider_impl.h"
  13 #include "chrome/browser/extensions/pending_extension_manager.h"
  14 #include "chrome/test/base/testing_profile.h"
  15 #include "extensions/browser/extension_registry.h"
  16 #include "extensions/common/extension_builder.h"
  17
  18 namespace extensions {
  19
  20 namespace {
  21
  22 const char kOldId[] = "oooooooooooooooooooooooooooooooo";
  23 const char kNewId[] = "nnnnnnnnnnnnnnnnnnnnnnnnnnnnnnnn";
  24
  25 scoped_refptr<Extension> CreateExtension(const std::string& id) {
  26   return ExtensionBuilder()
  27       .SetManifest(DictionaryBuilder().Set("name", "test").Set(
  28           "version", "0.1"))
  29       .SetID(id)
  30       .Build();
  31 }
  32
  33 }  // namespace
  34
  35 class ExtensionMigratorTest : public ExtensionServiceTestBase {
  36  public:
  37   ExtensionMigratorTest() {}
  38   ~ExtensionMigratorTest() override {}
  39
  40  protected:
  41   void InitWithExistingProfile() {
  42     ExtensionServiceInitParams params = CreateDefaultInitParams();
  43     params.is_first_run = false;
  44     // Create prefs file to make the profile not new.
  45     const char prefs[] = "{}";
  46     EXPECT_EQ(int(sizeof(prefs)),
  47               base::WriteFile(params.pref_file, prefs, sizeof(prefs)));
  48     InitializeExtensionService(params);
  49     service()->Init();
  50     AddMigratorProvider();
  51   }
  52
  53   void AddMigratorProvider() {
  54     service()->AddProviderForTesting(new ExternalProviderImpl(
  55         service(),
  56         new ExtensionMigrator(profile(), kOldId, kNewId),
  57         profile(),
  58         Manifest::EXTERNAL_PREF,
  59         Manifest::EXTERNAL_PREF_DOWNLOAD,
  60         Extension::FROM_WEBSTORE | Extension::WAS_INSTALLED_BY_DEFAULT));
  61   }
  62
  63   void AddExtension(const std::string& id) {
  64     scoped_refptr<Extension> fake_app = CreateExtension(id);
  65     service()->AddExtension(fake_app.get());
  66   }
  67
  68   bool HasNewExtension() {
  69     return service()->pending_extension_manager()->IsIdPending(kNewId) ||
  70            !!registry()->GetInstalledExtension(kNewId);
  71   }
  72
  73  private:
  74   DISALLOW_COPY_AND_ASSIGN(ExtensionMigratorTest);
  75 };
  76
  77 TEST_F(ExtensionMigratorTest, NoExistingOld) {
  78   InitWithExistingProfile();
  79   service()->CheckForExternalUpdates();
  80   base::RunLoop().RunUntilIdle();
  81   EXPECT_FALSE(HasNewExtension());
  82 }
  83
  84 TEST_F(ExtensionMigratorTest, HasExistingOld) {
  85   InitWithExistingProfile();
  86   AddExtension(kOldId);
  87   service()->CheckForExternalUpdates();
  88   base::RunLoop().RunUntilIdle();
  89   EXPECT_TRUE(HasNewExtension());
  90   EXPECT_TRUE(!!registry()->GetInstalledExtension(kOldId));
  91 }
  92
  93 TEST_F(ExtensionMigratorTest, KeepExistingNew) {
  94   InitWithExistingProfile();
  95   AddExtension(kNewId);
  96   service()->CheckForExternalUpdates();
  97   base::RunLoop().RunUntilIdle();
  98   EXPECT_TRUE(!!registry()->GetInstalledExtension(kNewId));
  99 }
 100
 101 TEST_F(ExtensionMigratorTest, HasBothOldAndNew) {
 102   InitWithExistingProfile();
 103   AddExtension(kOldId);
 104   AddExtension(kNewId);
 105   service()->CheckForExternalUpdates();
 106   base::RunLoop().RunUntilIdle();
 107   EXPECT_TRUE(!!registry()->GetInstalledExtension(kOldId));
 108   EXPECT_TRUE(!!registry()->GetInstalledExtension(kNewId));
 109 }
 110
 111 }  // namespace extensions