Index: chrome/browser/background/background_application_list_model_unittest.cc |
diff --git a/chrome/browser/background/background_application_list_model_unittest.cc b/chrome/browser/background/background_application_list_model_unittest.cc |
index 2e8690ec3f37da2bdf953c610e5be0fb355b0c7b..30efc20765f5f301daf492bd75d9be8c66d0b9d4 100644 |
--- a/chrome/browser/background/background_application_list_model_unittest.cc |
+++ b/chrome/browser/background/background_application_list_model_unittest.cc |
@@ -84,7 +84,10 @@ std::string GenerateUniqueExtensionName() { |
void AddBackgroundPermission(ExtensionService* service, |
Extension* extension) { |
- if (BackgroundApplicationListModel::IsBackgroundApp(*extension)) return; |
+ if (BackgroundApplicationListModel::IsBackgroundApp(*extension, |
Mihai Parparita -not on Chrome
2012/05/03 00:54:11
Nit: You may want to make this into a IsBackground
Andrew T Wilson (Slow)
2012/05/03 07:42:56
Done.
|
+ service->profile())) { |
+ return; |
+ } |
static scoped_refptr<Extension> temporary = |
CreateExtension(GenerateUniqueExtensionName(), true); |
@@ -96,7 +99,10 @@ void AddBackgroundPermission(ExtensionService* service, |
void RemoveBackgroundPermission(ExtensionService* service, |
Extension* extension) { |
- if (!BackgroundApplicationListModel::IsBackgroundApp(*extension)) return; |
+ if (!BackgroundApplicationListModel::IsBackgroundApp(*extension, |
+ service->profile())) { |
+ return; |
+ } |
extensions::PermissionsUpdater(service->profile()).RemovePermissions( |
extension, extension->GetActivePermissions()); |
} |
@@ -125,45 +131,55 @@ TEST_F(BackgroundApplicationListModelTest, ExplicitTest) { |
ASSERT_EQ(0U, model->size()); |
// Add alternating Extensions and Background Apps |
- ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext1)); |
+ ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext1, |
+ profile_.get())); |
service->AddExtension(ext1); |
ASSERT_EQ(1U, service->extensions()->size()); |
ASSERT_EQ(0U, model->size()); |
- ASSERT_TRUE(BackgroundApplicationListModel::IsBackgroundApp(*bgapp1)); |
+ ASSERT_TRUE(BackgroundApplicationListModel::IsBackgroundApp(*bgapp1, |
+ profile_.get())); |
service->AddExtension(bgapp1); |
ASSERT_EQ(2U, service->extensions()->size()); |
ASSERT_EQ(1U, model->size()); |
- ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext2)); |
+ ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext2, |
+ profile_.get())); |
service->AddExtension(ext2); |
ASSERT_EQ(3U, service->extensions()->size()); |
ASSERT_EQ(1U, model->size()); |
- ASSERT_TRUE(BackgroundApplicationListModel::IsBackgroundApp(*bgapp2)); |
+ ASSERT_TRUE(BackgroundApplicationListModel::IsBackgroundApp(*bgapp2, |
+ profile_.get())); |
service->AddExtension(bgapp2); |
ASSERT_EQ(4U, service->extensions()->size()); |
ASSERT_EQ(2U, model->size()); |
- ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext3)); |
+ ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext3, |
+ profile_.get())); |
service->AddExtension(ext3); |
ASSERT_EQ(5U, service->extensions()->size()); |
ASSERT_EQ(2U, model->size()); |
// Remove in FIFO order. |
- ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext1)); |
+ ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext1, |
+ profile_.get())); |
service->UninstallExtension(ext1->id(), false, NULL); |
ASSERT_EQ(4U, service->extensions()->size()); |
ASSERT_EQ(2U, model->size()); |
- ASSERT_TRUE(BackgroundApplicationListModel::IsBackgroundApp(*bgapp1)); |
+ ASSERT_TRUE(BackgroundApplicationListModel::IsBackgroundApp(*bgapp1, |
+ profile_.get())); |
service->UninstallExtension(bgapp1->id(), false, NULL); |
ASSERT_EQ(3U, service->extensions()->size()); |
ASSERT_EQ(1U, model->size()); |
- ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext2)); |
+ ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext2, |
+ profile_.get())); |
service->UninstallExtension(ext2->id(), false, NULL); |
ASSERT_EQ(2U, service->extensions()->size()); |
ASSERT_EQ(1U, model->size()); |
- ASSERT_TRUE(BackgroundApplicationListModel::IsBackgroundApp(*bgapp2)); |
+ ASSERT_TRUE(BackgroundApplicationListModel::IsBackgroundApp(*bgapp2, |
+ profile_.get())); |
service->UninstallExtension(bgapp2->id(), false, NULL); |
ASSERT_EQ(1U, service->extensions()->size()); |
ASSERT_EQ(0U, model->size()); |
- ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext3)); |
+ ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext3, |
+ profile_.get())); |
service->UninstallExtension(ext3->id(), false, NULL); |
ASSERT_EQ(0U, service->extensions()->size()); |
ASSERT_EQ(0U, model->size()); |
@@ -188,11 +204,13 @@ TEST_F(BackgroundApplicationListModelTest, AddRemovePermissionsTest) { |
ASSERT_EQ(0U, model->size()); |
// Add one (non-background) extension and one background application |
- ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext)); |
+ ASSERT_FALSE(BackgroundApplicationListModel::IsBackgroundApp(*ext, |
+ profile_.get())); |
service->AddExtension(ext); |
ASSERT_EQ(1U, service->extensions()->size()); |
ASSERT_EQ(0U, model->size()); |
- ASSERT_TRUE(BackgroundApplicationListModel::IsBackgroundApp(*bgapp)); |
+ ASSERT_TRUE(BackgroundApplicationListModel::IsBackgroundApp(*bgapp, |
+ profile_.get())); |
service->AddExtension(bgapp); |
ASSERT_EQ(2U, service->extensions()->size()); |
ASSERT_EQ(1U, model->size()); |
@@ -227,7 +245,8 @@ void AddExtension(ExtensionService* service, |
} |
scoped_refptr<Extension> extension = |
CreateExtension(GenerateUniqueExtensionName(), create_background); |
- ASSERT_EQ(BackgroundApplicationListModel::IsBackgroundApp(*extension), |
+ ASSERT_EQ(BackgroundApplicationListModel::IsBackgroundApp(*extension, |
+ service->profile()), |
create_background); |
extensions->insert(extension); |
++*count; |
@@ -258,8 +277,10 @@ void RemoveExtension(ExtensionService* service, |
} |
scoped_refptr<Extension> extension = cursor->get(); |
std::string id = extension->id(); |
- if (BackgroundApplicationListModel::IsBackgroundApp(*extension)) |
+ if (BackgroundApplicationListModel::IsBackgroundApp(*extension, |
+ service->profile())) { |
--*expected; |
+ } |
extensions->erase(cursor); |
--*count; |
ASSERT_EQ(*count, extensions->size()); |
@@ -290,7 +311,8 @@ void TogglePermission(ExtensionService* service, |
} |
scoped_refptr<Extension> extension = cursor->get(); |
std::string id = extension->id(); |
- if (BackgroundApplicationListModel::IsBackgroundApp(*extension)) { |
+ if (BackgroundApplicationListModel::IsBackgroundApp(*extension, |
+ service->profile())) { |
--*expected; |
ASSERT_EQ(*count, extensions->size()); |
RemoveBackgroundPermission(service, extension); |