OLD | NEW |
(Empty) | |
| 1 // Copyright (c) 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "content/browser/devtools/shared_worker_devtools_manager.h" |
| 6 |
| 7 #include "base/memory/scoped_ptr.h" |
| 8 #include "base/message_loop/message_loop.h" |
| 9 #include "base/run_loop.h" |
| 10 #include "content/browser/browser_thread_impl.h" |
| 11 #include "content/browser/devtools/devtools_manager_impl.h" |
| 12 #include "content/browser/shared_worker/shared_worker_instance.h" |
| 13 #include "content/browser/worker_host/worker_storage_partition.h" |
| 14 #include "content/public/browser/devtools_agent_host.h" |
| 15 #include "content/public/browser/devtools_client_host.h" |
| 16 #include "content/public/test/test_browser_context.h" |
| 17 #include "testing/gtest/include/gtest/gtest.h" |
| 18 |
| 19 namespace content { |
| 20 namespace { |
| 21 |
| 22 class TestDevToolsClientHost : public DevToolsClientHost { |
| 23 public: |
| 24 TestDevToolsClientHost() {} |
| 25 virtual ~TestDevToolsClientHost() {} |
| 26 virtual void DispatchOnInspectorFrontend(const std::string& message) |
| 27 OVERRIDE {} |
| 28 virtual void InspectedContentsClosing() OVERRIDE {} |
| 29 virtual void ReplacedWithAnotherClient() OVERRIDE {} |
| 30 |
| 31 private: |
| 32 DISALLOW_COPY_AND_ASSIGN(TestDevToolsClientHost); |
| 33 }; |
| 34 } |
| 35 |
| 36 class SharedWorkerDevToolsManagerTest : public testing::Test { |
| 37 public: |
| 38 SharedWorkerDevToolsManagerTest() |
| 39 : ui_thread_(BrowserThread::UI, &message_loop_), |
| 40 browser_context_(new TestBrowserContext()), |
| 41 partition_( |
| 42 new WorkerStoragePartition(browser_context_->GetRequestContext(), |
| 43 NULL, |
| 44 NULL, |
| 45 NULL, |
| 46 NULL, |
| 47 NULL, |
| 48 NULL, |
| 49 NULL)) {} |
| 50 |
| 51 protected: |
| 52 virtual void SetUp() OVERRIDE { |
| 53 manager_ = SharedWorkerDevToolsManager::GetInstance(); |
| 54 } |
| 55 virtual void TearDown() OVERRIDE { |
| 56 SharedWorkerDevToolsManager::GetInstance()->ResetForTesting(); |
| 57 } |
| 58 |
| 59 void CheckWorkerState(int worker_process_id, |
| 60 int worker_route_id, |
| 61 SharedWorkerDevToolsManager::WorkerState state) { |
| 62 const SharedWorkerDevToolsManager::WorkerId id(worker_process_id, |
| 63 worker_route_id); |
| 64 SharedWorkerDevToolsManager::WorkerInfoMap::iterator it = |
| 65 manager_->workers_.find(id); |
| 66 EXPECT_TRUE(manager_->workers_.end() != it); |
| 67 EXPECT_EQ(state, it->second->state()); |
| 68 } |
| 69 |
| 70 void CheckWorkerNotExist(int worker_process_id, int worker_route_id) { |
| 71 const SharedWorkerDevToolsManager::WorkerId id(worker_process_id, |
| 72 worker_route_id); |
| 73 EXPECT_TRUE(manager_->workers_.end() == manager_->workers_.find(id)); |
| 74 } |
| 75 |
| 76 void CheckWorkerCount(size_t size) { |
| 77 EXPECT_EQ(size, manager_->workers_.size()); |
| 78 } |
| 79 |
| 80 void RegisterDevToolsClientHostFor(DevToolsAgentHost* agent_host, |
| 81 DevToolsClientHost* client_host) { |
| 82 DevToolsManagerImpl::GetInstance()->RegisterDevToolsClientHostFor( |
| 83 agent_host, client_host); |
| 84 } |
| 85 |
| 86 void ClientHostClosing(DevToolsClientHost* client_host) { |
| 87 DevToolsManagerImpl::GetInstance()->ClientHostClosing(client_host); |
| 88 } |
| 89 |
| 90 base::MessageLoopForIO message_loop_; |
| 91 BrowserThreadImpl ui_thread_; |
| 92 scoped_ptr<TestBrowserContext> browser_context_; |
| 93 scoped_ptr<WorkerStoragePartition> partition_; |
| 94 SharedWorkerDevToolsManager* manager_; |
| 95 }; |
| 96 |
| 97 TEST_F(SharedWorkerDevToolsManagerTest, BasicTest) { |
| 98 scoped_refptr<DevToolsAgentHost> agent_host; |
| 99 |
| 100 SharedWorkerInstance instance1(GURL("http://example.com/w.js"), |
| 101 base::string16(), |
| 102 base::string16(), |
| 103 blink::WebContentSecurityPolicyTypeReport, |
| 104 browser_context_->GetResourceContext(), |
| 105 *partition_.get()); |
| 106 |
| 107 agent_host = manager_->GetDevToolsAgentHostForWorker(1, 1); |
| 108 EXPECT_FALSE(agent_host.get()); |
| 109 |
| 110 // Created -> Started -> Destroyed |
| 111 CheckWorkerNotExist(1, 1); |
| 112 manager_->WorkerCreated(1, 1, instance1); |
| 113 CheckWorkerState(1, 1, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 114 manager_->WorkerContextStarted(1, 1); |
| 115 CheckWorkerState(1, 1, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 116 manager_->WorkerDestroyed(1, 1); |
| 117 CheckWorkerNotExist(1, 1); |
| 118 |
| 119 // Created -> GetDevToolsAgentHost -> Started -> Destroyed |
| 120 CheckWorkerNotExist(1, 2); |
| 121 manager_->WorkerCreated(1, 2, instance1); |
| 122 CheckWorkerState(1, 2, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 123 agent_host = manager_->GetDevToolsAgentHostForWorker(1, 2); |
| 124 EXPECT_TRUE(agent_host.get()); |
| 125 CheckWorkerState(1, 2, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 126 EXPECT_EQ(agent_host.get(), manager_->GetDevToolsAgentHostForWorker(1, 2)); |
| 127 manager_->WorkerContextStarted(1, 2); |
| 128 CheckWorkerState(1, 2, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 129 manager_->WorkerDestroyed(1, 2); |
| 130 CheckWorkerNotExist(1, 2); |
| 131 agent_host = NULL; |
| 132 |
| 133 // Created -> Started -> GetDevToolsAgentHost -> Destroyed |
| 134 CheckWorkerNotExist(1, 3); |
| 135 manager_->WorkerCreated(1, 3, instance1); |
| 136 CheckWorkerState(1, 3, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 137 manager_->WorkerContextStarted(1, 3); |
| 138 CheckWorkerState(1, 3, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 139 agent_host = manager_->GetDevToolsAgentHostForWorker(1, 3); |
| 140 EXPECT_TRUE(agent_host.get()); |
| 141 CheckWorkerState(1, 3, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 142 manager_->WorkerDestroyed(1, 3); |
| 143 CheckWorkerNotExist(1, 3); |
| 144 agent_host = NULL; |
| 145 |
| 146 // Created -> Destroyed |
| 147 CheckWorkerNotExist(1, 4); |
| 148 manager_->WorkerCreated(1, 4, instance1); |
| 149 CheckWorkerState(1, 4, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 150 manager_->WorkerDestroyed(1, 4); |
| 151 CheckWorkerNotExist(1, 4); |
| 152 |
| 153 // Created -> GetDevToolsAgentHost -> Destroyed |
| 154 CheckWorkerNotExist(1, 5); |
| 155 manager_->WorkerCreated(1, 5, instance1); |
| 156 CheckWorkerState(1, 5, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 157 agent_host = manager_->GetDevToolsAgentHostForWorker(1, 5); |
| 158 EXPECT_TRUE(agent_host.get()); |
| 159 CheckWorkerState(1, 5, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 160 manager_->WorkerDestroyed(1, 5); |
| 161 CheckWorkerNotExist(1, 5); |
| 162 agent_host = NULL; |
| 163 |
| 164 // Created -> GetDevToolsAgentHost -> Free agent_host -> Destroyed |
| 165 CheckWorkerNotExist(1, 6); |
| 166 manager_->WorkerCreated(1, 6, instance1); |
| 167 CheckWorkerState(1, 6, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 168 agent_host = manager_->GetDevToolsAgentHostForWorker(1, 6); |
| 169 EXPECT_TRUE(agent_host.get()); |
| 170 CheckWorkerState(1, 6, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 171 agent_host = NULL; |
| 172 manager_->WorkerDestroyed(1, 6); |
| 173 CheckWorkerNotExist(1, 6); |
| 174 } |
| 175 |
| 176 TEST_F(SharedWorkerDevToolsManagerTest, AttachTest) { |
| 177 scoped_refptr<DevToolsAgentHost> agent_host1; |
| 178 scoped_refptr<DevToolsAgentHost> agent_host2; |
| 179 |
| 180 SharedWorkerInstance instance1(GURL("http://example.com/w1.js"), |
| 181 base::string16(), |
| 182 base::string16(), |
| 183 blink::WebContentSecurityPolicyTypeReport, |
| 184 browser_context_->GetResourceContext(), |
| 185 *partition_.get()); |
| 186 SharedWorkerInstance instance2(GURL("http://example.com/w2.js"), |
| 187 base::string16(), |
| 188 base::string16(), |
| 189 blink::WebContentSecurityPolicyTypeReport, |
| 190 browser_context_->GetResourceContext(), |
| 191 *partition_.get()); |
| 192 |
| 193 // Created -> GetDevToolsAgentHost -> Register -> Started -> Destroyed |
| 194 scoped_ptr<TestDevToolsClientHost> client_host1(new TestDevToolsClientHost()); |
| 195 CheckWorkerNotExist(2, 1); |
| 196 manager_->WorkerCreated(2, 1, instance1); |
| 197 CheckWorkerState(2, 1, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 198 agent_host1 = manager_->GetDevToolsAgentHostForWorker(2, 1); |
| 199 EXPECT_TRUE(agent_host1.get()); |
| 200 CheckWorkerState(2, 1, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 201 EXPECT_EQ(agent_host1.get(), manager_->GetDevToolsAgentHostForWorker(2, 1)); |
| 202 RegisterDevToolsClientHostFor(agent_host1.get(), client_host1.get()); |
| 203 CheckWorkerState(2, 1, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 204 manager_->WorkerContextStarted(2, 1); |
| 205 CheckWorkerState(2, 1, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 206 manager_->WorkerDestroyed(2, 1); |
| 207 CheckWorkerState(2, 1, SharedWorkerDevToolsManager::WORKER_TERMINATED); |
| 208 EXPECT_EQ(agent_host1.get(), manager_->GetDevToolsAgentHostForWorker(2, 1)); |
| 209 |
| 210 // Created -> Started -> GetDevToolsAgentHost -> Register -> Destroyed |
| 211 scoped_ptr<TestDevToolsClientHost> client_host2(new TestDevToolsClientHost()); |
| 212 manager_->WorkerCreated(2, 2, instance2); |
| 213 CheckWorkerState(2, 2, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 214 manager_->WorkerContextStarted(2, 2); |
| 215 CheckWorkerState(2, 2, SharedWorkerDevToolsManager::WORKER_UNINSPECTED); |
| 216 agent_host2 = manager_->GetDevToolsAgentHostForWorker(2, 2); |
| 217 EXPECT_TRUE(agent_host2.get()); |
| 218 EXPECT_NE(agent_host1.get(), agent_host2.get()); |
| 219 EXPECT_EQ(agent_host2.get(), manager_->GetDevToolsAgentHostForWorker(2, 2)); |
| 220 CheckWorkerState(2, 2, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 221 RegisterDevToolsClientHostFor(agent_host2.get(), client_host2.get()); |
| 222 CheckWorkerState(2, 2, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 223 manager_->WorkerDestroyed(2, 2); |
| 224 CheckWorkerState(2, 2, SharedWorkerDevToolsManager::WORKER_TERMINATED); |
| 225 EXPECT_EQ(agent_host2.get(), manager_->GetDevToolsAgentHostForWorker(2, 2)); |
| 226 |
| 227 // Re-created -> Started -> ClientHostClosing -> Destroyed |
| 228 CheckWorkerState(2, 1, SharedWorkerDevToolsManager::WORKER_TERMINATED); |
| 229 manager_->WorkerCreated(2, 3, instance1); |
| 230 CheckWorkerNotExist(2, 1); |
| 231 CheckWorkerState(2, 3, SharedWorkerDevToolsManager::WORKER_PAUSED); |
| 232 EXPECT_EQ(agent_host1.get(), manager_->GetDevToolsAgentHostForWorker(2, 3)); |
| 233 manager_->WorkerContextStarted(2, 3); |
| 234 CheckWorkerState(2, 3, SharedWorkerDevToolsManager::WORKER_INSPECTED); |
| 235 ClientHostClosing(client_host1.get()); |
| 236 manager_->WorkerDestroyed(2, 3); |
| 237 CheckWorkerNotExist(2, 3); |
| 238 agent_host1 = NULL; |
| 239 |
| 240 // Re-created -> Destroyed |
| 241 CheckWorkerState(2, 2, SharedWorkerDevToolsManager::WORKER_TERMINATED); |
| 242 manager_->WorkerCreated(2, 4, instance2); |
| 243 CheckWorkerNotExist(2, 2); |
| 244 CheckWorkerState(2, 4, SharedWorkerDevToolsManager::WORKER_PAUSED); |
| 245 EXPECT_EQ(agent_host2.get(), manager_->GetDevToolsAgentHostForWorker(2, 4)); |
| 246 manager_->WorkerDestroyed(2, 4); |
| 247 CheckWorkerNotExist(2, 4); |
| 248 CheckWorkerState(2, 2, SharedWorkerDevToolsManager::WORKER_TERMINATED); |
| 249 |
| 250 // Re-created -> ClientHostClosing -> Destroyed |
| 251 manager_->WorkerCreated(2, 5, instance2); |
| 252 CheckWorkerNotExist(2, 2); |
| 253 CheckWorkerState(2, 5, SharedWorkerDevToolsManager::WORKER_PAUSED); |
| 254 EXPECT_EQ(agent_host2.get(), manager_->GetDevToolsAgentHostForWorker(2, 5)); |
| 255 ClientHostClosing(client_host2.get()); |
| 256 CheckWorkerCount(1); |
| 257 agent_host2 = NULL; |
| 258 CheckWorkerCount(1); |
| 259 manager_->WorkerDestroyed(2, 5); |
| 260 CheckWorkerCount(0); |
| 261 } |
| 262 |
| 263 } // namespace content |
OLD | NEW |