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.cpp31
1 files changed, 11 insertions, 20 deletions
diff --git a/core/fpdfapi/parser/cpdf_document_unittest.cpp b/core/fpdfapi/parser/cpdf_document_unittest.cpp
index f06a7f42d3..fa52d3bc7f 100644
--- a/core/fpdfapi/parser/cpdf_document_unittest.cpp
+++ b/core/fpdfapi/parser/cpdf_document_unittest.cpp
@@ -87,20 +87,16 @@ class CPDF_TestDocumentForPages : public CPDF_Document {
CPDF_Dictionary* pagesDict =
CreatePageTreeNode(std::move(allPages), this, kNumTestPages);
- m_pOwnedRootDict = pdfium::MakeUnique<CPDF_Dictionary>();
- m_pOwnedRootDict->SetNewFor<CPDF_Reference>("Pages", this,
- pagesDict->GetObjNum());
- m_pRootDict = m_pOwnedRootDict.get();
+ m_pRootDict = NewIndirect<CPDF_Dictionary>();
+ m_pRootDict->SetNewFor<CPDF_Reference>("Pages", this,
+ pagesDict->GetObjNum());
m_PageList.resize(kNumTestPages);
}
void SetTreeSize(int size) {
- m_pOwnedRootDict->SetNewFor<CPDF_Number>("Count", size);
+ m_pRootDict->SetNewFor<CPDF_Number>("Count", size);
m_PageList.resize(size);
}
-
- private:
- std::unique_ptr<CPDF_Dictionary> m_pOwnedRootDict;
};
class CPDF_TestDocumentWithPageWithoutPageNum : public CPDF_Document {
@@ -116,17 +112,15 @@ class CPDF_TestDocumentWithPageWithoutPageNum : public CPDF_Document {
inlined_page_ = allPages->Add(CreateNumberedPage(2));
CPDF_Dictionary* pagesDict =
CreatePageTreeNode(std::move(allPages), this, 3);
- m_pOwnedRootDict = pdfium::MakeUnique<CPDF_Dictionary>();
- m_pOwnedRootDict->SetNewFor<CPDF_Reference>("Pages", this,
- pagesDict->GetObjNum());
- m_pRootDict = m_pOwnedRootDict.get();
+ m_pRootDict = NewIndirect<CPDF_Dictionary>();
+ m_pRootDict->SetNewFor<CPDF_Reference>("Pages", this,
+ pagesDict->GetObjNum());
m_PageList.resize(3);
}
const CPDF_Object* inlined_page() const { return inlined_page_; }
private:
- std::unique_ptr<CPDF_Dictionary> m_pOwnedRootDict;
const CPDF_Object* inlined_page_;
};
@@ -143,15 +137,12 @@ class CPDF_TestDocPagesWithoutKids : public CPDF_Document {
pagesDict->SetNewFor<CPDF_Name>("Type", "Pages");
pagesDict->SetNewFor<CPDF_Number>("Count", 3);
m_PageList.resize(10);
- m_pOwnedRootDict = pdfium::MakeUnique<CPDF_Dictionary>();
- m_pOwnedRootDict->SetNewFor<CPDF_Reference>("Pages", this,
- pagesDict->GetObjNum());
- m_pRootDict = m_pOwnedRootDict.get();
+ m_pRootDict = NewIndirect<CPDF_Dictionary>();
+ m_pRootDict->SetNewFor<CPDF_Reference>("Pages", this,
+ pagesDict->GetObjNum());
}
-
- private:
- std::unique_ptr<CPDF_Dictionary> m_pOwnedRootDict;
};
+
} // namespace
class cpdf_document_test : public testing::Test {