| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium OS Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium OS 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 <string> | 5 #include <string> |
| 6 #include <vector> | 6 #include <vector> |
| 7 | 7 |
| 8 #include <glib.h> | 8 #include <glib.h> |
| 9 | 9 |
| 10 #include "base/string_util.h" | 10 #include "base/string_util.h" |
| (...skipping 142 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 153 // returned. | 153 // returned. |
| 154 bool TestUpdateCheck(PrefsInterface* prefs, | 154 bool TestUpdateCheck(PrefsInterface* prefs, |
| 155 const OmahaRequestParams& params, | 155 const OmahaRequestParams& params, |
| 156 const string& http_response, | 156 const string& http_response, |
| 157 int fail_http_response_code, | 157 int fail_http_response_code, |
| 158 ActionExitCode expected_code, | 158 ActionExitCode expected_code, |
| 159 OmahaResponse* out_response, | 159 OmahaResponse* out_response, |
| 160 vector<char>* out_post_data) { | 160 vector<char>* out_post_data) { |
| 161 GMainLoop* loop = g_main_loop_new(g_main_context_default(), FALSE); | 161 GMainLoop* loop = g_main_loop_new(g_main_context_default(), FALSE); |
| 162 MockHttpFetcher* fetcher = new MockHttpFetcher(http_response.data(), | 162 MockHttpFetcher* fetcher = new MockHttpFetcher(http_response.data(), |
| 163 http_response.size()); | 163 http_response.size(), |
| 164 NULL); |
| 164 if (fail_http_response_code >= 0) { | 165 if (fail_http_response_code >= 0) { |
| 165 fetcher->FailTransfer(fail_http_response_code); | 166 fetcher->FailTransfer(fail_http_response_code); |
| 166 } | 167 } |
| 167 NiceMock<PrefsMock> local_prefs; | 168 NiceMock<PrefsMock> local_prefs; |
| 168 OmahaRequestAction action(prefs ? prefs : &local_prefs, | 169 OmahaRequestAction action(prefs ? prefs : &local_prefs, |
| 169 params, | 170 params, |
| 170 NULL, | 171 NULL, |
| 171 fetcher); | 172 fetcher); |
| 172 OmahaRequestActionTestProcessorDelegate delegate; | 173 OmahaRequestActionTestProcessorDelegate delegate; |
| 173 delegate.loop_ = loop; | 174 delegate.loop_ = loop; |
| (...skipping 19 matching lines...) Expand all Loading... |
| 193 | 194 |
| 194 // Tests Event requests -- they should always succeed. |out_post_data| | 195 // Tests Event requests -- they should always succeed. |out_post_data| |
| 195 // may be null; if non-null, the post-data received by the mock | 196 // may be null; if non-null, the post-data received by the mock |
| 196 // HttpFetcher is returned. | 197 // HttpFetcher is returned. |
| 197 void TestEvent(const OmahaRequestParams& params, | 198 void TestEvent(const OmahaRequestParams& params, |
| 198 OmahaEvent* event, | 199 OmahaEvent* event, |
| 199 const string& http_response, | 200 const string& http_response, |
| 200 vector<char>* out_post_data) { | 201 vector<char>* out_post_data) { |
| 201 GMainLoop* loop = g_main_loop_new(g_main_context_default(), FALSE); | 202 GMainLoop* loop = g_main_loop_new(g_main_context_default(), FALSE); |
| 202 MockHttpFetcher* fetcher = new MockHttpFetcher(http_response.data(), | 203 MockHttpFetcher* fetcher = new MockHttpFetcher(http_response.data(), |
| 203 http_response.size()); | 204 http_response.size(), |
| 205 NULL); |
| 204 NiceMock<PrefsMock> prefs; | 206 NiceMock<PrefsMock> prefs; |
| 205 OmahaRequestAction action(&prefs, params, event, fetcher); | 207 OmahaRequestAction action(&prefs, params, event, fetcher); |
| 206 OmahaRequestActionTestProcessorDelegate delegate; | 208 OmahaRequestActionTestProcessorDelegate delegate; |
| 207 delegate.loop_ = loop; | 209 delegate.loop_ = loop; |
| 208 ActionProcessor processor; | 210 ActionProcessor processor; |
| 209 processor.set_delegate(&delegate); | 211 processor.set_delegate(&delegate); |
| 210 processor.EnqueueAction(&action); | 212 processor.EnqueueAction(&action); |
| 211 | 213 |
| 212 g_timeout_add(0, &StartProcessorInRunLoop, &processor); | 214 g_timeout_add(0, &StartProcessorInRunLoop, &processor); |
| 213 g_main_loop_run(loop); | 215 g_main_loop_run(loop); |
| (...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 260 } | 262 } |
| 261 | 263 |
| 262 TEST(OmahaRequestActionTest, NoOutputPipeTest) { | 264 TEST(OmahaRequestActionTest, NoOutputPipeTest) { |
| 263 const string http_response(GetNoUpdateResponse(OmahaRequestParams::kAppId)); | 265 const string http_response(GetNoUpdateResponse(OmahaRequestParams::kAppId)); |
| 264 | 266 |
| 265 GMainLoop *loop = g_main_loop_new(g_main_context_default(), FALSE); | 267 GMainLoop *loop = g_main_loop_new(g_main_context_default(), FALSE); |
| 266 | 268 |
| 267 NiceMock<PrefsMock> prefs; | 269 NiceMock<PrefsMock> prefs; |
| 268 OmahaRequestAction action(&prefs, kDefaultTestParams, NULL, | 270 OmahaRequestAction action(&prefs, kDefaultTestParams, NULL, |
| 269 new MockHttpFetcher(http_response.data(), | 271 new MockHttpFetcher(http_response.data(), |
| 270 http_response.size())); | 272 http_response.size(), |
| 273 NULL)); |
| 271 OmahaRequestActionTestProcessorDelegate delegate; | 274 OmahaRequestActionTestProcessorDelegate delegate; |
| 272 delegate.loop_ = loop; | 275 delegate.loop_ = loop; |
| 273 ActionProcessor processor; | 276 ActionProcessor processor; |
| 274 processor.set_delegate(&delegate); | 277 processor.set_delegate(&delegate); |
| 275 processor.EnqueueAction(&action); | 278 processor.EnqueueAction(&action); |
| 276 | 279 |
| 277 g_timeout_add(0, &StartProcessorInRunLoop, &processor); | 280 g_timeout_add(0, &StartProcessorInRunLoop, &processor); |
| 278 g_main_loop_run(loop); | 281 g_main_loop_run(loop); |
| 279 g_main_loop_unref(loop); | 282 g_main_loop_unref(loop); |
| 280 EXPECT_FALSE(processor.IsRunning()); | 283 EXPECT_FALSE(processor.IsRunning()); |
| (...skipping 126 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 407 } | 410 } |
| 408 } // namespace {} | 411 } // namespace {} |
| 409 | 412 |
| 410 TEST(OmahaRequestActionTest, TerminateTransferTest) { | 413 TEST(OmahaRequestActionTest, TerminateTransferTest) { |
| 411 string http_response("doesn't matter"); | 414 string http_response("doesn't matter"); |
| 412 GMainLoop *loop = g_main_loop_new(g_main_context_default(), FALSE); | 415 GMainLoop *loop = g_main_loop_new(g_main_context_default(), FALSE); |
| 413 | 416 |
| 414 NiceMock<PrefsMock> prefs; | 417 NiceMock<PrefsMock> prefs; |
| 415 OmahaRequestAction action(&prefs, kDefaultTestParams, NULL, | 418 OmahaRequestAction action(&prefs, kDefaultTestParams, NULL, |
| 416 new MockHttpFetcher(http_response.data(), | 419 new MockHttpFetcher(http_response.data(), |
| 417 http_response.size())); | 420 http_response.size(), |
| 421 NULL)); |
| 418 TerminateEarlyTestProcessorDelegate delegate; | 422 TerminateEarlyTestProcessorDelegate delegate; |
| 419 delegate.loop_ = loop; | 423 delegate.loop_ = loop; |
| 420 ActionProcessor processor; | 424 ActionProcessor processor; |
| 421 processor.set_delegate(&delegate); | 425 processor.set_delegate(&delegate); |
| 422 processor.EnqueueAction(&action); | 426 processor.EnqueueAction(&action); |
| 423 | 427 |
| 424 g_timeout_add(0, &TerminateTransferTestStarter, &processor); | 428 g_timeout_add(0, &TerminateTransferTestStarter, &processor); |
| 425 g_main_loop_run(loop); | 429 g_main_loop_run(loop); |
| 426 g_main_loop_unref(loop); | 430 g_main_loop_unref(loop); |
| 427 } | 431 } |
| (...skipping 164 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 592 } | 596 } |
| 593 | 597 |
| 594 TEST(OmahaRequestActionTest, IsEventTest) { | 598 TEST(OmahaRequestActionTest, IsEventTest) { |
| 595 string http_response("doesn't matter"); | 599 string http_response("doesn't matter"); |
| 596 NiceMock<PrefsMock> prefs; | 600 NiceMock<PrefsMock> prefs; |
| 597 OmahaRequestAction update_check_action( | 601 OmahaRequestAction update_check_action( |
| 598 &prefs, | 602 &prefs, |
| 599 kDefaultTestParams, | 603 kDefaultTestParams, |
| 600 NULL, | 604 NULL, |
| 601 new MockHttpFetcher(http_response.data(), | 605 new MockHttpFetcher(http_response.data(), |
| 602 http_response.size())); | 606 http_response.size(), |
| 607 NULL)); |
| 603 EXPECT_FALSE(update_check_action.IsEvent()); | 608 EXPECT_FALSE(update_check_action.IsEvent()); |
| 604 | 609 |
| 605 OmahaRequestAction event_action( | 610 OmahaRequestAction event_action( |
| 606 &prefs, | 611 &prefs, |
| 607 kDefaultTestParams, | 612 kDefaultTestParams, |
| 608 new OmahaEvent(OmahaEvent::kTypeUpdateComplete), | 613 new OmahaEvent(OmahaEvent::kTypeUpdateComplete), |
| 609 new MockHttpFetcher(http_response.data(), | 614 new MockHttpFetcher(http_response.data(), |
| 610 http_response.size())); | 615 http_response.size(), |
| 616 NULL)); |
| 611 EXPECT_TRUE(event_action.IsEvent()); | 617 EXPECT_TRUE(event_action.IsEvent()); |
| 612 } | 618 } |
| 613 | 619 |
| 614 TEST(OmahaRequestActionTest, FormatDeltaOkayOutputTest) { | 620 TEST(OmahaRequestActionTest, FormatDeltaOkayOutputTest) { |
| 615 for (int i = 0; i < 2; i++) { | 621 for (int i = 0; i < 2; i++) { |
| 616 bool delta_okay = i == 1; | 622 bool delta_okay = i == 1; |
| 617 const char* delta_okay_str = delta_okay ? "true" : "false"; | 623 const char* delta_okay_str = delta_okay ? "true" : "false"; |
| 618 vector<char> post_data; | 624 vector<char> post_data; |
| 619 OmahaRequestParams params(OmahaRequestParams::kOsPlatform, | 625 OmahaRequestParams params(OmahaRequestParams::kOsPlatform, |
| 620 OmahaRequestParams::kOsVersion, | 626 OmahaRequestParams::kOsVersion, |
| (...skipping 263 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 884 "", | 890 "", |
| 885 1500, | 891 1500, |
| 886 static_cast<ActionExitCode>( | 892 static_cast<ActionExitCode>( |
| 887 kActionCodeOmahaRequestHTTPResponseBase + 999), | 893 kActionCodeOmahaRequestHTTPResponseBase + 999), |
| 888 &response, | 894 &response, |
| 889 NULL)); | 895 NULL)); |
| 890 EXPECT_FALSE(response.update_exists); | 896 EXPECT_FALSE(response.update_exists); |
| 891 } | 897 } |
| 892 | 898 |
| 893 } // namespace chromeos_update_engine | 899 } // namespace chromeos_update_engine |
| OLD | NEW |