Index: chrome/browser/extensions/updater/extension_updater_unittest.cc |
diff --git a/chrome/browser/extensions/updater/extension_updater_unittest.cc b/chrome/browser/extensions/updater/extension_updater_unittest.cc |
index 825921d65460c235f8834d711f80701ab6311fb9..79d0ed3ad98ea64ba9eed29945ed0d7f69d38759 100644 |
--- a/chrome/browser/extensions/updater/extension_updater_unittest.cc |
+++ b/chrome/browser/extensions/updater/extension_updater_unittest.cc |
@@ -260,13 +260,20 @@ class ServiceForManifestTests : public MockService { |
virtual const Extension* GetExtensionById( |
const std::string& id, bool include_disabled) const OVERRIDE { |
- return extensions_.GetByID(id); |
+ const Extension* result = extensions_.GetByID(id); |
+ if (result || !include_disabled) |
+ return result; |
+ return disabled_extensions_.GetByID(id); |
} |
virtual const ExtensionSet* extensions() const OVERRIDE { |
return &extensions_; |
} |
+ virtual const ExtensionSet* disabled_extensions() const OVERRIDE { |
+ return &disabled_extensions_; |
+ } |
+ |
virtual PendingExtensionManager* pending_extension_manager() OVERRIDE { |
return &pending_extension_manager_; |
} |
@@ -278,8 +285,16 @@ class ServiceForManifestTests : public MockService { |
} |
} |
+ void set_disabled_extensions(ExtensionList disabled_extensions) { |
+ for (ExtensionList::const_iterator it = disabled_extensions.begin(); |
+ it != disabled_extensions.end(); ++it) { |
+ disabled_extensions_.Insert(*it); |
+ } |
+ } |
+ |
private: |
ExtensionSet extensions_; |
+ ExtensionSet disabled_extensions_; |
}; |
class ServiceForDownloadTests : public MockService { |
@@ -1340,14 +1355,26 @@ TEST_F(ExtensionUpdaterTest, TestNonAutoUpdateableLocations) { |
service.CreateTestExtensions(1, 1, &extensions, NULL, Extension::INVALID); |
service.CreateTestExtensions(2, 1, &extensions, NULL, Extension::INTERNAL); |
ASSERT_EQ(2u, extensions.size()); |
- const std::string& id = extensions[1]->id(); |
+ const std::string& enabled_id = extensions[1]->id(); |
+ |
+ ExtensionList disabled_extensions; |
+ service.CreateTestExtensions(3, 1, &disabled_extensions, NULL, |
+ Extension::INVALID); |
+ service.CreateTestExtensions(4, 1, &disabled_extensions, NULL, |
+ Extension::INTERNAL); |
+ ASSERT_EQ(2u, disabled_extensions.size()); |
+ const std::string& disabled_id = disabled_extensions[1]->id(); |
asargent_no_longer_on_chrome
2012/03/20 00:24:01
Since this test ("TestNonAutoUpdateableLocations")
Matt Tytel
2012/03/20 20:34:28
Done.
|
// These expectations fail if the delegate's methods are invoked for the |
// first extension, which has a non-matching id. |
- EXPECT_CALL(delegate, GetUpdateUrlData(id)).WillOnce(Return("")); |
- EXPECT_CALL(delegate, GetPingDataForExtension(id, _)); |
+ EXPECT_CALL(delegate, GetUpdateUrlData(enabled_id)).WillOnce(Return("")); |
+ EXPECT_CALL(delegate, GetPingDataForExtension(enabled_id, _)); |
+ |
+ EXPECT_CALL(delegate, GetUpdateUrlData(disabled_id)).WillOnce(Return("")); |
+ EXPECT_CALL(delegate, GetPingDataForExtension(disabled_id, _)); |
service.set_extensions(extensions); |
+ service.set_disabled_extensions(disabled_extensions); |
updater.set_blacklist_checks_enabled(false); |
updater.Start(); |
updater.CheckNow(); |