blob: 42e4eccddfc26af146ca5c55a032eb19b3425753 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
// Copyright (c) 2013 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 "base/bind.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/chrome_notification_types.h"
#include "chrome/browser/extensions/extension_browsertest.h"
#include "chrome/browser/extensions/extension_host.h"
#include "chrome/browser/extensions/extension_process_manager.h"
#include "chrome/browser/extensions/extension_service.h"
#include "chrome/browser/extensions/extension_system.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_commands.h"
#include "chrome/test/base/ui_test_utils.h"
#include "content/public/browser/render_process_host.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_test_utils.h"
using content::RenderViewHost;
using content::WebContents;
using extensions::Extension;
class GtalkExtensionTest : public ExtensionBrowserTest {
protected:
ExtensionProcessManager* GetExtensionProcessManager() {
return extensions::ExtensionSystem::Get(browser()->profile())->
process_manager();
}
void InstallGtalkExtension(const std::string& version) {
const Extension* extension = InstallExtensionWithUIAutoConfirm(
test_data_dir_.AppendASCII("gtalk").AppendASCII(version + ".crx"),
1, browser());
installed_extension_id_ = extension->id();
}
const std::string& GetInstalledExtensionId() {
return installed_extension_id_;
}
RenderViewHost* GetViewer() {
std::vector<RenderViewHost*> views = GetMatchingViews(GetViewerUrl());
EXPECT_EQ(1U, views.size());
if (views.empty())
return NULL;
return views.front();
}
std::string GetViewerUrl() {
return "chrome-extension://" + GetInstalledExtensionId() + "/viewer.html";
}
std::vector<RenderViewHost*> GetMatchingViews(std::string url_query) {
ExtensionProcessManager* manager = GetExtensionProcessManager();
ExtensionProcessManager::ViewSet all_views = manager->GetAllViews();
std::vector<RenderViewHost*> matching_views;
for (ExtensionProcessManager::ViewSet::const_iterator iter =
all_views.begin(); iter != all_views.end(); ++iter) {
WebContents* web_contents = WebContents::FromRenderViewHost(*iter);
std::string url = web_contents->GetURL().spec();
if (url.find(url_query) != std::string::npos)
matching_views.push_back(*iter);
}
return matching_views;
}
std::string ReadCurrentVersion() {
std::string response;
EXPECT_TRUE(file_util::ReadFileToString(
test_data_dir_.AppendASCII("gtalk").AppendASCII("current_version"),
&response));
return response;
}
private:
std::string installed_extension_id_;
};
IN_PROC_BROWSER_TEST_F(GtalkExtensionTest, InstallCurrent) {
content::WindowedNotificationObserver panel_loaded(
chrome::NOTIFICATION_EXTENSION_VIEW_REGISTERED,
content::NotificationService::AllSources());
InstallGtalkExtension(ReadCurrentVersion());
panel_loaded.Wait();
ASSERT_TRUE(GetViewer());
}
|
