| Index: content/browser/loader/resource_dispatcher_host_unittest.cc
|
| diff --git a/content/browser/loader/resource_dispatcher_host_unittest.cc b/content/browser/loader/resource_dispatcher_host_unittest.cc
|
| index 3c9d8811ffc3fb59b981d9b58714b69c6ccc9217..75144d89d0b5bcac0e22f3e961f9c9805b55f08b 100644
|
| --- a/content/browser/loader/resource_dispatcher_host_unittest.cc
|
| +++ b/content/browser/loader/resource_dispatcher_host_unittest.cc
|
| @@ -8,6 +8,7 @@
|
| #include "base/file_path.h"
|
| #include "base/memory/scoped_vector.h"
|
| #include "base/message_loop.h"
|
| +#include "base/pickle.h"
|
| #include "base/process_util.h"
|
| #include "base/string_number_conversions.h"
|
| #include "base/string_split.h"
|
| @@ -75,9 +76,11 @@ static int RequestIDForMessage(const IPC::Message& msg) {
|
| case ResourceMsg_ReceivedRedirect::ID:
|
| case ResourceMsg_SetDataBuffer::ID:
|
| case ResourceMsg_DataReceived::ID:
|
| - case ResourceMsg_RequestComplete::ID:
|
| - request_id = IPC::MessageIterator(msg).NextInt();
|
| + case ResourceMsg_RequestComplete::ID: {
|
| + bool result = PickleIterator(msg).ReadInt(&request_id);
|
| + DCHECK(result);
|
| break;
|
| + }
|
| }
|
| return request_id;
|
| }
|
| @@ -688,7 +691,9 @@ class ResourceDispatcherHostTest : public testing::Test,
|
| void GenerateDataReceivedACK(const IPC::Message& msg) {
|
| EXPECT_EQ(ResourceMsg_DataReceived::ID, msg.type());
|
|
|
| - int request_id = IPC::MessageIterator(msg).NextInt();
|
| + int request_id = -1;
|
| + bool result = PickleIterator(msg).ReadInt(&request_id);
|
| + DCHECK(result);
|
| scoped_ptr<IPC::Message> ack(
|
| new ResourceHostMsg_DataReceived_ACK(msg.routing_id(), request_id));
|
|
|
|
|