| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "chrome/browser/extensions/user_script_master.h" | 5 #include "chrome/browser/extensions/user_script_master.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/file_util.h" | 9 #include "base/file_util.h" |
| 10 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
| (...skipping 22 matching lines...) Expand all Loading... |
| 33 | 33 |
| 34 namespace extensions { | 34 namespace extensions { |
| 35 | 35 |
| 36 // Test bringing up a master on a specific directory, putting a script | 36 // Test bringing up a master on a specific directory, putting a script |
| 37 // in there, etc. | 37 // in there, etc. |
| 38 | 38 |
| 39 class UserScriptMasterTest : public testing::Test, | 39 class UserScriptMasterTest : public testing::Test, |
| 40 public content::NotificationObserver { | 40 public content::NotificationObserver { |
| 41 public: | 41 public: |
| 42 UserScriptMasterTest() | 42 UserScriptMasterTest() |
| 43 : message_loop_(MessageLoop::TYPE_UI), | 43 : message_loop_(base::MessageLoop::TYPE_UI), |
| 44 shared_memory_(NULL) { | 44 shared_memory_(NULL) { |
| 45 } | 45 } |
| 46 | 46 |
| 47 virtual void SetUp() { | 47 virtual void SetUp() { |
| 48 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 48 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
| 49 | 49 |
| 50 // Register for all user script notifications. | 50 // Register for all user script notifications. |
| 51 registrar_.Add(this, chrome::NOTIFICATION_USER_SCRIPTS_UPDATED, | 51 registrar_.Add(this, chrome::NOTIFICATION_USER_SCRIPTS_UPDATED, |
| 52 content::NotificationService::AllSources()); | 52 content::NotificationService::AllSources()); |
| 53 | 53 |
| 54 // UserScriptMaster posts tasks to the file thread so make the current | 54 // UserScriptMaster posts tasks to the file thread so make the current |
| 55 // thread look like one. | 55 // thread look like one. |
| 56 file_thread_.reset(new content::TestBrowserThread( | 56 file_thread_.reset(new content::TestBrowserThread( |
| 57 BrowserThread::FILE, MessageLoop::current())); | 57 BrowserThread::FILE, base::MessageLoop::current())); |
| 58 ui_thread_.reset(new content::TestBrowserThread( | 58 ui_thread_.reset(new content::TestBrowserThread( |
| 59 BrowserThread::UI, MessageLoop::current())); | 59 BrowserThread::UI, base::MessageLoop::current())); |
| 60 } | 60 } |
| 61 | 61 |
| 62 virtual void TearDown() { | 62 virtual void TearDown() { |
| 63 file_thread_.reset(); | 63 file_thread_.reset(); |
| 64 ui_thread_.reset(); | 64 ui_thread_.reset(); |
| 65 } | 65 } |
| 66 | 66 |
| 67 virtual void Observe(int type, | 67 virtual void Observe(int type, |
| 68 const content::NotificationSource& source, | 68 const content::NotificationSource& source, |
| 69 const content::NotificationDetails& details) OVERRIDE { | 69 const content::NotificationDetails& details) OVERRIDE { |
| 70 DCHECK(type == chrome::NOTIFICATION_USER_SCRIPTS_UPDATED); | 70 DCHECK(type == chrome::NOTIFICATION_USER_SCRIPTS_UPDATED); |
| 71 | 71 |
| 72 shared_memory_ = content::Details<base::SharedMemory>(details).ptr(); | 72 shared_memory_ = content::Details<base::SharedMemory>(details).ptr(); |
| 73 if (MessageLoop::current() == &message_loop_) | 73 if (base::MessageLoop::current() == &message_loop_) |
| 74 MessageLoop::current()->Quit(); | 74 base::MessageLoop::current()->Quit(); |
| 75 } | 75 } |
| 76 | 76 |
| 77 // Directory containing user scripts. | 77 // Directory containing user scripts. |
| 78 base::ScopedTempDir temp_dir_; | 78 base::ScopedTempDir temp_dir_; |
| 79 | 79 |
| 80 content::NotificationRegistrar registrar_; | 80 content::NotificationRegistrar registrar_; |
| 81 | 81 |
| 82 // MessageLoop used in tests. | 82 // MessageLoop used in tests. |
| 83 MessageLoop message_loop_; | 83 base::MessageLoop message_loop_; |
| 84 | 84 |
| 85 scoped_ptr<content::TestBrowserThread> file_thread_; | 85 scoped_ptr<content::TestBrowserThread> file_thread_; |
| 86 scoped_ptr<content::TestBrowserThread> ui_thread_; | 86 scoped_ptr<content::TestBrowserThread> ui_thread_; |
| 87 | 87 |
| 88 // Updated to the script shared memory when we get notified. | 88 // Updated to the script shared memory when we get notified. |
| 89 base::SharedMemory* shared_memory_; | 89 base::SharedMemory* shared_memory_; |
| 90 }; | 90 }; |
| 91 | 91 |
| 92 // Test that we get notified even when there are no scripts. | 92 // Test that we get notified even when there are no scripts. |
| 93 TEST_F(UserScriptMasterTest, NoScripts) { | 93 TEST_F(UserScriptMasterTest, NoScripts) { |
| 94 TestingProfile profile; | 94 TestingProfile profile; |
| 95 scoped_refptr<UserScriptMaster> master(new UserScriptMaster(&profile)); | 95 scoped_refptr<UserScriptMaster> master(new UserScriptMaster(&profile)); |
| 96 master->StartLoad(); | 96 master->StartLoad(); |
| 97 message_loop_.PostTask(FROM_HERE, MessageLoop::QuitClosure()); | 97 message_loop_.PostTask(FROM_HERE, base::MessageLoop::QuitClosure()); |
| 98 message_loop_.Run(); | 98 message_loop_.Run(); |
| 99 | 99 |
| 100 ASSERT_TRUE(shared_memory_ != NULL); | 100 ASSERT_TRUE(shared_memory_ != NULL); |
| 101 } | 101 } |
| 102 | 102 |
| 103 TEST_F(UserScriptMasterTest, Parse1) { | 103 TEST_F(UserScriptMasterTest, Parse1) { |
| 104 const std::string text( | 104 const std::string text( |
| 105 "// This is my awesome script\n" | 105 "// This is my awesome script\n" |
| 106 "// It does stuff.\n" | 106 "// It does stuff.\n" |
| 107 "// ==UserScript== trailing garbage\n" | 107 "// ==UserScript== trailing garbage\n" |
| (...skipping 162 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 270 UserScriptMaster::ScriptReloader* script_reloader = | 270 UserScriptMaster::ScriptReloader* script_reloader = |
| 271 new UserScriptMaster::ScriptReloader(NULL); | 271 new UserScriptMaster::ScriptReloader(NULL); |
| 272 script_reloader->AddRef(); | 272 script_reloader->AddRef(); |
| 273 script_reloader->LoadUserScripts(&user_scripts); | 273 script_reloader->LoadUserScripts(&user_scripts); |
| 274 script_reloader->Release(); | 274 script_reloader->Release(); |
| 275 | 275 |
| 276 EXPECT_EQ(content, user_scripts[0].js_scripts()[0].GetContent().as_string()); | 276 EXPECT_EQ(content, user_scripts[0].js_scripts()[0].GetContent().as_string()); |
| 277 } | 277 } |
| 278 | 278 |
| 279 } // namespace extensions | 279 } // namespace extensions |
| OLD | NEW |