OLD | NEW |
1 // Copyright 2016 PDFium Authors. All rights reserved. | 1 // Copyright 2016 PDFium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "public/fpdf_doc.h" | 5 #include "public/fpdf_doc.h" |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "core/fpdfapi/fpdf_parser/include/cpdf_document.h" | 10 #include "core/fpdfapi/fpdf_parser/include/cpdf_document.h" |
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
57 | 57 |
58 void SetUp() override { | 58 void SetUp() override { |
59 // We don't need page module or render module, but | 59 // We don't need page module or render module, but |
60 // initialize them to keep the code sane. | 60 // initialize them to keep the code sane. |
61 CPDF_ModuleMgr* module_mgr = CPDF_ModuleMgr::Get(); | 61 CPDF_ModuleMgr* module_mgr = CPDF_ModuleMgr::Get(); |
62 module_mgr->InitPageModule(); | 62 module_mgr->InitPageModule(); |
63 | 63 |
64 m_pDoc.reset(new CPDF_TestPdfDocument()); | 64 m_pDoc.reset(new CPDF_TestPdfDocument()); |
65 m_pIndirectObjs = m_pDoc->GetHolder(); | 65 m_pIndirectObjs = m_pDoc->GetHolder(); |
66 // Setup the root directory. | 66 // Setup the root directory. |
67 m_pRootObj.reset(new CPDF_Dictionary()); | 67 m_pRootObj.reset(new CPDF_Dictionary(CFX_WeakPtr<CFX_ByteStringPool>())); |
68 m_pDoc->SetRoot(m_pRootObj.get()); | 68 m_pDoc->SetRoot(m_pRootObj.get()); |
69 } | 69 } |
70 | 70 |
71 void TearDown() override { | 71 void TearDown() override { |
72 m_pRootObj.reset(); | 72 m_pRootObj.reset(); |
73 m_pIndirectObjs = nullptr; | 73 m_pIndirectObjs = nullptr; |
74 m_pDoc.reset(); | 74 m_pDoc.reset(); |
75 CPDF_ModuleMgr::Destroy(); | 75 CPDF_ModuleMgr::Destroy(); |
76 } | 76 } |
77 | 77 |
78 std::vector<DictObjInfo> CreateDictObjs(int num) { | 78 std::vector<DictObjInfo> CreateDictObjs(int num) { |
79 std::vector<DictObjInfo> info; | 79 std::vector<DictObjInfo> info; |
80 for (int i = 0; i < num; ++i) { | 80 for (int i = 0; i < num; ++i) { |
81 // Objects created will be released by the document. | 81 // Objects created will be released by the document. |
82 CPDF_Dictionary* obj = new CPDF_Dictionary; | 82 CPDF_Dictionary* obj( |
| 83 new CPDF_Dictionary(CFX_WeakPtr<CFX_ByteStringPool>())); |
83 info.push_back({m_pIndirectObjs->AddIndirectObject(obj), obj}); | 84 info.push_back({m_pIndirectObjs->AddIndirectObject(obj), obj}); |
84 } | 85 } |
85 return info; | 86 return info; |
86 } | 87 } |
87 | 88 |
88 protected: | 89 protected: |
89 std::unique_ptr<CPDF_TestPdfDocument> m_pDoc; | 90 std::unique_ptr<CPDF_TestPdfDocument> m_pDoc; |
90 CPDF_IndirectObjectHolder* m_pIndirectObjs; | 91 CPDF_IndirectObjectHolder* m_pIndirectObjs; |
91 std::unique_ptr<CPDF_Dictionary, ReleaseDeleter<CPDF_Dictionary>> m_pRootObj; | 92 std::unique_ptr<CPDF_Dictionary, ReleaseDeleter<CPDF_Dictionary>> m_pRootObj; |
92 }; | 93 }; |
93 | 94 |
94 TEST_F(PDFDocTest, FindBookmark) { | 95 TEST_F(PDFDocTest, FindBookmark) { |
95 { | 96 { |
96 // No bookmark information. | 97 // No bookmark information. |
97 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = | 98 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = |
98 GetFPDFWideString(L""); | 99 GetFPDFWideString(L""); |
99 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); | 100 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); |
100 | 101 |
101 title = GetFPDFWideString(L"Preface"); | 102 title = GetFPDFWideString(L"Preface"); |
102 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); | 103 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); |
103 } | 104 } |
104 { | 105 { |
105 // Empty bookmark tree. | 106 // Empty bookmark tree. |
106 m_pRootObj->SetFor("Outlines", new CPDF_Dictionary()); | 107 m_pRootObj->SetFor("Outlines", |
| 108 new CPDF_Dictionary(CFX_WeakPtr<CFX_ByteStringPool>())); |
107 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = | 109 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = |
108 GetFPDFWideString(L""); | 110 GetFPDFWideString(L""); |
109 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); | 111 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); |
110 | 112 |
111 title = GetFPDFWideString(L"Preface"); | 113 title = GetFPDFWideString(L"Preface"); |
112 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); | 114 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); |
113 } | 115 } |
114 { | 116 { |
115 // Check on a regular bookmark tree. | 117 // Check on a regular bookmark tree. |
116 auto bookmarks = CreateDictObjs(3); | 118 auto bookmarks = CreateDictObjs(3); |
(...skipping 107 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
224 // Title with no match. | 226 // Title with no match. |
225 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = | 227 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = |
226 GetFPDFWideString(L"Chapter 8"); | 228 GetFPDFWideString(L"Chapter 8"); |
227 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); | 229 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); |
228 | 230 |
229 // Title with a match. | 231 // Title with a match. |
230 title = GetFPDFWideString(L"Chapter 3"); | 232 title = GetFPDFWideString(L"Chapter 3"); |
231 EXPECT_EQ(bookmarks[3].obj, FPDFBookmark_Find(m_pDoc.get(), title.get())); | 233 EXPECT_EQ(bookmarks[3].obj, FPDFBookmark_Find(m_pDoc.get(), title.get())); |
232 } | 234 } |
233 } | 235 } |
OLD | NEW |