Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(38)

Side by Side Diff: fpdfsdk/fpdfdoc_unittest.cpp

Issue 2345063002: Use string pools in some dictionaries (Closed)
Patch Set: windows compile Created 4 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « fpdfsdk/fpdf_transformpage.cpp ('k') | fpdfsdk/fpdfeditpage.cpp » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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
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
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 }
OLDNEW
« no previous file with comments | « fpdfsdk/fpdf_transformpage.cpp ('k') | fpdfsdk/fpdfeditpage.cpp » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698