| 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/basictypes.h" | 5 #include "base/basictypes.h" |
| 6 #include "base/values.h" | 6 #include "base/values.h" |
| 7 #include "chrome/browser/extensions/api/web_request/upload_data_presenter.h" | 7 #include "chrome/browser/extensions/api/web_request/upload_data_presenter.h" |
| 8 #include "chrome/browser/extensions/api/web_request/web_request_api_constants.h" | 8 #include "chrome/browser/extensions/api/web_request/web_request_api_constants.h" |
| 9 #include "net/base/upload_bytes_element_reader.h" | 9 #include "net/base/upload_bytes_element_reader.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 | 11 |
| 12 namespace keys = extension_web_request_api_constants; | 12 namespace keys = extension_web_request_api_constants; |
| 13 | 13 |
| 14 namespace extensions { | 14 namespace extensions { |
| 15 | 15 |
| 16 // This only tests the handling of dots in keys. Other functionality is covered | 16 // This only tests the handling of dots in keys. Other functionality is covered |
| 17 // by ExtensionWebRequestTest.AccessRequestBodyData and | 17 // by ExtensionWebRequestTest.AccessRequestBodyData and |
| 18 // WebRequestFormDataParserTest. | 18 // WebRequestFormDataParserTest. |
| 19 TEST(WebRequestUploadDataPresenterTest, ParsedData) { | 19 TEST(WebRequestUploadDataPresenterTest, ParsedData) { |
| 20 // Input. | 20 // Input. |
| 21 const char block[] = "key.with.dots=value"; | 21 const char block[] = "key.with.dots=value"; |
| 22 net::UploadBytesElementReader element(block, sizeof(block) - 1); | 22 net::UploadBytesElementReader element(block, sizeof(block) - 1); |
| 23 | 23 |
| 24 // Expected output. | 24 // Expected output. |
| 25 scoped_ptr<base::ListValue> values(new base::ListValue); | 25 scoped_ptr<base::ListValue> values(new base::ListValue); |
| 26 values->Append(base::Value::CreateStringValue("value")); | 26 values->Append(new base::StringValue("value")); |
| 27 base::DictionaryValue expected_form; | 27 base::DictionaryValue expected_form; |
| 28 expected_form.SetWithoutPathExpansion("key.with.dots", values.release()); | 28 expected_form.SetWithoutPathExpansion("key.with.dots", values.release()); |
| 29 | 29 |
| 30 // Real output. | 30 // Real output. |
| 31 scoped_ptr<ParsedDataPresenter> parsed_data_presenter( | 31 scoped_ptr<ParsedDataPresenter> parsed_data_presenter( |
| 32 ParsedDataPresenter::CreateForTests()); | 32 ParsedDataPresenter::CreateForTests()); |
| 33 ASSERT_TRUE(parsed_data_presenter.get() != NULL); | 33 ASSERT_TRUE(parsed_data_presenter.get() != NULL); |
| 34 parsed_data_presenter->FeedNext(element); | 34 parsed_data_presenter->FeedNext(element); |
| 35 EXPECT_TRUE(parsed_data_presenter->Succeeded()); | 35 EXPECT_TRUE(parsed_data_presenter->Succeeded()); |
| 36 scoped_ptr<base::Value> result = parsed_data_presenter->Result(); | 36 scoped_ptr<base::Value> result = parsed_data_presenter->Result(); |
| 37 ASSERT_TRUE(result.get() != NULL); | 37 ASSERT_TRUE(result.get() != NULL); |
| 38 | 38 |
| 39 EXPECT_TRUE(result->Equals(&expected_form)); | 39 EXPECT_TRUE(result->Equals(&expected_form)); |
| 40 } | 40 } |
| 41 | 41 |
| 42 TEST(WebRequestUploadDataPresenterTest, RawData) { | 42 TEST(WebRequestUploadDataPresenterTest, RawData) { |
| 43 // Input. | 43 // Input. |
| 44 const char block1[] = "test"; | 44 const char block1[] = "test"; |
| 45 const size_t block1_size = sizeof(block1) - 1; | 45 const size_t block1_size = sizeof(block1) - 1; |
| 46 const char kFilename[] = "path/test_filename.ext"; | 46 const char kFilename[] = "path/test_filename.ext"; |
| 47 const char block2[] = "another test"; | 47 const char block2[] = "another test"; |
| 48 const size_t block2_size = sizeof(block2) - 1; | 48 const size_t block2_size = sizeof(block2) - 1; |
| 49 | 49 |
| 50 // Expected output. | 50 // Expected output. |
| 51 scoped_ptr<base::BinaryValue> expected_a( | 51 scoped_ptr<base::BinaryValue> expected_a( |
| 52 base::BinaryValue::CreateWithCopiedBuffer(block1, block1_size)); | 52 base::BinaryValue::CreateWithCopiedBuffer(block1, block1_size)); |
| 53 ASSERT_TRUE(expected_a.get() != NULL); | 53 ASSERT_TRUE(expected_a.get() != NULL); |
| 54 scoped_ptr<base::StringValue> expected_b( | 54 scoped_ptr<base::StringValue> expected_b( |
| 55 base::Value::CreateStringValue(kFilename)); | 55 new base::StringValue(kFilename)); |
| 56 ASSERT_TRUE(expected_b.get() != NULL); | 56 ASSERT_TRUE(expected_b.get() != NULL); |
| 57 scoped_ptr<base::BinaryValue> expected_c( | 57 scoped_ptr<base::BinaryValue> expected_c( |
| 58 base::BinaryValue::CreateWithCopiedBuffer(block2, block2_size)); | 58 base::BinaryValue::CreateWithCopiedBuffer(block2, block2_size)); |
| 59 ASSERT_TRUE(expected_c.get() != NULL); | 59 ASSERT_TRUE(expected_c.get() != NULL); |
| 60 | 60 |
| 61 base::ListValue expected_list; | 61 base::ListValue expected_list; |
| 62 subtle::AppendKeyValuePair( | 62 subtle::AppendKeyValuePair( |
| 63 keys::kRequestBodyRawBytesKey, expected_a.release(), &expected_list); | 63 keys::kRequestBodyRawBytesKey, expected_a.release(), &expected_list); |
| 64 subtle::AppendKeyValuePair( | 64 subtle::AppendKeyValuePair( |
| 65 keys::kRequestBodyRawFileKey, expected_b.release(), &expected_list); | 65 keys::kRequestBodyRawFileKey, expected_b.release(), &expected_list); |
| 66 subtle::AppendKeyValuePair( | 66 subtle::AppendKeyValuePair( |
| 67 keys::kRequestBodyRawBytesKey, expected_c.release(), &expected_list); | 67 keys::kRequestBodyRawBytesKey, expected_c.release(), &expected_list); |
| 68 | 68 |
| 69 // Real output. | 69 // Real output. |
| 70 RawDataPresenter raw_presenter; | 70 RawDataPresenter raw_presenter; |
| 71 raw_presenter.FeedNextBytes(block1, block1_size); | 71 raw_presenter.FeedNextBytes(block1, block1_size); |
| 72 raw_presenter.FeedNextFile(kFilename); | 72 raw_presenter.FeedNextFile(kFilename); |
| 73 raw_presenter.FeedNextBytes(block2, block2_size); | 73 raw_presenter.FeedNextBytes(block2, block2_size); |
| 74 EXPECT_TRUE(raw_presenter.Succeeded()); | 74 EXPECT_TRUE(raw_presenter.Succeeded()); |
| 75 scoped_ptr<base::Value> result = raw_presenter.Result(); | 75 scoped_ptr<base::Value> result = raw_presenter.Result(); |
| 76 ASSERT_TRUE(result.get() != NULL); | 76 ASSERT_TRUE(result.get() != NULL); |
| 77 | 77 |
| 78 EXPECT_TRUE(result->Equals(&expected_list)); | 78 EXPECT_TRUE(result->Equals(&expected_list)); |
| 79 } | 79 } |
| 80 | 80 |
| 81 } // namespace extensions | 81 } // namespace extensions |
| OLD | NEW |