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 "base/bind.h" | 5 #include "base/bind.h" |
6 #include "base/files/file.h" | 6 #include "base/files/file.h" |
7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
8 #include "base/files/file_util.h" | 8 #include "base/files/file_util.h" |
9 #include "base/files/scoped_file.h" | 9 #include "base/files/scoped_file.h" |
10 #include "base/files/scoped_temp_dir.h" | 10 #include "base/files/scoped_temp_dir.h" |
(...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
74 write_flags |= base::File::FLAG_ASYNC; | 74 write_flags |= base::File::FLAG_ASYNC; |
75 #endif | 75 #endif |
76 | 76 |
77 return scoped_ptr<NativeProcessLauncher>(new FakeLauncher( | 77 return scoped_ptr<NativeProcessLauncher>(new FakeLauncher( |
78 read_pipe.Pass(), | 78 read_pipe.Pass(), |
79 base::File(write_file, write_flags))); | 79 base::File(write_file, write_flags))); |
80 } | 80 } |
81 | 81 |
82 virtual void Launch(const GURL& origin, | 82 virtual void Launch(const GURL& origin, |
83 const std::string& native_host_name, | 83 const std::string& native_host_name, |
84 LaunchedCallback callback) const OVERRIDE { | 84 LaunchedCallback callback) const override { |
85 callback.Run(NativeProcessLauncher::RESULT_SUCCESS, | 85 callback.Run(NativeProcessLauncher::RESULT_SUCCESS, |
86 base::kNullProcessHandle, | 86 base::kNullProcessHandle, |
87 read_file_.Pass(), write_file_.Pass()); | 87 read_file_.Pass(), write_file_.Pass()); |
88 } | 88 } |
89 | 89 |
90 private: | 90 private: |
91 mutable base::File read_file_; | 91 mutable base::File read_file_; |
92 mutable base::File write_file_; | 92 mutable base::File write_file_; |
93 }; | 93 }; |
94 | 94 |
95 class NativeMessagingTest : public ::testing::Test, | 95 class NativeMessagingTest : public ::testing::Test, |
96 public NativeMessageProcessHost::Client, | 96 public NativeMessageProcessHost::Client, |
97 public base::SupportsWeakPtr<NativeMessagingTest> { | 97 public base::SupportsWeakPtr<NativeMessagingTest> { |
98 protected: | 98 protected: |
99 NativeMessagingTest() | 99 NativeMessagingTest() |
100 : current_channel_(chrome::VersionInfo::CHANNEL_DEV), | 100 : current_channel_(chrome::VersionInfo::CHANNEL_DEV), |
101 thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP), | 101 thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP), |
102 channel_closed_(false) {} | 102 channel_closed_(false) {} |
103 | 103 |
104 virtual void SetUp() OVERRIDE { | 104 virtual void SetUp() override { |
105 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | 105 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); |
106 } | 106 } |
107 | 107 |
108 virtual void TearDown() OVERRIDE { | 108 virtual void TearDown() override { |
109 if (native_message_process_host_.get()) { | 109 if (native_message_process_host_.get()) { |
110 BrowserThread::DeleteSoon(BrowserThread::IO, FROM_HERE, | 110 BrowserThread::DeleteSoon(BrowserThread::IO, FROM_HERE, |
111 native_message_process_host_.release()); | 111 native_message_process_host_.release()); |
112 } | 112 } |
113 base::RunLoop().RunUntilIdle(); | 113 base::RunLoop().RunUntilIdle(); |
114 } | 114 } |
115 | 115 |
116 virtual void PostMessageFromNativeProcess( | 116 virtual void PostMessageFromNativeProcess( |
117 int port_id, | 117 int port_id, |
118 const std::string& message) OVERRIDE { | 118 const std::string& message) override { |
119 last_message_ = message; | 119 last_message_ = message; |
120 | 120 |
121 // Parse the message. | 121 // Parse the message. |
122 base::Value* parsed = base::JSONReader::Read(message); | 122 base::Value* parsed = base::JSONReader::Read(message); |
123 base::DictionaryValue* dict_value; | 123 base::DictionaryValue* dict_value; |
124 if (parsed && parsed->GetAsDictionary(&dict_value)) { | 124 if (parsed && parsed->GetAsDictionary(&dict_value)) { |
125 last_message_parsed_.reset(dict_value); | 125 last_message_parsed_.reset(dict_value); |
126 } else { | 126 } else { |
127 LOG(ERROR) << "Failed to parse " << message; | 127 LOG(ERROR) << "Failed to parse " << message; |
128 last_message_parsed_.reset(); | 128 last_message_parsed_.reset(); |
129 delete parsed; | 129 delete parsed; |
130 } | 130 } |
131 | 131 |
132 if (run_loop_) | 132 if (run_loop_) |
133 run_loop_->Quit(); | 133 run_loop_->Quit(); |
134 } | 134 } |
135 | 135 |
136 virtual void CloseChannel(int port_id, | 136 virtual void CloseChannel(int port_id, |
137 const std::string& error_message) OVERRIDE { | 137 const std::string& error_message) override { |
138 channel_closed_ = true; | 138 channel_closed_ = true; |
139 if (run_loop_) | 139 if (run_loop_) |
140 run_loop_->Quit(); | 140 run_loop_->Quit(); |
141 } | 141 } |
142 | 142 |
143 protected: | 143 protected: |
144 std::string FormatMessage(const std::string& message) { | 144 std::string FormatMessage(const std::string& message) { |
145 uint32_t length = message.length(); | 145 uint32_t length = message.length(); |
146 return std::string(reinterpret_cast<char*>(&length), 4).append(message); | 146 return std::string(reinterpret_cast<char*>(&length), 4).append(message); |
147 } | 147 } |
(...skipping 164 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
312 ScopedTestNativeMessagingHost::kHostName, 0, false); | 312 ScopedTestNativeMessagingHost::kHostName, 0, false); |
313 ASSERT_TRUE(native_message_process_host_.get()); | 313 ASSERT_TRUE(native_message_process_host_.get()); |
314 run_loop_.reset(new base::RunLoop()); | 314 run_loop_.reset(new base::RunLoop()); |
315 run_loop_->Run(); | 315 run_loop_->Run(); |
316 | 316 |
317 // The host should fail to start. | 317 // The host should fail to start. |
318 ASSERT_TRUE(channel_closed_); | 318 ASSERT_TRUE(channel_closed_); |
319 } | 319 } |
320 | 320 |
321 } // namespace extensions | 321 } // namespace extensions |
OLD | NEW |