src/chrome/browser/extensions/api/web_request/upload_data_presenter_unittest.cc

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