| OLD | NEW |
| 1 // Copyright 2015 PDFium Authors. All rights reserved. | 1 // Copyright 2015 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 <string> | 5 #include <string> |
| 6 | 6 |
| 7 #include "core/fxcrt/include/fx_string.h" | 7 #include "core/fxcrt/include/fx_string.h" |
| 8 #include "public/fpdf_doc.h" | 8 #include "public/fpdf_doc.h" |
| 9 #include "public/fpdfview.h" | 9 #include "public/fpdfview.h" |
| 10 #include "testing/embedder_test.h" | 10 #include "testing/embedder_test.h" |
| (...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 65 | 65 |
| 66 FPDF_ClosePage(page); | 66 FPDF_ClosePage(page); |
| 67 } | 67 } |
| 68 | 68 |
| 69 TEST_F(FPDFDocEmbeddertest, NoBookmarks) { | 69 TEST_F(FPDFDocEmbeddertest, NoBookmarks) { |
| 70 // Open a file with no bookmarks. | 70 // Open a file with no bookmarks. |
| 71 EXPECT_TRUE(OpenDocument("named_dests.pdf")); | 71 EXPECT_TRUE(OpenDocument("named_dests.pdf")); |
| 72 | 72 |
| 73 // The non-existent top-level bookmark has no title. | 73 // The non-existent top-level bookmark has no title. |
| 74 unsigned short buf[128]; | 74 unsigned short buf[128]; |
| 75 EXPECT_EQ(0, FPDFBookmark_GetTitle(nullptr, buf, sizeof(buf))); | 75 EXPECT_EQ(0u, FPDFBookmark_GetTitle(nullptr, buf, sizeof(buf))); |
| 76 | 76 |
| 77 // The non-existent top-level bookmark has no children. | 77 // The non-existent top-level bookmark has no children. |
| 78 EXPECT_EQ(nullptr, FPDFBookmark_GetFirstChild(document(), nullptr)); | 78 EXPECT_EQ(nullptr, FPDFBookmark_GetFirstChild(document(), nullptr)); |
| 79 } | 79 } |
| 80 | 80 |
| 81 TEST_F(FPDFDocEmbeddertest, Bookmarks) { | 81 TEST_F(FPDFDocEmbeddertest, Bookmarks) { |
| 82 // Open a file with two bookmarks. | 82 // Open a file with two bookmarks. |
| 83 EXPECT_TRUE(OpenDocument("bookmarks.pdf")); | 83 EXPECT_TRUE(OpenDocument("bookmarks.pdf")); |
| 84 | 84 |
| 85 // The existent top-level bookmark has no title. | 85 // The existent top-level bookmark has no title. |
| 86 unsigned short buf[128]; | 86 unsigned short buf[128]; |
| 87 EXPECT_EQ(0, FPDFBookmark_GetTitle(nullptr, buf, sizeof(buf))); | 87 EXPECT_EQ(0u, FPDFBookmark_GetTitle(nullptr, buf, sizeof(buf))); |
| 88 | 88 |
| 89 FPDF_BOOKMARK child = FPDFBookmark_GetFirstChild(document(), nullptr); | 89 FPDF_BOOKMARK child = FPDFBookmark_GetFirstChild(document(), nullptr); |
| 90 EXPECT_NE(nullptr, child); | 90 EXPECT_NE(nullptr, child); |
| 91 EXPECT_EQ(34, FPDFBookmark_GetTitle(child, buf, sizeof(buf))); | 91 EXPECT_EQ(34u, FPDFBookmark_GetTitle(child, buf, sizeof(buf))); |
| 92 EXPECT_EQ(CFX_WideString(L"A Good Beginning"), | 92 EXPECT_EQ(CFX_WideString(L"A Good Beginning"), |
| 93 CFX_WideString::FromUTF16LE(buf, 16)); | 93 CFX_WideString::FromUTF16LE(buf, 16)); |
| 94 | 94 |
| 95 EXPECT_EQ(nullptr, FPDFBookmark_GetFirstChild(document(), child)); | 95 EXPECT_EQ(nullptr, FPDFBookmark_GetFirstChild(document(), child)); |
| 96 | 96 |
| 97 FPDF_BOOKMARK sibling = FPDFBookmark_GetNextSibling(document(), child); | 97 FPDF_BOOKMARK sibling = FPDFBookmark_GetNextSibling(document(), child); |
| 98 EXPECT_NE(nullptr, sibling); | 98 EXPECT_NE(nullptr, sibling); |
| 99 EXPECT_EQ(28, FPDFBookmark_GetTitle(sibling, buf, sizeof(buf))); | 99 EXPECT_EQ(28u, FPDFBookmark_GetTitle(sibling, buf, sizeof(buf))); |
| 100 EXPECT_EQ(CFX_WideString(L"A Good Ending"), | 100 EXPECT_EQ(CFX_WideString(L"A Good Ending"), |
| 101 CFX_WideString::FromUTF16LE(buf, 13)); | 101 CFX_WideString::FromUTF16LE(buf, 13)); |
| 102 | 102 |
| 103 EXPECT_EQ(nullptr, FPDFBookmark_GetNextSibling(document(), sibling)); | 103 EXPECT_EQ(nullptr, FPDFBookmark_GetNextSibling(document(), sibling)); |
| 104 } | 104 } |
| 105 | 105 |
| 106 TEST_F(FPDFDocEmbeddertest, FindBookmarks) { | 106 TEST_F(FPDFDocEmbeddertest, FindBookmarks) { |
| 107 // Open a file with two bookmarks. | 107 // Open a file with two bookmarks. |
| 108 EXPECT_TRUE(OpenDocument("bookmarks.pdf")); | 108 EXPECT_TRUE(OpenDocument("bookmarks.pdf")); |
| 109 | 109 |
| 110 // Find the first one, based on its known title. | 110 // Find the first one, based on its known title. |
| 111 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = | 111 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = |
| 112 GetFPDFWideString(L"A Good Beginning"); | 112 GetFPDFWideString(L"A Good Beginning"); |
| 113 FPDF_BOOKMARK child = FPDFBookmark_Find(document(), title.get()); | 113 FPDF_BOOKMARK child = FPDFBookmark_Find(document(), title.get()); |
| 114 EXPECT_NE(nullptr, child); | 114 EXPECT_NE(nullptr, child); |
| 115 | 115 |
| 116 // Check that the string matches. | 116 // Check that the string matches. |
| 117 unsigned short buf[128]; | 117 unsigned short buf[128]; |
| 118 EXPECT_EQ(34, FPDFBookmark_GetTitle(child, buf, sizeof(buf))); | 118 EXPECT_EQ(34u, FPDFBookmark_GetTitle(child, buf, sizeof(buf))); |
| 119 EXPECT_EQ(CFX_WideString(L"A Good Beginning"), | 119 EXPECT_EQ(CFX_WideString(L"A Good Beginning"), |
| 120 CFX_WideString::FromUTF16LE(buf, 16)); | 120 CFX_WideString::FromUTF16LE(buf, 16)); |
| 121 | 121 |
| 122 // Check that it is them same as the one returned by GetFirstChild. | 122 // Check that it is them same as the one returned by GetFirstChild. |
| 123 EXPECT_EQ(child, FPDFBookmark_GetFirstChild(document(), nullptr)); | 123 EXPECT_EQ(child, FPDFBookmark_GetFirstChild(document(), nullptr)); |
| 124 | 124 |
| 125 // Try to find one using a non-existent title. | 125 // Try to find one using a non-existent title. |
| 126 std::unique_ptr<unsigned short, pdfium::FreeDeleter> bad_title = | 126 std::unique_ptr<unsigned short, pdfium::FreeDeleter> bad_title = |
| 127 GetFPDFWideString(L"A BAD Beginning"); | 127 GetFPDFWideString(L"A BAD Beginning"); |
| 128 EXPECT_EQ(nullptr, FPDFBookmark_Find(document(), bad_title.get())); | 128 EXPECT_EQ(nullptr, FPDFBookmark_Find(document(), bad_title.get())); |
| 129 } | 129 } |
| 130 | 130 |
| 131 // Check circular bookmarks will not cause infinite loop. | 131 // Check circular bookmarks will not cause infinite loop. |
| 132 TEST_F(FPDFDocEmbeddertest, FindBookmarks_bug420) { | 132 TEST_F(FPDFDocEmbeddertest, FindBookmarks_bug420) { |
| 133 // Open a file with circular bookmarks. | 133 // Open a file with circular bookmarks. |
| 134 EXPECT_TRUE(OpenDocument("bookmarks_circular.pdf")); | 134 EXPECT_TRUE(OpenDocument("bookmarks_circular.pdf")); |
| 135 | 135 |
| 136 // Try to find a title. | 136 // Try to find a title. |
| 137 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = | 137 std::unique_ptr<unsigned short, pdfium::FreeDeleter> title = |
| 138 GetFPDFWideString(L"anything"); | 138 GetFPDFWideString(L"anything"); |
| 139 EXPECT_EQ(nullptr, FPDFBookmark_Find(document(), title.get())); | 139 EXPECT_EQ(nullptr, FPDFBookmark_Find(document(), title.get())); |
| 140 } | 140 } |
| OLD | NEW |