| 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" |
| 11 #include "core/fpdfapi/fpdf_parser/include/cpdf_name.h" | 11 #include "core/fpdfapi/fpdf_parser/include/cpdf_name.h" |
| 12 #include "core/fpdfapi/fpdf_parser/include/cpdf_number.h" | 12 #include "core/fpdfapi/fpdf_parser/include/cpdf_number.h" |
| 13 #include "core/fpdfapi/fpdf_parser/include/cpdf_parser.h" |
| 13 #include "core/fpdfapi/fpdf_parser/include/cpdf_reference.h" | 14 #include "core/fpdfapi/fpdf_parser/include/cpdf_reference.h" |
| 14 #include "core/fpdfapi/fpdf_parser/include/cpdf_string.h" | 15 #include "core/fpdfapi/fpdf_parser/include/cpdf_string.h" |
| 15 #include "core/fpdfapi/include/cpdf_modulemgr.h" | 16 #include "core/fpdfapi/include/cpdf_modulemgr.h" |
| 16 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
| 17 #include "testing/test_support.h" | 18 #include "testing/test_support.h" |
| 18 | 19 |
| 19 #ifdef PDF_ENABLE_XFA | 20 #ifdef PDF_ENABLE_XFA |
| 20 #include "fpdfsdk/fpdfxfa/include/fpdfxfa_app.h" | 21 #include "fpdfsdk/fpdfxfa/include/fpdfxfa_app.h" |
| 21 #include "fpdfsdk/fpdfxfa/include/fpdfxfa_doc.h" | 22 #include "fpdfsdk/fpdfxfa/include/fpdfxfa_doc.h" |
| 22 #endif // PDF_ENABLE_XFA | 23 #endif // PDF_ENABLE_XFA |
| 23 | 24 |
| 24 class CPDF_TestDocument : public CPDF_Document { | 25 class CPDF_TestDocument : public CPDF_Document { |
| 25 public: | 26 public: |
| 26 CPDF_TestDocument() : CPDF_Document(nullptr) {} | 27 CPDF_TestDocument() : CPDF_Document(std::unique_ptr<CPDF_Parser>()) {} |
| 27 | 28 |
| 28 void SetRoot(CPDF_Dictionary* root) { m_pRootDict = root; } | 29 void SetRoot(CPDF_Dictionary* root) { m_pRootDict = root; } |
| 29 CPDF_IndirectObjectHolder* GetHolder() { return this; } | 30 CPDF_IndirectObjectHolder* GetHolder() { return this; } |
| 30 }; | 31 }; |
| 31 | 32 |
| 32 #ifdef PDF_ENABLE_XFA | 33 #ifdef PDF_ENABLE_XFA |
| 33 class CPDF_TestXFADocument : public CPDFXFA_Document { | 34 class CPDF_TestXFADocument : public CPDFXFA_Document { |
| 34 public: | 35 public: |
| 35 CPDF_TestXFADocument() | 36 CPDF_TestXFADocument() |
| 36 : CPDFXFA_Document(new CPDF_TestDocument(), CPDFXFA_App::GetInstance()) {} | 37 : CPDFXFA_Document(WrapUnique(new CPDF_TestDocument()), |
| 38 CPDFXFA_App::GetInstance()) {} |
| 37 | 39 |
| 38 void SetRoot(CPDF_Dictionary* root) { | 40 void SetRoot(CPDF_Dictionary* root) { |
| 39 reinterpret_cast<CPDF_TestDocument*>(GetPDFDoc())->SetRoot(root); | 41 reinterpret_cast<CPDF_TestDocument*>(GetPDFDoc())->SetRoot(root); |
| 40 } | 42 } |
| 41 | 43 |
| 42 CPDF_IndirectObjectHolder* GetHolder() { return GetPDFDoc(); } | 44 CPDF_IndirectObjectHolder* GetHolder() { return GetPDFDoc(); } |
| 43 }; | 45 }; |
| 44 using CPDF_TestPdfDocument = CPDF_TestXFADocument; | 46 using CPDF_TestPdfDocument = CPDF_TestXFADocument; |
| 45 #else // PDF_ENABLE_XFA | 47 #else // PDF_ENABLE_XFA |
| 46 using CPDF_TestPdfDocument = CPDF_TestDocument; | 48 using CPDF_TestPdfDocument = CPDF_TestDocument; |
| (...skipping 176 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 223 // Title with no match. | 225 // Title with no match. |
| 224 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = | 226 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = |
| 225 GetFPDFWideString(L"Chapter 8"); | 227 GetFPDFWideString(L"Chapter 8"); |
| 226 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); | 228 EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); |
| 227 | 229 |
| 228 // Title with a match. | 230 // Title with a match. |
| 229 title = GetFPDFWideString(L"Chapter 3"); | 231 title = GetFPDFWideString(L"Chapter 3"); |
| 230 EXPECT_EQ(bookmarks[3].obj, FPDFBookmark_Find(m_pDoc.get(), title.get())); | 232 EXPECT_EQ(bookmarks[3].obj, FPDFBookmark_Find(m_pDoc.get(), title.get())); |
| 231 } | 233 } |
| 232 } | 234 } |
| OLD | NEW |