| // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <stddef.h> |
| |
| #include <utility> |
| |
| #include "base/values.h" |
| #include "extensions/browser/api/web_request/upload_data_presenter.h" |
| #include "extensions/browser/api/web_request/web_request_api_constants.h" |
| #include "net/base/upload_bytes_element_reader.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace keys = extension_web_request_api_constants; |
| |
| namespace extensions { |
| |
| // This only tests the handling of dots in keys. Other functionality is covered |
| // by ExtensionWebRequestTest.AccessRequestBodyData and |
| // WebRequestFormDataParserTest. |
| TEST(WebRequestUploadDataPresenterTest, ParsedData) { |
| // Input. |
| const char block[] = "key.with.dots=value"; |
| net::UploadBytesElementReader element(block, sizeof(block) - 1); |
| |
| // Expected output. |
| std::unique_ptr<base::ListValue> values(new base::ListValue); |
| values->AppendString("value"); |
| base::DictionaryValue expected_form; |
| expected_form.SetWithoutPathExpansion("key.with.dots", values.release()); |
| |
| // Real output. |
| std::unique_ptr<ParsedDataPresenter> parsed_data_presenter( |
| ParsedDataPresenter::CreateForTests()); |
| ASSERT_TRUE(parsed_data_presenter.get() != NULL); |
| parsed_data_presenter->FeedNext(element); |
| EXPECT_TRUE(parsed_data_presenter->Succeeded()); |
| std::unique_ptr<base::Value> result = parsed_data_presenter->Result(); |
| ASSERT_TRUE(result.get() != NULL); |
| |
| EXPECT_TRUE(result->Equals(&expected_form)); |
| } |
| |
| TEST(WebRequestUploadDataPresenterTest, RawData) { |
| // Input. |
| const char block1[] = "test"; |
| const size_t block1_size = sizeof(block1) - 1; |
| const char kFilename[] = "path/test_filename.ext"; |
| const char block2[] = "another test"; |
| const size_t block2_size = sizeof(block2) - 1; |
| |
| // Expected output. |
| std::unique_ptr<base::BinaryValue> expected_a( |
| base::BinaryValue::CreateWithCopiedBuffer(block1, block1_size)); |
| ASSERT_TRUE(expected_a.get() != NULL); |
| std::unique_ptr<base::StringValue> expected_b( |
| new base::StringValue(kFilename)); |
| ASSERT_TRUE(expected_b.get() != NULL); |
| std::unique_ptr<base::BinaryValue> expected_c( |
| base::BinaryValue::CreateWithCopiedBuffer(block2, block2_size)); |
| ASSERT_TRUE(expected_c.get() != NULL); |
| |
| base::ListValue expected_list; |
| subtle::AppendKeyValuePair(keys::kRequestBodyRawBytesKey, |
| std::move(expected_a), &expected_list); |
| subtle::AppendKeyValuePair(keys::kRequestBodyRawFileKey, |
| std::move(expected_b), &expected_list); |
| subtle::AppendKeyValuePair(keys::kRequestBodyRawBytesKey, |
| std::move(expected_c), &expected_list); |
| |
| // Real output. |
| RawDataPresenter raw_presenter; |
| raw_presenter.FeedNextBytes(block1, block1_size); |
| raw_presenter.FeedNextFile(kFilename); |
| raw_presenter.FeedNextBytes(block2, block2_size); |
| EXPECT_TRUE(raw_presenter.Succeeded()); |
| std::unique_ptr<base::Value> result = raw_presenter.Result(); |
| ASSERT_TRUE(result.get() != NULL); |
| |
| EXPECT_TRUE(result->Equals(&expected_list)); |
| } |
| |
| } // namespace extensions |