| Index: core/src/fpdfapi/fpdf_parser/fpdf_parser_objects_unittest.cpp
|
| diff --git a/core/src/fpdfapi/fpdf_parser/fpdf_parser_objects_unittest.cpp b/core/src/fpdfapi/fpdf_parser/fpdf_parser_objects_unittest.cpp
|
| index 9de31c8673fae309e8e8eb5a410aed5e20f3b332..d2c30fa6ae2857a1a061b3075a8231146af6eb14 100644
|
| --- a/core/src/fpdfapi/fpdf_parser/fpdf_parser_objects_unittest.cpp
|
| +++ b/core/src/fpdfapi/fpdf_parser/fpdf_parser_objects_unittest.cpp
|
| @@ -10,6 +10,31 @@
|
| #include "core/include/fxcrt/fx_basic.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
|
|
| +namespace {
|
| +
|
| +using ScopedArray = std::unique_ptr<CPDF_Array, ReleaseDeleter<CPDF_Array>>;
|
| +
|
| +void TestArrayAccessors(const CPDF_Array* arr,
|
| + size_t index,
|
| + const char* str_val,
|
| + const char* const_str_val,
|
| + int int_val,
|
| + float float_val,
|
| + CPDF_Array* arr_val,
|
| + CPDF_Dictionary* dict_val,
|
| + CPDF_Stream* stream_val) {
|
| + EXPECT_STREQ(str_val, arr->GetStringAt(index).c_str());
|
| + EXPECT_STREQ(const_str_val, arr->GetConstStringAt(index).GetCStr());
|
| + EXPECT_EQ(int_val, arr->GetIntegerAt(index));
|
| + EXPECT_EQ(float_val, arr->GetNumberAt(index));
|
| + EXPECT_EQ(float_val, arr->GetFloatAt(index));
|
| + EXPECT_EQ(arr_val, arr->GetArrayAt(index));
|
| + EXPECT_EQ(dict_val, arr->GetDictAt(index));
|
| + EXPECT_EQ(stream_val, arr->GetStreamAt(index));
|
| +}
|
| +
|
| +} // namespace
|
| +
|
| class PDFObjectsTest : public testing::Test {
|
| public:
|
| void SetUp() override {
|
| @@ -154,13 +179,13 @@ TEST_F(PDFObjectsTest, GetString) {
|
| "", "", "", ""};
|
| // Check for direct objects.
|
| for (size_t i = 0; i < m_DirectObjs.size(); ++i)
|
| - EXPECT_STREQ(m_DirectObjs[i]->GetString().c_str(), direct_obj_results[i]);
|
| + EXPECT_STREQ(direct_obj_results[i], m_DirectObjs[i]->GetString().c_str());
|
|
|
| // Check indirect references.
|
| const char* indirect_obj_results[] = {"true", "1245", "\t\n", "space",
|
| "", "", ""};
|
| for (size_t i = 0; i < m_RefObjs.size(); ++i) {
|
| - EXPECT_STREQ(m_RefObjs[i]->GetString().c_str(), indirect_obj_results[i]);
|
| + EXPECT_STREQ(indirect_obj_results[i], m_RefObjs[i]->GetString().c_str());
|
| }
|
| }
|
|
|
| @@ -171,11 +196,11 @@ TEST_F(PDFObjectsTest, GetConstString) {
|
| // Check for direct objects.
|
| for (size_t i = 0; i < m_DirectObjs.size(); ++i) {
|
| if (!direct_obj_results[i]) {
|
| - EXPECT_EQ(m_DirectObjs[i]->GetConstString().GetCStr(),
|
| - direct_obj_results[i]);
|
| + EXPECT_EQ(direct_obj_results[i],
|
| + m_DirectObjs[i]->GetConstString().GetCStr());
|
| } else {
|
| - EXPECT_STREQ(m_DirectObjs[i]->GetConstString().GetCStr(),
|
| - direct_obj_results[i]);
|
| + EXPECT_STREQ(direct_obj_results[i],
|
| + m_DirectObjs[i]->GetConstString().GetCStr());
|
| }
|
| }
|
| // Check indirect references.
|
| @@ -183,10 +208,10 @@ TEST_F(PDFObjectsTest, GetConstString) {
|
| nullptr, nullptr, nullptr};
|
| for (size_t i = 0; i < m_RefObjs.size(); ++i) {
|
| if (!indirect_obj_results[i]) {
|
| - EXPECT_EQ(m_RefObjs[i]->GetConstString().GetCStr(), nullptr);
|
| + EXPECT_EQ(nullptr, m_RefObjs[i]->GetConstString().GetCStr());
|
| } else {
|
| - EXPECT_STREQ(m_RefObjs[i]->GetConstString().GetCStr(),
|
| - indirect_obj_results[i]);
|
| + EXPECT_STREQ(indirect_obj_results[i],
|
| + m_RefObjs[i]->GetConstString().GetCStr());
|
| }
|
| }
|
| }
|
| @@ -198,12 +223,12 @@ TEST_F(PDFObjectsTest, GetUnicodeText) {
|
| L""};
|
| // Check for direct objects.
|
| for (size_t i = 0; i < m_DirectObjs.size(); ++i)
|
| - EXPECT_STREQ(m_DirectObjs[i]->GetUnicodeText().c_str(),
|
| - direct_obj_results[i]);
|
| + EXPECT_STREQ(direct_obj_results[i],
|
| + m_DirectObjs[i]->GetUnicodeText().c_str());
|
|
|
| // Check indirect references.
|
| for (const auto& it : m_RefObjs)
|
| - EXPECT_STREQ(it->GetUnicodeText().c_str(), L"");
|
| + EXPECT_STREQ(L"", it->GetUnicodeText().c_str());
|
| }
|
|
|
| TEST_F(PDFObjectsTest, GetNumber) {
|
| @@ -211,24 +236,24 @@ TEST_F(PDFObjectsTest, GetNumber) {
|
| 0, 0, 0, 0, 0};
|
| // Check for direct objects.
|
| for (size_t i = 0; i < m_DirectObjs.size(); ++i)
|
| - EXPECT_EQ(m_DirectObjs[i]->GetNumber(), direct_obj_results[i]);
|
| + EXPECT_EQ(direct_obj_results[i], m_DirectObjs[i]->GetNumber());
|
|
|
| // Check indirect references.
|
| const FX_FLOAT indirect_obj_results[] = {0, 1245, 0, 0, 0, 0, 0};
|
| for (size_t i = 0; i < m_RefObjs.size(); ++i)
|
| - EXPECT_EQ(m_RefObjs[i]->GetNumber(), indirect_obj_results[i]);
|
| + EXPECT_EQ(indirect_obj_results[i], m_RefObjs[i]->GetNumber());
|
| }
|
|
|
| TEST_F(PDFObjectsTest, GetInteger) {
|
| const int direct_obj_results[] = {0, 1, 1245, 9, 0, 0, 0, 0, 0, 0, 0};
|
| // Check for direct objects.
|
| for (size_t i = 0; i < m_DirectObjs.size(); ++i)
|
| - EXPECT_EQ(m_DirectObjs[i]->GetInteger(), direct_obj_results[i]);
|
| + EXPECT_EQ(direct_obj_results[i], m_DirectObjs[i]->GetInteger());
|
|
|
| // Check indirect references.
|
| const int indirect_obj_results[] = {1, 1245, 0, 0, 0, 0, 0};
|
| for (size_t i = 0; i < m_RefObjs.size(); ++i)
|
| - EXPECT_EQ(m_RefObjs[i]->GetInteger(), indirect_obj_results[i]);
|
| + EXPECT_EQ(indirect_obj_results[i], m_RefObjs[i]->GetInteger());
|
| }
|
|
|
| TEST_F(PDFObjectsTest, GetDict) {
|
| @@ -237,13 +262,13 @@ TEST_F(PDFObjectsTest, GetDict) {
|
| nullptr, nullptr, m_DictObj, m_StreamDictObj, nullptr};
|
| // Check for direct objects.
|
| for (size_t i = 0; i < m_DirectObjs.size(); ++i)
|
| - EXPECT_EQ(m_DirectObjs[i]->GetDict(), direct_obj_results[i]);
|
| + EXPECT_EQ(direct_obj_results[i], m_DirectObjs[i]->GetDict());
|
|
|
| // Check indirect references.
|
| const CPDF_Dictionary* indirect_obj_results[] = {
|
| nullptr, nullptr, nullptr, nullptr, nullptr, m_DictObj, m_StreamDictObj};
|
| for (size_t i = 0; i < m_RefObjs.size(); ++i)
|
| - EXPECT_EQ(m_RefObjs[i]->GetDict(), indirect_obj_results[i]);
|
| + EXPECT_EQ(indirect_obj_results[i], m_RefObjs[i]->GetDict());
|
| }
|
|
|
| TEST_F(PDFObjectsTest, GetArray) {
|
| @@ -252,11 +277,11 @@ TEST_F(PDFObjectsTest, GetArray) {
|
| nullptr, m_ArrayObj, nullptr, nullptr, nullptr};
|
| // Check for direct objects.
|
| for (size_t i = 0; i < m_DirectObjs.size(); ++i)
|
| - EXPECT_EQ(m_DirectObjs[i]->GetArray(), direct_obj_results[i]);
|
| + EXPECT_EQ(direct_obj_results[i], m_DirectObjs[i]->GetArray());
|
|
|
| // Check indirect references.
|
| for (const auto& it : m_RefObjs)
|
| - EXPECT_EQ(it->GetArray(), nullptr);
|
| + EXPECT_EQ(nullptr, it->GetArray());
|
| }
|
|
|
| TEST_F(PDFObjectsTest, Clone) {
|
| @@ -276,21 +301,21 @@ TEST_F(PDFObjectsTest, Clone) {
|
| TEST_F(PDFObjectsTest, GetType) {
|
| // Check for direct objects.
|
| for (size_t i = 0; i < m_DirectObjs.size(); ++i)
|
| - EXPECT_EQ(m_DirectObjs[i]->GetType(), m_DirectObjTypes[i]);
|
| + EXPECT_EQ(m_DirectObjTypes[i], m_DirectObjs[i]->GetType());
|
|
|
| // Check indirect references.
|
| for (const auto& it : m_RefObjs)
|
| - EXPECT_EQ(it->GetType(), CPDF_Object::REFERENCE);
|
| + EXPECT_EQ(CPDF_Object::REFERENCE, it->GetType());
|
| }
|
|
|
| TEST_F(PDFObjectsTest, GetDirect) {
|
| // Check for direct objects.
|
| for (size_t i = 0; i < m_DirectObjs.size(); ++i)
|
| - EXPECT_EQ(m_DirectObjs[i]->GetDirect(), m_DirectObjs[i].get());
|
| + EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->GetDirect());
|
|
|
| // Check indirect references.
|
| for (size_t i = 0; i < m_RefObjs.size(); ++i)
|
| - EXPECT_EQ(m_RefObjs[i]->GetDirect(), m_IndirectObjs[i]);
|
| + EXPECT_EQ(m_IndirectObjs[i], m_RefObjs[i]->GetDirect());
|
| }
|
|
|
| TEST_F(PDFObjectsTest, SetString) {
|
| @@ -301,7 +326,7 @@ TEST_F(PDFObjectsTest, SetString) {
|
| "changed", "", "NewName"};
|
| for (size_t i = 0; i < FX_ArraySize(set_values); ++i) {
|
| m_DirectObjs[i]->SetString(set_values[i]);
|
| - EXPECT_STREQ(m_DirectObjs[i]->GetString().c_str(), expected[i]);
|
| + EXPECT_STREQ(expected[i], m_DirectObjs[i]->GetString().c_str());
|
| }
|
| }
|
|
|
| @@ -310,66 +335,432 @@ TEST_F(PDFObjectsTest, IsTypeAndAsType) {
|
| for (size_t i = 0; i < m_DirectObjs.size(); ++i) {
|
| if (m_DirectObjTypes[i] == CPDF_Object::ARRAY) {
|
| EXPECT_TRUE(m_DirectObjs[i]->IsArray());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsArray(), m_DirectObjs[i].get());
|
| + EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsArray());
|
| } else {
|
| EXPECT_FALSE(m_DirectObjs[i]->IsArray());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsArray(), nullptr);
|
| + EXPECT_EQ(nullptr, m_DirectObjs[i]->AsArray());
|
| }
|
|
|
| if (m_DirectObjTypes[i] == CPDF_Object::BOOLEAN) {
|
| EXPECT_TRUE(m_DirectObjs[i]->IsBoolean());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsBoolean(), m_DirectObjs[i].get());
|
| + EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsBoolean());
|
| } else {
|
| EXPECT_FALSE(m_DirectObjs[i]->IsBoolean());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsBoolean(), nullptr);
|
| + EXPECT_EQ(nullptr, m_DirectObjs[i]->AsBoolean());
|
| }
|
|
|
| if (m_DirectObjTypes[i] == CPDF_Object::NAME) {
|
| EXPECT_TRUE(m_DirectObjs[i]->IsName());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsName(), m_DirectObjs[i].get());
|
| + EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsName());
|
| } else {
|
| EXPECT_FALSE(m_DirectObjs[i]->IsName());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsName(), nullptr);
|
| + EXPECT_EQ(nullptr, m_DirectObjs[i]->AsName());
|
| }
|
|
|
| if (m_DirectObjTypes[i] == CPDF_Object::NUMBER) {
|
| EXPECT_TRUE(m_DirectObjs[i]->IsNumber());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsNumber(), m_DirectObjs[i].get());
|
| + EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsNumber());
|
| } else {
|
| EXPECT_FALSE(m_DirectObjs[i]->IsNumber());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsNumber(), nullptr);
|
| + EXPECT_EQ(nullptr, m_DirectObjs[i]->AsNumber());
|
| }
|
|
|
| if (m_DirectObjTypes[i] == CPDF_Object::STRING) {
|
| EXPECT_TRUE(m_DirectObjs[i]->IsString());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsString(), m_DirectObjs[i].get());
|
| + EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsString());
|
| } else {
|
| EXPECT_FALSE(m_DirectObjs[i]->IsString());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsString(), nullptr);
|
| + EXPECT_EQ(nullptr, m_DirectObjs[i]->AsString());
|
| }
|
|
|
| if (m_DirectObjTypes[i] == CPDF_Object::DICTIONARY) {
|
| EXPECT_TRUE(m_DirectObjs[i]->IsDictionary());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsDictionary(), m_DirectObjs[i].get());
|
| + EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsDictionary());
|
| } else {
|
| EXPECT_FALSE(m_DirectObjs[i]->IsDictionary());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsDictionary(), nullptr);
|
| + EXPECT_EQ(nullptr, m_DirectObjs[i]->AsDictionary());
|
| }
|
|
|
| if (m_DirectObjTypes[i] == CPDF_Object::STREAM) {
|
| EXPECT_TRUE(m_DirectObjs[i]->IsStream());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsStream(), m_DirectObjs[i].get());
|
| + EXPECT_EQ(m_DirectObjs[i].get(), m_DirectObjs[i]->AsStream());
|
| } else {
|
| EXPECT_FALSE(m_DirectObjs[i]->IsStream());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsStream(), nullptr);
|
| + EXPECT_EQ(nullptr, m_DirectObjs[i]->AsStream());
|
| }
|
|
|
| EXPECT_FALSE(m_DirectObjs[i]->IsReference());
|
| - EXPECT_EQ(m_DirectObjs[i]->AsReference(), nullptr);
|
| + EXPECT_EQ(nullptr, m_DirectObjs[i]->AsReference());
|
| }
|
| // Check indirect references.
|
| for (size_t i = 0; i < m_RefObjs.size(); ++i) {
|
| EXPECT_TRUE(m_RefObjs[i]->IsReference());
|
| - EXPECT_EQ(m_RefObjs[i]->AsReference(), m_RefObjs[i].get());
|
| + EXPECT_EQ(m_RefObjs[i].get(), m_RefObjs[i]->AsReference());
|
| + }
|
| +}
|
| +
|
| +TEST(PDFArrayTest, GetMatrix) {
|
| + float elems[][6] = {{0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f},
|
| + {1, 2, 3, 4, 5, 6},
|
| + {2.3f, 4.05f, 3, -2, -3, 0.0f},
|
| + {0.05f, 0.1f, 0.56f, 0.67f, 1.34f, 99.9f}};
|
| + for (size_t i = 0; i < FX_ArraySize(elems); ++i) {
|
| + ScopedArray arr(new CPDF_Array);
|
| + CFX_Matrix matrix(elems[i][0], elems[i][1], elems[i][2], elems[i][3],
|
| + elems[i][4], elems[i][5]);
|
| + for (size_t j = 0; j < 6; ++j)
|
| + arr->AddNumber(elems[i][j]);
|
| + CFX_Matrix arr_matrix = arr->GetMatrix();
|
| + EXPECT_EQ(matrix.GetA(), arr_matrix.GetA());
|
| + EXPECT_EQ(matrix.GetB(), arr_matrix.GetB());
|
| + EXPECT_EQ(matrix.GetC(), arr_matrix.GetC());
|
| + EXPECT_EQ(matrix.GetD(), arr_matrix.GetD());
|
| + EXPECT_EQ(matrix.GetE(), arr_matrix.GetE());
|
| + EXPECT_EQ(matrix.GetF(), arr_matrix.GetF());
|
| + }
|
| +}
|
| +
|
| +TEST(PDFArrayTest, GetRect) {
|
| + float elems[][4] = {{0.0f, 0.0f, 0.0f, 0.0f},
|
| + {1, 2, 5, 6},
|
| + {2.3f, 4.05f, -3, 0.0f},
|
| + {0.05f, 0.1f, 1.34f, 99.9f}};
|
| + for (size_t i = 0; i < FX_ArraySize(elems); ++i) {
|
| + ScopedArray arr(new CPDF_Array);
|
| + CFX_FloatRect rect(elems[i]);
|
| + for (size_t j = 0; j < 4; ++j)
|
| + arr->AddNumber(elems[i][j]);
|
| + CFX_FloatRect arr_rect = arr->GetRect();
|
| + EXPECT_EQ(rect.left, arr_rect.left);
|
| + EXPECT_EQ(rect.right, arr_rect.right);
|
| + EXPECT_EQ(rect.bottom, arr_rect.bottom);
|
| + EXPECT_EQ(rect.top, arr_rect.top);
|
| + }
|
| +}
|
| +
|
| +TEST(PDFArrayTest, GetTypeAt) {
|
| + {
|
| + // Boolean array.
|
| + bool vals[] = {true, false, false, true, true};
|
| + ScopedArray arr(new CPDF_Array);
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i)
|
| + arr->InsertAt(i, new CPDF_Boolean(vals[i]));
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i) {
|
| + TestArrayAccessors(arr.get(), i, // Array and index.
|
| + vals[i] ? "true" : "false", // String value.
|
| + nullptr, // Const string value.
|
| + vals[i] ? 1 : 0, // Integer value.
|
| + 0, // Float value.
|
| + nullptr, // Array value.
|
| + nullptr, // Dictionary value.
|
| + nullptr); // Stream value.
|
| + }
|
| + }
|
| + {
|
| + // Integer array.
|
| + int vals[] = {10, 0, -345, 2089345456, -1000000000, 567, 93658767};
|
| + ScopedArray arr(new CPDF_Array);
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i)
|
| + arr->InsertAt(i, new CPDF_Number(vals[i]));
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i) {
|
| + char buf[33];
|
| + TestArrayAccessors(arr.get(), i, // Array and index.
|
| + FXSYS_itoa(vals[i], buf, 10), // String value.
|
| + nullptr, // Const string value.
|
| + vals[i], // Integer value.
|
| + vals[i], // Float value.
|
| + nullptr, // Array value.
|
| + nullptr, // Dictionary value.
|
| + nullptr); // Stream value.
|
| + }
|
| + }
|
| + {
|
| + // Float array.
|
| + float vals[] = {0.0f, 0, 10, 10.0f, 0.0345f,
|
| + 897.34f, -2.5f, -1.0f, -345.0f, -0.0f};
|
| + const char* expected_str[] = {"0", "0", "10", "10", "0.0345",
|
| + "897.34", "-2.5", "-1", "-345", "0"};
|
| + ScopedArray arr(new CPDF_Array);
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i) {
|
| + arr->InsertAt(i, new CPDF_Number(vals[i]));
|
| + }
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i) {
|
| + TestArrayAccessors(arr.get(), i, // Array and index.
|
| + expected_str[i], // String value.
|
| + nullptr, // Const string value.
|
| + vals[i], // Integer value.
|
| + vals[i], // Float value.
|
| + nullptr, // Array value.
|
| + nullptr, // Dictionary value.
|
| + nullptr); // Stream value.
|
| + }
|
| + }
|
| + {
|
| + // String and name array
|
| + const char* const vals[] = {"this", "adsde$%^", "\r\t", "\"012",
|
| + ".", "EYREW", "It is a joke :)"};
|
| + ScopedArray string_array(new CPDF_Array);
|
| + ScopedArray name_array(new CPDF_Array);
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i) {
|
| + string_array->InsertAt(i, new CPDF_String(vals[i], false));
|
| + name_array->InsertAt(i, new CPDF_Name(vals[i]));
|
| + }
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i) {
|
| + TestArrayAccessors(string_array.get(), i, // Array and index.
|
| + vals[i], // String value.
|
| + vals[i], // Const string value.
|
| + 0, // Integer value.
|
| + 0, // Float value.
|
| + nullptr, // Array value.
|
| + nullptr, // Dictionary value.
|
| + nullptr); // Stream value.
|
| + TestArrayAccessors(name_array.get(), i, // Array and index.
|
| + vals[i], // String value.
|
| + vals[i], // Const string value.
|
| + 0, // Integer value.
|
| + 0, // Float value.
|
| + nullptr, // Array value.
|
| + nullptr, // Dictionary value.
|
| + nullptr); // Stream value.
|
| + }
|
| + }
|
| + {
|
| + // Null element array.
|
| + ScopedArray arr(new CPDF_Array);
|
| + for (size_t i = 0; i < 3; ++i)
|
| + arr->InsertAt(i, new CPDF_Null);
|
| + for (size_t i = 0; i < 3; ++i) {
|
| + TestArrayAccessors(arr.get(), i, // Array and index.
|
| + "", // String value.
|
| + nullptr, // Const string value.
|
| + 0, // Integer value.
|
| + 0, // Float value.
|
| + nullptr, // Array value.
|
| + nullptr, // Dictionary value.
|
| + nullptr); // Stream value.
|
| + }
|
| + }
|
| + {
|
| + // Array of array.
|
| + CPDF_Array* vals[3];
|
| + ScopedArray arr(new CPDF_Array);
|
| + for (size_t i = 0; i < 3; ++i) {
|
| + vals[i] = new CPDF_Array;
|
| + for (size_t j = 0; j < 3; ++j) {
|
| + int value = j + 100;
|
| + vals[i]->InsertAt(i, new CPDF_Number(value));
|
| + }
|
| + arr->InsertAt(i, vals[i]);
|
| + }
|
| + for (size_t i = 0; i < 3; ++i) {
|
| + TestArrayAccessors(arr.get(), i, // Array and index.
|
| + "", // String value.
|
| + nullptr, // Const string value.
|
| + 0, // Integer value.
|
| + 0, // Float value.
|
| + vals[i], // Array value.
|
| + nullptr, // Dictionary value.
|
| + nullptr); // Stream value.
|
| + }
|
| + }
|
| + {
|
| + // Dictionary array.
|
| + CPDF_Dictionary* vals[3];
|
| + ScopedArray arr(new CPDF_Array);
|
| + for (size_t i = 0; i < 3; ++i) {
|
| + vals[i] = new CPDF_Dictionary;
|
| + for (size_t j = 0; j < 3; ++j) {
|
| + std::string key("key");
|
| + char buf[33];
|
| + key.append(FXSYS_itoa(j, buf, 10));
|
| + int value = j + 200;
|
| + vals[i]->SetAt(CFX_ByteStringC(key.c_str()), new CPDF_Number(value));
|
| + }
|
| + arr->InsertAt(i, vals[i]);
|
| + }
|
| + for (size_t i = 0; i < 3; ++i) {
|
| + TestArrayAccessors(arr.get(), i, // Array and index.
|
| + "", // String value.
|
| + nullptr, // Const string value.
|
| + 0, // Integer value.
|
| + 0, // Float value.
|
| + nullptr, // Array value.
|
| + vals[i], // Dictionary value.
|
| + nullptr); // Stream value.
|
| + }
|
| + }
|
| + {
|
| + // Stream array.
|
| + CPDF_Dictionary* vals[3];
|
| + CPDF_Stream* stream_vals[3];
|
| + ScopedArray arr(new CPDF_Array);
|
| + for (size_t i = 0; i < 3; ++i) {
|
| + vals[i] = new CPDF_Dictionary;
|
| + for (size_t j = 0; j < 3; ++j) {
|
| + std::string key("key");
|
| + char buf[33];
|
| + key.append(FXSYS_itoa(j, buf, 10));
|
| + int value = j + 200;
|
| + vals[i]->SetAt(CFX_ByteStringC(key.c_str()), new CPDF_Number(value));
|
| + }
|
| + uint8_t content[] = "content: this is a stream";
|
| + size_t data_size = FX_ArraySize(content);
|
| + uint8_t* data = reinterpret_cast<uint8_t*>(malloc(data_size));
|
| + memcpy(data, content, data_size);
|
| + stream_vals[i] = new CPDF_Stream(data, data_size, vals[i]);
|
| + arr->InsertAt(i, stream_vals[i]);
|
| + }
|
| + for (size_t i = 0; i < 3; ++i) {
|
| + TestArrayAccessors(arr.get(), i, // Array and index.
|
| + "", // String value.
|
| + nullptr, // Const string value.
|
| + 0, // Integer value.
|
| + 0, // Float value.
|
| + nullptr, // Array value.
|
| + vals[i], // Dictionary value.
|
| + stream_vals[i]); // Stream value.
|
| + }
|
| + }
|
| + {
|
| + // Mixed array.
|
| + ScopedArray arr(new CPDF_Array);
|
| + // Array arr will take ownership of all the objects inserted.
|
| + arr->InsertAt(0, new CPDF_Boolean(true));
|
| + arr->InsertAt(1, new CPDF_Boolean(false));
|
| + arr->InsertAt(2, new CPDF_Number(0));
|
| + arr->InsertAt(3, new CPDF_Number(-1234));
|
| + arr->InsertAt(4, new CPDF_Number(2345.0f));
|
| + arr->InsertAt(5, new CPDF_Number(0.05f));
|
| + arr->InsertAt(6, new CPDF_String("", false));
|
| + arr->InsertAt(7, new CPDF_String("It is a test!", false));
|
| + arr->InsertAt(8, new CPDF_Name("NAME"));
|
| + arr->InsertAt(9, new CPDF_Name("test"));
|
| + arr->InsertAt(10, new CPDF_Null());
|
| + CPDF_Array* arr_val = new CPDF_Array;
|
| + arr_val->AddNumber(1);
|
| + arr_val->AddNumber(2);
|
| + arr->InsertAt(11, arr_val);
|
| + CPDF_Dictionary* dict_val = new CPDF_Dictionary;
|
| + dict_val->SetAt("key1", new CPDF_String("Linda", false));
|
| + dict_val->SetAt("key2", new CPDF_String("Zoe", false));
|
| + arr->InsertAt(12, dict_val);
|
| + CPDF_Dictionary* stream_dict = new CPDF_Dictionary;
|
| + stream_dict->SetAt("key1", new CPDF_String("John", false));
|
| + stream_dict->SetAt("key2", new CPDF_String("King", false));
|
| + uint8_t data[] = "A stream for test";
|
| + // The data buffer will be owned by stream object, so it needs to be
|
| + // dynamically allocated.
|
| + size_t buf_size = sizeof(data);
|
| + uint8_t* buf = reinterpret_cast<uint8_t*>(malloc(buf_size));
|
| + memcpy(buf, data, buf_size);
|
| + CPDF_Stream* stream_val = new CPDF_Stream(buf, buf_size, stream_dict);
|
| + arr->InsertAt(13, stream_val);
|
| + const char* const expected_str[] = {
|
| + "true", "false", "0", "-1234", "2345", "0.05", "",
|
| + "It is a test!", "NAME", "test", "", "", "", ""};
|
| + const char* const expected_cstr[] = {
|
| + nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr,
|
| + "It is a test!", "NAME", "test", nullptr, nullptr, nullptr, nullptr};
|
| + const int expected_int[] = {1, 0, 0, -1234, 2345, 0, 0,
|
| + 0, 0, 0, 0, 0, 0, 0};
|
| + const float expected_float[] = {0, 0, 0, -1234, 2345, 0.05f, 0,
|
| + 0, 0, 0, 0, 0, 0, 0};
|
| + for (size_t i = 0; i < arr->GetCount(); ++i) {
|
| + EXPECT_STREQ(expected_str[i], arr->GetStringAt(i).c_str());
|
| + EXPECT_STREQ(expected_cstr[i], arr->GetConstStringAt(i).GetCStr());
|
| + EXPECT_EQ(expected_int[i], arr->GetIntegerAt(i));
|
| + EXPECT_EQ(expected_float[i], arr->GetNumberAt(i));
|
| + EXPECT_EQ(expected_float[i], arr->GetFloatAt(i));
|
| + if (i == 11)
|
| + EXPECT_EQ(arr_val, arr->GetArrayAt(i));
|
| + else
|
| + EXPECT_EQ(nullptr, arr->GetArrayAt(i));
|
| + if (i == 13) {
|
| + EXPECT_EQ(stream_dict, arr->GetDictAt(i));
|
| + EXPECT_EQ(stream_val, arr->GetStreamAt(i));
|
| + } else {
|
| + EXPECT_EQ(nullptr, arr->GetStreamAt(i));
|
| + if (i == 12)
|
| + EXPECT_EQ(dict_val, arr->GetDictAt(i));
|
| + else
|
| + EXPECT_EQ(nullptr, arr->GetDictAt(i));
|
| + }
|
| + }
|
| + }
|
| +}
|
| +
|
| +TEST(PDFArrayTest, AddNumber) {
|
| + float vals[] = {1.0f, -1.0f, 0, 0.456734f,
|
| + 12345.54321f, 0.5f, 1000, 0.000045f};
|
| + ScopedArray arr(new CPDF_Array);
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i)
|
| + arr->AddNumber(vals[i]);
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i) {
|
| + EXPECT_EQ(CPDF_Object::NUMBER, arr->GetElement(i)->GetType());
|
| + EXPECT_EQ(vals[i], arr->GetElement(i)->GetNumber());
|
| + }
|
| +}
|
| +
|
| +TEST(PDFArrayTest, AddInteger) {
|
| + int vals[] = {0, 1, 934435456, 876, 10000, -1, -24354656, -100};
|
| + ScopedArray arr(new CPDF_Array);
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i)
|
| + arr->AddInteger(vals[i]);
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i) {
|
| + EXPECT_EQ(CPDF_Object::NUMBER, arr->GetElement(i)->GetType());
|
| + EXPECT_EQ(vals[i], arr->GetElement(i)->GetNumber());
|
| + }
|
| +}
|
| +
|
| +TEST(PDFArrayTest, AddStringAndName) {
|
| + const char* vals[] = {"", "a", "ehjhRIOYTTFdfcdnv", "122323",
|
| + "$#%^&**", " ", "This is a test.\r\n"};
|
| + ScopedArray string_array(new CPDF_Array);
|
| + ScopedArray name_array(new CPDF_Array);
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i) {
|
| + string_array->AddString(vals[i]);
|
| + name_array->AddName(vals[i]);
|
| + }
|
| + for (size_t i = 0; i < FX_ArraySize(vals); ++i) {
|
| + EXPECT_EQ(CPDF_Object::STRING, string_array->GetElement(i)->GetType());
|
| + EXPECT_STREQ(vals[i], string_array->GetElement(i)->GetString().c_str());
|
| + EXPECT_EQ(CPDF_Object::NAME, name_array->GetElement(i)->GetType());
|
| + EXPECT_STREQ(vals[i], name_array->GetElement(i)->GetString().c_str());
|
| + }
|
| +}
|
| +
|
| +TEST(PDFArrayTest, AddReferenceAndGetElement) {
|
| + std::unique_ptr<CPDF_IndirectObjectHolder> holder(
|
| + new CPDF_IndirectObjectHolder(nullptr));
|
| + CPDF_Boolean* boolean_obj = new CPDF_Boolean(true);
|
| + CPDF_Number* int_obj = new CPDF_Number(-1234);
|
| + CPDF_Number* float_obj = new CPDF_Number(2345.089f);
|
| + CPDF_String* str_obj = new CPDF_String("Adsfdsf 343434 %&&*\n", false);
|
| + CPDF_Name* name_obj = new CPDF_Name("Title:");
|
| + CPDF_Null* null_obj = new CPDF_Null();
|
| + CPDF_Object* indirect_objs[] = {boolean_obj, int_obj, float_obj,
|
| + str_obj, name_obj, null_obj};
|
| + unsigned int obj_nums[] = {2, 4, 7, 2345, 799887, 1};
|
| + ScopedArray arr(new CPDF_Array);
|
| + ScopedArray arr1(new CPDF_Array);
|
| + // Create two arrays of references by different AddReference() APIs.
|
| + for (size_t i = 0; i < FX_ArraySize(indirect_objs); ++i) {
|
| + // All the indirect objects inserted will be owned by holder.
|
| + holder->InsertIndirectObject(obj_nums[i], indirect_objs[i]);
|
| + arr->AddReference(holder.get(), obj_nums[i]);
|
| + arr1->AddReference(holder.get(), indirect_objs[i]);
|
| + }
|
| + // Check indirect objects.
|
| + for (size_t i = 0; i < FX_ArraySize(obj_nums); ++i)
|
| + EXPECT_EQ(indirect_objs[i], holder->GetIndirectObject(obj_nums[i]));
|
| + // Check arrays.
|
| + EXPECT_EQ(arr->GetCount(), arr1->GetCount());
|
| + for (size_t i = 0; i < arr->GetCount(); ++i) {
|
| + EXPECT_EQ(CPDF_Object::REFERENCE, arr->GetElement(i)->GetType());
|
| + EXPECT_EQ(indirect_objs[i], arr->GetElement(i)->GetDirect());
|
| + EXPECT_EQ(indirect_objs[i], arr->GetElementValue(i));
|
| + EXPECT_EQ(CPDF_Object::REFERENCE, arr1->GetElement(i)->GetType());
|
| + EXPECT_EQ(indirect_objs[i], arr1->GetElement(i)->GetDirect());
|
| + EXPECT_EQ(indirect_objs[i], arr1->GetElementValue(i));
|
| }
|
| }
|
|
|