You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
3.4 KiB
101 lines
3.4 KiB
// 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_save.h" |
|
|
|
#include <string> |
|
|
|
#include "core/fxcrt/fx_string.h" |
|
#include "public/fpdf_edit.h" |
|
#include "public/fpdf_ppo.h" |
|
#include "public/fpdfview.h" |
|
#include "testing/embedder_test.h" |
|
#include "testing/gmock/include/gmock/gmock-matchers.h" |
|
#include "testing/gtest/include/gtest/gtest.h" |
|
#include "testing/test_support.h" |
|
|
|
class FPDFSaveEmbedderTest : public EmbedderTest {}; |
|
|
|
TEST_F(FPDFSaveEmbedderTest, SaveSimpleDoc) { |
|
EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
|
EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
|
EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.7\r\n")); |
|
EXPECT_EQ(805u, GetString().length()); |
|
} |
|
|
|
TEST_F(FPDFSaveEmbedderTest, SaveSimpleDocWithVersion) { |
|
EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
|
EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, 14)); |
|
EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.4\r\n")); |
|
EXPECT_EQ(805u, GetString().length()); |
|
} |
|
TEST_F(FPDFSaveEmbedderTest, SaveSimpleDocWithBadVersion) { |
|
EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
|
EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, -1)); |
|
EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.7\r\n")); |
|
|
|
ClearString(); |
|
EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, 0)); |
|
EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.7\r\n")); |
|
|
|
ClearString(); |
|
EXPECT_TRUE(FPDF_SaveWithVersion(document(), this, 0, 18)); |
|
EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.7\r\n")); |
|
} |
|
|
|
TEST_F(FPDFSaveEmbedderTest, SaveCopiedDoc) { |
|
EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
|
|
|
FPDF_PAGE page = LoadPage(0); |
|
EXPECT_TRUE(page); |
|
|
|
FPDF_DOCUMENT output_doc = FPDF_CreateNewDocument(); |
|
EXPECT_TRUE(output_doc); |
|
EXPECT_TRUE(FPDF_ImportPages(output_doc, document(), "1", 0)); |
|
EXPECT_TRUE(FPDF_SaveAsCopy(output_doc, this, 0)); |
|
FPDF_CloseDocument(output_doc); |
|
|
|
UnloadPage(page); |
|
} |
|
|
|
TEST_F(FPDFSaveEmbedderTest, SaveLinearizedDoc) { |
|
const int kPageCount = 3; |
|
std::string original_md5[kPageCount]; |
|
|
|
EXPECT_TRUE(OpenDocument("linearized.pdf")); |
|
for (int i = 0; i < kPageCount; ++i) { |
|
FPDF_PAGE page = LoadPage(i); |
|
FPDF_BITMAP bitmap = RenderPage(page); |
|
EXPECT_EQ(612, FPDFBitmap_GetWidth(bitmap)); |
|
EXPECT_EQ(792, FPDFBitmap_GetHeight(bitmap)); |
|
original_md5[i] = HashBitmap(bitmap); |
|
FPDFBitmap_Destroy(bitmap); |
|
UnloadPage(page); |
|
} |
|
|
|
EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
|
EXPECT_THAT(GetString(), testing::StartsWith("%PDF-1.6\r\n")); |
|
EXPECT_THAT(GetString(), testing::HasSubstr("/Root ")); |
|
EXPECT_THAT(GetString(), testing::HasSubstr("/Info ")); |
|
EXPECT_EQ(8219u, GetString().length()); |
|
|
|
// Make sure new document renders the same as the old one. |
|
EXPECT_TRUE(OpenSavedDocument()); |
|
for (int i = 0; i < kPageCount; ++i) { |
|
FPDF_PAGE page = LoadSavedPage(i); |
|
FPDF_BITMAP bitmap = RenderSavedPage(page); |
|
EXPECT_EQ(original_md5[i], HashBitmap(bitmap)); |
|
FPDFBitmap_Destroy(bitmap); |
|
CloseSavedPage(page); |
|
} |
|
CloseSavedDocument(); |
|
} |
|
|
|
TEST_F(FPDFSaveEmbedderTest, BUG_342) { |
|
EXPECT_TRUE(OpenDocument("hello_world.pdf")); |
|
EXPECT_TRUE(FPDF_SaveAsCopy(document(), this, 0)); |
|
EXPECT_THAT(GetString(), testing::HasSubstr("0000000000 65535 f\r\n")); |
|
EXPECT_THAT(GetString(), |
|
testing::Not(testing::HasSubstr("0000000000 65536 f\r\n"))); |
|
}
|
|
|