summaryrefslogtreecommitdiffstats
path: root/printing/pdf_ps_metafile_cairo_unittest.cc
blob: 927306617c18a974ebac898f7349083b8a1f53ab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
// Copyright (c) 2009 The Chromium 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 "printing/pdf_ps_metafile_cairo.h"

#include <fcntl.h>
#include <string>
#include <vector>

#include "base/file_descriptor_posix.h"
#include "base/file_util.h"
#include "base/string_util.h"
#include "testing/gtest/include/gtest/gtest.h"

typedef struct _cairo cairo_t;

class PdfPsTest : public testing::Test {
 protected:
  base::FileDescriptor DevNullFD() {
    return base::FileDescriptor(open("/dev/null", O_WRONLY), true);
  }
};

TEST_F(PdfPsTest, Pdf) {
  // Tests in-renderer constructor.
  printing::PdfPsMetafile pdf(printing::PdfPsMetafile::PDF);
  EXPECT_TRUE(pdf.Init());

  // Renders page 1.
  cairo_t* context = pdf.StartPage(72, 72, 1, 2, 3, 4);
  EXPECT_TRUE(context != NULL);
  EXPECT_EQ(printing::PdfPsMetafile::FromCairoContext(context), &pdf);
  // In theory, we should use Cairo to draw something on |context|.
  EXPECT_TRUE(pdf.FinishPage());

  // Renders page 2.
  context = pdf.StartPage(64, 64, 1, 2, 3, 4);
  EXPECT_TRUE(context != NULL);
  // In theory, we should use Cairo to draw something on |context|.
  EXPECT_TRUE(pdf.FinishPage());

  // Closes the file.
  pdf.Close();

  // Checks data size.
  uint32 size = pdf.GetDataSize();
  EXPECT_GT(size, 0u);

  // Gets resulting data.
  std::vector<char> buffer(size, 0x00);
  pdf.GetData(&buffer.front(), size);

  // Tests another constructor.
  printing::PdfPsMetafile pdf2(printing::PdfPsMetafile::PDF);
  EXPECT_TRUE(pdf2.Init(&buffer.front(), size));

  // Tries to get the first 4 characters from pdf2.
  std::vector<char> buffer2(4, 0x00);
  pdf2.GetData(&buffer2.front(), 4);

  // Tests if the header begins with "%PDF".
  std::string header(&buffer2.front(), 4);
  EXPECT_EQ(header.find("%PDF", 0), 0u);

  // Tests if we can save data.
  EXPECT_TRUE(pdf.SaveTo(DevNullFD()));

  // Test overriding the metafile with raw data.
  printing::PdfPsMetafile pdf3(printing::PdfPsMetafile::PDF);
  EXPECT_TRUE(pdf3.Init());
  context = pdf3.StartPage(72, 72, 1, 2, 3, 4);
  EXPECT_TRUE(context != NULL);
  std::string test_raw_data = "Dummy PDF";
  EXPECT_TRUE(pdf3.SetRawData(test_raw_data.c_str(), test_raw_data.size()));
  EXPECT_TRUE(pdf3.FinishPage());
  pdf3.Close();
  size = pdf3.GetDataSize();
  EXPECT_EQ(test_raw_data.size(), size);
  std::string output;
  pdf3.GetData(WriteInto(&output, size + 1), size);
  EXPECT_EQ(test_raw_data, output);
}

TEST_F(PdfPsTest, Ps) {
  // Tests in-renderer constructor.
  printing::PdfPsMetafile ps(printing::PdfPsMetafile::PS);
  EXPECT_TRUE(ps.Init());

  // Renders page 1.
  cairo_t* context = ps.StartPage(72, 72, 1, 2, 3, 4);
  EXPECT_TRUE(context != NULL);
  EXPECT_EQ(printing::PdfPsMetafile::FromCairoContext(context), &ps);
  // In theory, we should use Cairo to draw something on |context|.
  EXPECT_TRUE(ps.FinishPage());

  // Renders page 2.
  context = ps.StartPage(64, 64, 1, 2, 3, 4);
  EXPECT_TRUE(context != NULL);
  // In theory, we should use Cairo to draw something on |context|.
  EXPECT_TRUE(ps.FinishPage());

  // Closes the file.
  ps.Close();

  // Checks data size.
  uint32 size = ps.GetDataSize();
  EXPECT_GT(size, 0u);

  // Gets resulting data.
  std::vector<char> buffer(size, 0x00);
  ps.GetData(&buffer.front(), size);

  // Tests another constructor.
  printing::PdfPsMetafile ps2(printing::PdfPsMetafile::PS);
  EXPECT_TRUE(ps2.Init(&buffer.front(), size));

  // Tries to get the first 4 characters from ps2.
  std::vector<char> buffer2(4, 0x00);
  ps2.GetData(&buffer2.front(), 4);

  // Tests if the header begins with "%!PS".
  std::string header(&buffer2.front(), 4);
  EXPECT_EQ(header.find("%!PS", 0), 0u);

  // Tests if we can save data.
  EXPECT_TRUE(ps.SaveTo(DevNullFD()));
}