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