Correctly whitelist ftp directory listings.

webkit/glue/multipart_response_delegate_unittest.cc

 // Copyright (c) 2011 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 <vector>
 
 #include "base/basictypes.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebURL.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebURLLoaderClient.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebURLResponse.h"
 #include "webkit/glue/multipart_response_delegate.h"
+#include "webkit/glue/weburlresponse_extradata_impl.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using std::string;
 using WebKit::WebString;
 using WebKit::WebURL;
 using WebKit::WebURLError;
 using WebKit::WebURLLoader;
 using WebKit::WebURLLoaderClient;
 using WebKit::WebURLRequest;
 using WebKit::WebURLResponse;
 using webkit_glue::MultipartResponseDelegate;
 using webkit_glue::MultipartResponseDelegateTester;
+using webkit_glue::WebURLResponseExtraDataImpl;
 
 namespace webkit_glue {
 
 class MultipartResponseDelegateTester {
  public:
   MultipartResponseDelegateTester(MultipartResponseDelegate* delegate)
       : delegate_(delegate) {
   }
 
   int PushOverLine(const std::string& data, size_t pos) {
@@ skipping 600 matching lines @@
       &content_range_upper_bound,
       &content_range_instance_size);
 
   EXPECT_EQ(result, false);
 }
 
 TEST(MultipartResponseTest, MultipartPayloadSet) {
   WebURLResponse response;
   response.initialize();
   response.setMIMEType("multipart/x-mixed-replace");
+  response.setExtraData(new WebURLResponseExtraDataImpl(""));
   MockWebURLLoaderClient client;
   MultipartResponseDelegate delegate(&client, NULL, response, "bound");
 
   string data(
       "--bound\n"
       "Content-type: text/plain\n\n"
       "response data\n"
       "--bound\n");
   delegate.OnReceivedData(data.c_str(),
                           static_cast<int>(data.length()),
                           static_cast<int>(data.length()));
   EXPECT_EQ(1, client.received_response_);
   EXPECT_EQ(string("response data"), client.data_);
   EXPECT_EQ(static_cast<int>(data.length()), client.total_encoded_data_length_);
-  EXPECT_FALSE(client.response_.isMultipartPayload());
+  WebURLResponseExtraDataImpl* extra_data =
+      static_cast<WebURLResponseExtraDataImpl*>(client.response_.extraData());
+  EXPECT_FALSE(extra_data->is_multipart_payload());
 
   string data2(
       "Content-type: text/plain\n\n"
       "response data2\n"
       "--bound\n");
   delegate.OnReceivedData(data2.c_str(),
                           static_cast<int>(data2.length()),
                           static_cast<int>(data2.length()));
   EXPECT_EQ(2, client.received_response_);
   EXPECT_EQ(string("response data2"), client.data_);
   EXPECT_EQ(static_cast<int>(data.length()) + static_cast<int>(data2.length()),
             client.total_encoded_data_length_);
-  EXPECT_TRUE(client.response_.isMultipartPayload());
+  extra_data = static_cast<WebURLResponseExtraDataImpl*>(
+      client.response_.extraData());
+  EXPECT_TRUE(extra_data->is_multipart_payload());
 }
 
 }  // namespace