| Index: chrome/browser/extensions/test_extension_loader.cc
|
| diff --git a/chrome/browser/extensions/test_extension_loader.cc b/chrome/browser/extensions/test_extension_loader.cc
|
| new file mode 100755
|
| index 0000000000000000000000000000000000000000..d8bacb7bd6ac539971b32684c086c8113d9ad824
|
| --- /dev/null
|
| +++ b/chrome/browser/extensions/test_extension_loader.cc
|
| @@ -0,0 +1,63 @@
|
| +// Copyright (c) 2009 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#include "chrome/browser/extensions/test_extension_loader.h"
|
| +
|
| +#include "base/file_path.h"
|
| +#include "base/message_loop.h"
|
| +#include "chrome/browser/profile.h"
|
| +#include "chrome/browser/extensions/extensions_service.h"
|
| +#include "chrome/common/notification_service.h"
|
| +#include "chrome/test/ui_test_utils.h"
|
| +
|
| +namespace {
|
| +
|
| +// How long to wait for the extension to load before giving up.
|
| +const int kLoadTimeoutMs = 5000;
|
| +
|
| +} // namespace
|
| +
|
| +TestExtensionLoader::TestExtensionLoader(Profile* profile)
|
| + : profile_(profile),
|
| + extension_(NULL) {
|
| + registrar_.Add(this, NotificationType::EXTENSIONS_LOADED,
|
| + NotificationService::AllSources());
|
| +
|
| + profile_->GetExtensionsService()->Init();
|
| + DCHECK(profile_->GetExtensionsService()->extensions()->empty());
|
| +}
|
| +
|
| +Extension* TestExtensionLoader::Load(const char* extension_id,
|
| + const FilePath& path) {
|
| + loading_extension_id_ = extension_id;
|
| +
|
| + // Load the extension.
|
| + profile_->GetExtensionsService()->LoadExtension(path);
|
| +
|
| + // Wait for the load to complete. Stick a QuitTask into the message loop
|
| + // with the timeout so it will exit if the extension never loads.
|
| + extension_ = NULL;
|
| + MessageLoop::current()->PostDelayedTask(FROM_HERE,
|
| + new MessageLoop::QuitTask, kLoadTimeoutMs);
|
| + ui_test_utils::RunMessageLoop();
|
| +
|
| + return extension_;
|
| +}
|
| +
|
| +void TestExtensionLoader::Observe(NotificationType type,
|
| + const NotificationSource& source,
|
| + const NotificationDetails& details) {
|
| + if (type == NotificationType::EXTENSIONS_LOADED) {
|
| + ExtensionList* extensions = Details<ExtensionList>(details).ptr();
|
| + for (size_t i = 0; i < (*extensions).size(); ++i) {
|
| + if ((*extensions)[i]->id() == loading_extension_id_) {
|
| + extension_ = (*extensions)[i];
|
| + MessageLoopForUI::current()->Quit();
|
| + break;
|
| + }
|
| + }
|
| + } else {
|
| + NOTREACHED();
|
| + }
|
| +}
|
|
|