| Index: core/src/fpdfapi/fpdf_parser/fpdf_parser_parser_unittest.cpp
|
| diff --git a/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser_unittest.cpp b/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser_unittest.cpp
|
| index 827d1cec286a07ebe86f10fa01410c2d7edca7b4..73b46b52c2cbe78bef5628790ade1f4955d33c20 100644
|
| --- a/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser_unittest.cpp
|
| +++ b/core/src/fpdfapi/fpdf_parser/fpdf_parser_parser_unittest.cpp
|
| @@ -5,6 +5,28 @@
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| #include "../../../include/fpdfapi/fpdf_parser.h"
|
| +#include "../../../include/fxcrt/fx_stream.h"
|
| +
|
| +class CPDF_TestParser : public CPDF_Parser {
|
| + public:
|
| + CPDF_TestParser() {}
|
| + ~CPDF_TestParser() {}
|
| +
|
| + bool InitTest(const FX_CHAR* path) {
|
| + IFX_FileRead* pFileAccess = FX_CreateFileRead(path);
|
| + if (!pFileAccess)
|
| + return false;
|
| +
|
| + // For the test file, the header is set at the beginning.
|
| + m_Syntax.InitParser(pFileAccess, 0);
|
| + return true;
|
| + }
|
| +
|
| + private:
|
| + // Add test case as private friend so that RebuildCrossRef in CPDF_Parser
|
| + // can be accessed.
|
| + FRIEND_TEST(fpdf_parser_parser, RebuildCrossRefCorrectly);
|
| +};
|
|
|
| // TODO(thestig) Using unique_ptr with ReleaseDeleter is still not ideal.
|
| // Come up or wait for something better.
|
| @@ -168,3 +190,22 @@ TEST(fpdf_parser_parser, ReadHexString) {
|
| EXPECT_EQ(1, parser.SavePos());
|
| }
|
| }
|
| +
|
| +TEST(fpdf_parser_parser, RebuildCrossRefCorrectly) {
|
| + CPDF_TestParser parser;
|
| + ASSERT_TRUE(
|
| + parser.InitTest("testing/resources/parser_rebuildxref_correct.pdf"));
|
| +
|
| + ASSERT_TRUE(parser.RebuildCrossRef());
|
| + const FX_FILESIZE offsets[] = {0, 15, 61, 154, 296, 374, 450};
|
| + const FX_WORD versions[] = {0, 0, 2, 4, 6, 8, 0};
|
| + static_assert(FX_ArraySize(offsets) == FX_ArraySize(versions),
|
| + "numbers of offsets and versions should be same.");
|
| + ASSERT_EQ(FX_ArraySize(offsets), parser.m_CrossRef.GetSize());
|
| + ASSERT_EQ(FX_ArraySize(versions), parser.m_ObjVersion.GetSize());
|
| +
|
| + for (int i = 0; i < FX_ArraySize(offsets); ++i) {
|
| + EXPECT_EQ(offsets[i], parser.m_CrossRef.GetAt(i));
|
| + EXPECT_EQ(versions[i], parser.m_ObjVersion.GetAt(i));
|
| + }
|
| +}
|
|
|