Serenity Operating System
1/*
2 * Copyright (c) 2021, Simon Woertz <simon@woertz.at>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#include <AK/DeprecatedString.h>
8#include <AK/Forward.h>
9#include <LibCore/MappedFile.h>
10#include <LibPDF/Document.h>
11#include <LibTest/Macros.h>
12#include <LibTest/TestCase.h>
13
14TEST_CASE(linearized_pdf)
15{
16 auto file = Core::MappedFile::map("linearized.pdf"sv).release_value();
17 auto document = PDF::Document::create(file->bytes());
18 EXPECT(!document.is_error());
19 EXPECT(!document.value()->initialize().is_error());
20 EXPECT_EQ(document.value()->get_page_count(), 1U);
21}
22
23TEST_CASE(non_linearized_pdf)
24{
25 auto file = Core::MappedFile::map("non-linearized.pdf"sv).release_value();
26 auto document = PDF::Document::create(file->bytes());
27 EXPECT(!document.is_error());
28 EXPECT(!document.value()->initialize().is_error());
29 EXPECT_EQ(document.value()->get_page_count(), 1U);
30}
31
32TEST_CASE(complex_pdf)
33{
34 auto file = Core::MappedFile::map("complex.pdf"sv).release_value();
35 auto document = PDF::Document::create(file->bytes());
36 EXPECT(!document.is_error());
37 EXPECT(!document.value()->initialize().is_error());
38 EXPECT_EQ(document.value()->get_page_count(), 3U);
39}
40
41TEST_CASE(empty_file_issue_10702)
42{
43 AK::ReadonlyBytes empty;
44 auto document = PDF::Document::create(empty);
45 EXPECT(document.is_error());
46}
47
48TEST_CASE(truncated_pdf_header_issue_10717)
49{
50 AK::DeprecatedString string { "%PDF-2.11%" };
51 auto document = PDF::Document::create(string.bytes());
52 EXPECT(document.is_error());
53}