| Index: fpdfsdk/src/fpdfdoc_embeddertest.cpp
|
| diff --git a/fpdfsdk/src/fpdfdoc_embeddertest.cpp b/fpdfsdk/src/fpdfdoc_embeddertest.cpp
|
| index 38a366268c59588599eff7e873f707dfea1bcb40..260f25f3099958afd12665a33ba3b9d228c7690b 100644
|
| --- a/fpdfsdk/src/fpdfdoc_embeddertest.cpp
|
| +++ b/fpdfsdk/src/fpdfdoc_embeddertest.cpp
|
| @@ -7,6 +7,7 @@
|
| #include "public/fpdfview.h"
|
| #include "testing/embedder_test.h"
|
| #include "testing/fx_string_testhelpers.h"
|
| +#include "testing/test_support.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| class FPDFDocEmbeddertest : public EmbedderTest {};
|
| @@ -101,20 +102,28 @@ TEST_F(FPDFDocEmbeddertest, Bookmarks) {
|
| }
|
|
|
| TEST_F(FPDFDocEmbeddertest, FindBookmarks) {
|
| - // Open a file with two bookmarks, and extract the first.
|
| + // Open a file with two bookmarks.
|
| EXPECT_TRUE(OpenDocument("bookmarks.pdf"));
|
|
|
| - unsigned short buf[128];
|
| - FPDF_BOOKMARK child = FPDFBookmark_GetFirstChild(document(), nullptr);
|
| + // Find the first one, based on its known title.
|
| + FPDF_WIDESTRING title = GetFPDFWideString(L"A Good Beginning");
|
| + FPDF_BOOKMARK child = FPDFBookmark_Find(document(), title);
|
| EXPECT_NE(nullptr, child);
|
| +
|
| + // Check that the string matches.
|
| + unsigned short buf[128];
|
| EXPECT_EQ(34, FPDFBookmark_GetTitle(child, buf, sizeof(buf)));
|
| EXPECT_EQ(CFX_WideString(L"A Good Beginning"),
|
| CFX_WideString::FromUTF16LE(buf, 16));
|
|
|
| - // Find the same one again using the title.
|
| - EXPECT_EQ(child, FPDFBookmark_Find(document(), buf));
|
| + // Check that it is them same as the one returned by GetFirstChild.
|
| + EXPECT_EQ(child, FPDFBookmark_GetFirstChild(document(), nullptr));
|
|
|
| // Try to find one using a non-existent title.
|
| - buf[0] = 'X';
|
| - EXPECT_EQ(nullptr, FPDFBookmark_Find(document(), buf));
|
| + FPDF_WIDESTRING bad_title = GetFPDFWideString(L"A BAD Beginning");
|
| + EXPECT_EQ(nullptr, FPDFBookmark_Find(document(), bad_title));
|
| +
|
| + // Alas, the typedef includes the "const".
|
| + free(const_cast<unsigned short*>(title));
|
| + free(const_cast<unsigned short*>(bad_title));
|
| }
|
|
|