summaryrefslogtreecommitdiff
path: root/core/fpdfapi/parser/cpdf_document_unittest.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'core/fpdfapi/parser/cpdf_document_unittest.cpp')
-rw-r--r--core/fpdfapi/parser/cpdf_document_unittest.cpp33
1 files changed, 33 insertions, 0 deletions
diff --git a/core/fpdfapi/parser/cpdf_document_unittest.cpp b/core/fpdfapi/parser/cpdf_document_unittest.cpp
index 7b5ab43b5c..f27e7403f6 100644
--- a/core/fpdfapi/parser/cpdf_document_unittest.cpp
+++ b/core/fpdfapi/parser/cpdf_document_unittest.cpp
@@ -76,6 +76,27 @@ class CPDF_TestDocumentForPages : public CPDF_Document {
std::unique_ptr<CPDF_Dictionary> m_pOwnedRootDict;
};
+class CPDF_TestDocumentWithPageWithoutPageNum : public CPDF_Document {
+ public:
+ CPDF_TestDocumentWithPageWithoutPageNum() : CPDF_Document(nullptr) {
+ // Set up test
+ CPDF_Array* allPages = new CPDF_Array();
+ allPages->AddReference(this, AddIndirectObject(CreateNumberedPage(0)));
+ allPages->AddReference(this, AddIndirectObject(CreateNumberedPage(1)));
+ // Page without pageNum.
+ allPages->Add(CreateNumberedPage(2));
+ CPDF_Dictionary* pagesDict = CreatePageTreeNode(allPages, this, 3);
+
+ m_pOwnedRootDict.reset(new CPDF_Dictionary());
+ m_pOwnedRootDict->SetReferenceFor("Pages", this, pagesDict->GetObjNum());
+ m_pRootDict = m_pOwnedRootDict.get();
+ m_PageList.SetSize(3);
+ }
+
+ private:
+ std::unique_ptr<CPDF_Dictionary> m_pOwnedRootDict;
+};
+
class TestLinearized : public CPDF_LinearizedHeader {
public:
explicit TestLinearized(CPDF_Dictionary* dict)
@@ -105,6 +126,18 @@ TEST_F(cpdf_document_test, GetPages) {
EXPECT_FALSE(page);
}
+TEST_F(cpdf_document_test, GetPageWithoutObjNumTwice) {
+ std::unique_ptr<CPDF_TestDocumentWithPageWithoutPageNum> document =
+ pdfium::MakeUnique<CPDF_TestDocumentWithPageWithoutPageNum>();
+ const CPDF_Dictionary* page = document->GetPage(2);
+ ASSERT_TRUE(page);
+ // This is page without obj num.
+ ASSERT_EQ(0ul, page->GetObjNum());
+ const CPDF_Dictionary* second_call_page = document->GetPage(2);
+ EXPECT_TRUE(second_call_page);
+ EXPECT_EQ(page, second_call_page);
+}
+
TEST_F(cpdf_document_test, GetPagesReverseOrder) {
std::unique_ptr<CPDF_TestDocumentForPages> document =
pdfium::MakeUnique<CPDF_TestDocumentForPages>();