Index: webkit/glue/multipart_response_delegate_unittest.cc |
diff --git a/webkit/glue/multipart_response_delegate_unittest.cc b/webkit/glue/multipart_response_delegate_unittest.cc |
index 474daa07bc587bb59f4af7c691be84ab89e5d8fb..6cd16229f4aac63cf5c7e0d352147a45a3ea781a 100644 |
--- a/webkit/glue/multipart_response_delegate_unittest.cc |
+++ b/webkit/glue/multipart_response_delegate_unittest.cc |
@@ -10,6 +10,7 @@ |
#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; |
@@ -22,6 +23,7 @@ using WebKit::WebURLRequest; |
using WebKit::WebURLResponse; |
using webkit_glue::MultipartResponseDelegate; |
using webkit_glue::MultipartResponseDelegateTester; |
+using webkit_glue::WebURLResponseExtraDataImpl; |
namespace webkit_glue { |
@@ -642,6 +644,7 @@ 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"); |
@@ -656,7 +659,9 @@ TEST(MultipartResponseTest, MultipartPayloadSet) { |
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" |
@@ -669,7 +674,9 @@ TEST(MultipartResponseTest, MultipartPayloadSet) { |
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 |