| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "extensions/browser/runtime_data.h" | 5 #include "extensions/browser/runtime_data.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
| 10 #include "extensions/browser/extension_registry.h" | 10 #include "extensions/browser/extension_registry.h" |
| (...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 46 ExtensionRegistry registry_; | 46 ExtensionRegistry registry_; |
| 47 RuntimeData runtime_data_; | 47 RuntimeData runtime_data_; |
| 48 | 48 |
| 49 private: | 49 private: |
| 50 DISALLOW_COPY_AND_ASSIGN(RuntimeDataTest); | 50 DISALLOW_COPY_AND_ASSIGN(RuntimeDataTest); |
| 51 }; | 51 }; |
| 52 | 52 |
| 53 TEST_F(RuntimeDataTest, IsBackgroundPageReady) { | 53 TEST_F(RuntimeDataTest, IsBackgroundPageReady) { |
| 54 // An extension without a background page is always considered ready. | 54 // An extension without a background page is always considered ready. |
| 55 scoped_refptr<Extension> no_background = CreateExtension(); | 55 scoped_refptr<Extension> no_background = CreateExtension(); |
| 56 EXPECT_TRUE(runtime_data_.IsBackgroundPageReady(no_background)); | 56 EXPECT_TRUE(runtime_data_.IsBackgroundPageReady(no_background.get())); |
| 57 | 57 |
| 58 // An extension with a background page is not ready until the flag is set. | 58 // An extension with a background page is not ready until the flag is set. |
| 59 scoped_refptr<Extension> with_background = | 59 scoped_refptr<Extension> with_background = |
| 60 CreateExtensionWithBackgroundPage(); | 60 CreateExtensionWithBackgroundPage(); |
| 61 EXPECT_FALSE(runtime_data_.IsBackgroundPageReady(with_background)); | 61 EXPECT_FALSE(runtime_data_.IsBackgroundPageReady(with_background.get())); |
| 62 | 62 |
| 63 // The flag can be toggled. | 63 // The flag can be toggled. |
| 64 runtime_data_.SetBackgroundPageReady(with_background, true); | 64 runtime_data_.SetBackgroundPageReady(with_background.get(), true); |
| 65 EXPECT_TRUE(runtime_data_.IsBackgroundPageReady(with_background)); | 65 EXPECT_TRUE(runtime_data_.IsBackgroundPageReady(with_background.get())); |
| 66 runtime_data_.SetBackgroundPageReady(with_background, false); | 66 runtime_data_.SetBackgroundPageReady(with_background.get(), false); |
| 67 EXPECT_FALSE(runtime_data_.IsBackgroundPageReady(with_background)); | 67 EXPECT_FALSE(runtime_data_.IsBackgroundPageReady(with_background.get())); |
| 68 } | 68 } |
| 69 | 69 |
| 70 TEST_F(RuntimeDataTest, IsBeingUpgraded) { | 70 TEST_F(RuntimeDataTest, IsBeingUpgraded) { |
| 71 scoped_refptr<Extension> extension = CreateExtension(); | 71 scoped_refptr<Extension> extension = CreateExtension(); |
| 72 | 72 |
| 73 // An extension is not being upgraded until the flag is set. | 73 // An extension is not being upgraded until the flag is set. |
| 74 EXPECT_FALSE(runtime_data_.IsBeingUpgraded(extension)); | 74 EXPECT_FALSE(runtime_data_.IsBeingUpgraded(extension.get())); |
| 75 | 75 |
| 76 // The flag can be toggled. | 76 // The flag can be toggled. |
| 77 runtime_data_.SetBeingUpgraded(extension, true); | 77 runtime_data_.SetBeingUpgraded(extension.get(), true); |
| 78 EXPECT_TRUE(runtime_data_.IsBeingUpgraded(extension)); | 78 EXPECT_TRUE(runtime_data_.IsBeingUpgraded(extension.get())); |
| 79 runtime_data_.SetBeingUpgraded(extension, false); | 79 runtime_data_.SetBeingUpgraded(extension.get(), false); |
| 80 EXPECT_FALSE(runtime_data_.IsBeingUpgraded(extension)); | 80 EXPECT_FALSE(runtime_data_.IsBeingUpgraded(extension.get())); |
| 81 } | 81 } |
| 82 | 82 |
| 83 TEST_F(RuntimeDataTest, HasUsedWebRequest) { | 83 TEST_F(RuntimeDataTest, HasUsedWebRequest) { |
| 84 scoped_refptr<Extension> extension = CreateExtension(); | 84 scoped_refptr<Extension> extension = CreateExtension(); |
| 85 | 85 |
| 86 // An extension has not used web request until the flag is set. | 86 // An extension has not used web request until the flag is set. |
| 87 EXPECT_FALSE(runtime_data_.HasUsedWebRequest(extension)); | 87 EXPECT_FALSE(runtime_data_.HasUsedWebRequest(extension.get())); |
| 88 | 88 |
| 89 // The flag can be toggled. | 89 // The flag can be toggled. |
| 90 runtime_data_.SetHasUsedWebRequest(extension, true); | 90 runtime_data_.SetHasUsedWebRequest(extension.get(), true); |
| 91 EXPECT_TRUE(runtime_data_.HasUsedWebRequest(extension)); | 91 EXPECT_TRUE(runtime_data_.HasUsedWebRequest(extension.get())); |
| 92 runtime_data_.SetHasUsedWebRequest(extension, false); | 92 runtime_data_.SetHasUsedWebRequest(extension.get(), false); |
| 93 EXPECT_FALSE(runtime_data_.HasUsedWebRequest(extension)); | 93 EXPECT_FALSE(runtime_data_.HasUsedWebRequest(extension.get())); |
| 94 } | 94 } |
| 95 | 95 |
| 96 // Unloading an extension stops tracking it. | 96 // Unloading an extension stops tracking it. |
| 97 TEST_F(RuntimeDataTest, OnExtensionUnloaded) { | 97 TEST_F(RuntimeDataTest, OnExtensionUnloaded) { |
| 98 scoped_refptr<Extension> extension = CreateExtensionWithBackgroundPage(); | 98 scoped_refptr<Extension> extension = CreateExtensionWithBackgroundPage(); |
| 99 runtime_data_.SetBackgroundPageReady(extension, true); | 99 runtime_data_.SetBackgroundPageReady(extension.get(), true); |
| 100 ASSERT_TRUE(runtime_data_.HasExtensionForTesting(extension)); | 100 ASSERT_TRUE(runtime_data_.HasExtensionForTesting(extension.get())); |
| 101 | 101 |
| 102 runtime_data_.OnExtensionUnloaded( | 102 runtime_data_.OnExtensionUnloaded( |
| 103 NULL, extension, UnloadedExtensionInfo::REASON_DISABLE); | 103 NULL, extension.get(), UnloadedExtensionInfo::REASON_DISABLE); |
| 104 EXPECT_FALSE(runtime_data_.HasExtensionForTesting(extension)); | 104 EXPECT_FALSE(runtime_data_.HasExtensionForTesting(extension.get())); |
| 105 } | 105 } |
| 106 | 106 |
| 107 } // namespace | 107 } // namespace |
| 108 } // namespace extensions | 108 } // namespace extensions |
| OLD | NEW |