diff options
Diffstat (limited to 'fpdfsdk/src/fpdfdoc_unittest.cpp')
-rw-r--r-- | fpdfsdk/src/fpdfdoc_unittest.cpp | 226 |
1 files changed, 0 insertions, 226 deletions
diff --git a/fpdfsdk/src/fpdfdoc_unittest.cpp b/fpdfsdk/src/fpdfdoc_unittest.cpp deleted file mode 100644 index 04f885264e..0000000000 --- a/fpdfsdk/src/fpdfdoc_unittest.cpp +++ /dev/null @@ -1,226 +0,0 @@ -// Copyright 2016 PDFium Authors. All rights reserved. -// Use of this source code is governed by a BSD-style license that can be -// found in the LICENSE file. - -#include "public/fpdf_doc.h" - -#include <memory> -#include <vector> - -#include "core/include/fpdfapi/cpdf_document.h" -#include "core/include/fpdfapi/cpdf_name.h" -#include "core/include/fpdfapi/cpdf_number.h" -#include "core/include/fpdfapi/cpdf_reference.h" -#include "core/include/fpdfapi/cpdf_string.h" -#include "core/include/fpdfapi/fpdf_module.h" -#include "core/include/fpdfdoc/fpdf_doc.h" -#include "testing/gtest/include/gtest/gtest.h" -#include "testing/test_support.h" - -#ifdef PDF_ENABLE_XFA -#include "fpdfsdk/include/fpdfxfa/fpdfxfa_app.h" -#include "fpdfsdk/include/fpdfxfa/fpdfxfa_doc.h" -#endif // PDF_ENABLE_XFA - -class CPDF_TestDocument : public CPDF_Document { - public: - void SetRoot(CPDF_Dictionary* root) { m_pRootDict = root; } - CPDF_IndirectObjectHolder* GetHolder() { return this; } -}; - -#ifdef PDF_ENABLE_XFA -class CPDF_TestXFADocument : public CPDFXFA_Document { - public: - CPDF_TestXFADocument() - : CPDFXFA_Document(new CPDF_TestDocument(), CPDFXFA_App::GetInstance()) {} - - void SetRoot(CPDF_Dictionary* root) { - reinterpret_cast<CPDF_TestDocument*>(GetPDFDoc())->SetRoot(root); - } - - CPDF_IndirectObjectHolder* GetHolder() { return GetPDFDoc(); } -}; -using CPDF_TestPdfDocument = CPDF_TestXFADocument; -#else // PDF_ENABLE_XFA -using CPDF_TestPdfDocument = CPDF_TestDocument; -#endif // PDF_ENABLE_XFA - -class PDFDocTest : public testing::Test { - public: - struct DictObjInfo { - FX_DWORD num; - CPDF_Dictionary* obj; - }; - - void SetUp() override { - // We don't need page module or render module, but - // initialize them to keep the code sane. - CPDF_ModuleMgr::Create(); - CPDF_ModuleMgr* module_mgr = CPDF_ModuleMgr::Get(); - module_mgr->InitPageModule(); - module_mgr->InitRenderModule(); - - m_pDoc.reset(new CPDF_TestPdfDocument()); - m_pIndirectObjs = m_pDoc->GetHolder(); - // Setup the root directory. - m_pRootObj.reset(new CPDF_Dictionary()); - m_pDoc->SetRoot(m_pRootObj.get()); - } - - std::vector<DictObjInfo> CreateDictObjs(int num) { - std::vector<DictObjInfo> info; - for (int i = 0; i < num; ++i) { - // Objects created will be released by the document. - CPDF_Dictionary* obj(new CPDF_Dictionary()); - m_pIndirectObjs->AddIndirectObject(obj); - info.push_back({obj->GetObjNum(), obj}); - } - return info; - } - - protected: - std::unique_ptr<CPDF_TestPdfDocument> m_pDoc; - CPDF_IndirectObjectHolder* m_pIndirectObjs; - std::unique_ptr<CPDF_Dictionary, ReleaseDeleter<CPDF_Dictionary>> m_pRootObj; -}; - -TEST_F(PDFDocTest, FindBookmark) { - { - // No bookmark information. - std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = - GetFPDFWideString(L""); - EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); - - title = GetFPDFWideString(L"Preface"); - EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); - } - { - // Empty bookmark tree. - m_pRootObj->SetAt("Outlines", new CPDF_Dictionary()); - std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = - GetFPDFWideString(L""); - EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); - - title = GetFPDFWideString(L"Preface"); - EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); - } - { - // Check on a regular bookmark tree. - auto bookmarks = CreateDictObjs(3); - - bookmarks[1].obj->SetAt("Title", new CPDF_String(L"Chapter 1")); - bookmarks[1].obj->SetAt( - "Parent", new CPDF_Reference(m_pIndirectObjs, bookmarks[0].num)); - bookmarks[1].obj->SetAt( - "Next", new CPDF_Reference(m_pIndirectObjs, bookmarks[2].num)); - - bookmarks[2].obj->SetAt("Title", new CPDF_String(L"Chapter 2")); - bookmarks[2].obj->SetAt( - "Parent", new CPDF_Reference(m_pIndirectObjs, bookmarks[0].num)); - bookmarks[2].obj->SetAt( - "Prev", new CPDF_Reference(m_pIndirectObjs, bookmarks[1].num)); - - bookmarks[0].obj->SetAt("Type", new CPDF_Name("Outlines")); - bookmarks[0].obj->SetAt("Count", new CPDF_Number(2)); - bookmarks[0].obj->SetAt( - "First", new CPDF_Reference(m_pIndirectObjs, bookmarks[1].num)); - bookmarks[0].obj->SetAt( - "Last", new CPDF_Reference(m_pIndirectObjs, bookmarks[2].num)); - - m_pRootObj->SetAt("Outlines", - new CPDF_Reference(m_pIndirectObjs, bookmarks[0].num)); - - // Title with no match. - std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = - GetFPDFWideString(L"Chapter 3"); - EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); - - // Title with partial match only. - title = GetFPDFWideString(L"Chapter"); - EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); - - // Title with a match. - title = GetFPDFWideString(L"Chapter 2"); - EXPECT_EQ(bookmarks[2].obj, FPDFBookmark_Find(m_pDoc.get(), title.get())); - - // Title match is case insensitive. - title = GetFPDFWideString(L"cHaPter 2"); - EXPECT_EQ(bookmarks[2].obj, FPDFBookmark_Find(m_pDoc.get(), title.get())); - } - { - // Circular bookmarks in depth. - auto bookmarks = CreateDictObjs(3); - - bookmarks[1].obj->SetAt("Title", new CPDF_String(L"Chapter 1")); - bookmarks[1].obj->SetAt( - "Parent", new CPDF_Reference(m_pIndirectObjs, bookmarks[0].num)); - bookmarks[1].obj->SetAt( - "First", new CPDF_Reference(m_pIndirectObjs, bookmarks[2].num)); - - bookmarks[2].obj->SetAt("Title", new CPDF_String(L"Chapter 2")); - bookmarks[2].obj->SetAt( - "Parent", new CPDF_Reference(m_pIndirectObjs, bookmarks[1].num)); - bookmarks[2].obj->SetAt( - "First", new CPDF_Reference(m_pIndirectObjs, bookmarks[1].num)); - - bookmarks[0].obj->SetAt("Type", new CPDF_Name("Outlines")); - bookmarks[0].obj->SetAt("Count", new CPDF_Number(2)); - bookmarks[0].obj->SetAt( - "First", new CPDF_Reference(m_pIndirectObjs, bookmarks[1].num)); - bookmarks[0].obj->SetAt( - "Last", new CPDF_Reference(m_pIndirectObjs, bookmarks[2].num)); - - m_pRootObj->SetAt("Outlines", - new CPDF_Reference(m_pIndirectObjs, bookmarks[0].num)); - - // Title with no match. - std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = - GetFPDFWideString(L"Chapter 3"); - EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); - - // Title with a match. - title = GetFPDFWideString(L"Chapter 2"); - EXPECT_EQ(bookmarks[2].obj, FPDFBookmark_Find(m_pDoc.get(), title.get())); - } - { - // Circular bookmarks in breadth. - auto bookmarks = CreateDictObjs(4); - - bookmarks[1].obj->SetAt("Title", new CPDF_String(L"Chapter 1")); - bookmarks[1].obj->SetAt( - "Parent", new CPDF_Reference(m_pIndirectObjs, bookmarks[0].num)); - bookmarks[1].obj->SetAt( - "Next", new CPDF_Reference(m_pIndirectObjs, bookmarks[2].num)); - - bookmarks[2].obj->SetAt("Title", new CPDF_String(L"Chapter 2")); - bookmarks[2].obj->SetAt( - "Parent", new CPDF_Reference(m_pIndirectObjs, bookmarks[0].num)); - bookmarks[2].obj->SetAt( - "Next", new CPDF_Reference(m_pIndirectObjs, bookmarks[3].num)); - - bookmarks[3].obj->SetAt("Title", new CPDF_String(L"Chapter 3")); - bookmarks[3].obj->SetAt( - "Parent", new CPDF_Reference(m_pIndirectObjs, bookmarks[0].num)); - bookmarks[3].obj->SetAt( - "Next", new CPDF_Reference(m_pIndirectObjs, bookmarks[1].num)); - - bookmarks[0].obj->SetAt("Type", new CPDF_Name("Outlines")); - bookmarks[0].obj->SetAt("Count", new CPDF_Number(2)); - bookmarks[0].obj->SetAt( - "First", new CPDF_Reference(m_pIndirectObjs, bookmarks[1].num)); - bookmarks[0].obj->SetAt( - "Last", new CPDF_Reference(m_pIndirectObjs, bookmarks[2].num)); - - m_pRootObj->SetAt("Outlines", - new CPDF_Reference(m_pIndirectObjs, bookmarks[0].num)); - - // Title with no match. - std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = - GetFPDFWideString(L"Chapter 8"); - EXPECT_EQ(nullptr, FPDFBookmark_Find(m_pDoc.get(), title.get())); - - // Title with a match. - title = GetFPDFWideString(L"Chapter 3"); - EXPECT_EQ(bookmarks[3].obj, FPDFBookmark_Find(m_pDoc.get(), title.get())); - } -} |