Chromium Code Reviews| Index: third_party/WebKit/Source/web/tests/PageSerializerTest.cpp |
| diff --git a/third_party/WebKit/Source/web/tests/PageSerializerTest.cpp b/third_party/WebKit/Source/web/tests/PageSerializerTest.cpp |
| index 6cbf3fdc036e473793546b9d18599a1bafebf62c..f94ae4054b8130026805fb1b3b9d1942afc9e149 100644 |
| --- a/third_party/WebKit/Source/web/tests/PageSerializerTest.cpp |
| +++ b/third_party/WebKit/Source/web/tests/PageSerializerTest.cpp |
| @@ -47,6 +47,7 @@ |
| #include "web/WebLocalFrameImpl.h" |
| #include "web/WebViewImpl.h" |
| #include "web/tests/FrameTestHelpers.h" |
| +#include "wtf/Assertions.h" |
| #include "wtf/Vector.h" |
| #include <gtest/gtest.h> |
| @@ -123,7 +124,13 @@ protected: |
| for (const auto& rewriteURL: m_rewriteURLs) |
| serializer.registerRewriteURL(rewriteURL.key, rewriteURL.value); |
| - serializer.serialize(m_helper.webViewImpl()->mainFrameImpl()->frame()->page()); |
| + Frame* frame = m_helper.webViewImpl()->mainFrameImpl()->frame(); |
| + for (; frame; frame = frame->tree().traverseNext()) { |
| + // This is safe, because tests do not do cross-site navigation |
| + // (and therefore don't have remote frames). |
| + LocalFrame* localFrame = toLocalFrame(frame); |
| + serializer.serializeFrame(*localFrame); |
|
dcheng
2015/11/18 03:27:12
Nit: just inline toLocalFrame(*frame) here.
Łukasz Anforowicz
2015/11/18 18:40:41
Done.
|
| + } |
| } |
| Vector<SerializedResource>& getResources() |
| @@ -131,7 +138,6 @@ protected: |
| return m_resources; |
| } |
| - |
| const SerializedResource* getResource(const char* url, const char* mimeType) |
| { |
| KURL kURL = KURL(m_baseUrl, url); |