Index: core/fpdfapi/parser/cpdf_document_unittest.cpp |
diff --git a/core/fpdfapi/parser/cpdf_document_unittest.cpp b/core/fpdfapi/parser/cpdf_document_unittest.cpp |
index 9336626f45989be83afd6f10b4dfd305df8f35f4..dfa6a136185094457a725bc68d2dfe5bb011c60c 100644 |
--- a/core/fpdfapi/parser/cpdf_document_unittest.cpp |
+++ b/core/fpdfapi/parser/cpdf_document_unittest.cpp |
@@ -9,15 +9,13 @@ |
#include "core/fpdfapi/cpdf_modulemgr.h" |
#include "core/fpdfapi/parser/cpdf_array.h" |
#include "core/fpdfapi/parser/cpdf_dictionary.h" |
+#include "core/fpdfapi/parser/cpdf_linearized.h" |
#include "core/fpdfapi/parser/cpdf_parser.h" |
#include "core/fxcrt/fx_memory.h" |
#include "testing/gtest/include/gtest/gtest.h" |
namespace { |
-using ScopedDictionary = |
- std::unique_ptr<CPDF_Dictionary, ReleaseDeleter<CPDF_Dictionary>>; |
- |
CPDF_Dictionary* CreatePageTreeNode(CPDF_Array* kids, |
CPDF_Document* pDoc, |
int count) { |
@@ -79,6 +77,11 @@ class CPDF_TestDocumentForPages : public CPDF_Document { |
std::unique_ptr<CPDF_Dictionary, ReleaseDeleter<CPDF_Dictionary>> |
m_pOwnedRootDict; |
}; |
+ |
+class TestLinearized : public CPDF_Linearized { |
+ public: |
+ explicit TestLinearized(CPDF_Dictionary* dict) : CPDF_Linearized(dict) {} |
+}; |
} // namespace |
class cpdf_document_test : public testing::Test { |
@@ -121,10 +124,12 @@ TEST_F(cpdf_document_test, UseCachedPageObjNumIfHaveNotPagesDict) { |
// can be not exists in this case. |
// (case, when hint table is used to page check in CPDF_DataAvail). |
CPDF_Document document(pdfium::MakeUnique<CPDF_Parser>()); |
- ScopedDictionary dict(new CPDF_Dictionary()); |
+ UniqueDictionary dict(new CPDF_Dictionary()); |
+ dict->SetBooleanFor("Linearized", true); |
const int page_count = 100; |
dict->SetIntegerFor("N", page_count); |
- document.LoadLinearizedDoc(dict.get()); |
+ TestLinearized linearized(dict.get()); |
+ document.LoadLinearizedDoc(&linearized); |
ASSERT_EQ(page_count, document.GetPageCount()); |
CPDF_Object* page_stub = new CPDF_Dictionary(); |
const uint32_t obj_num = document.AddIndirectObject(page_stub); |