1 #include "Test.h"
2 #include "TestClassDef.h"
3
4 #include "SkCanvas.h"
5 #include "SkDocument.h"
6 #include "SkOSFile.h"
7 #include "SkStream.h"
8
test_empty(skiatest::Reporter * reporter)9 static void test_empty(skiatest::Reporter* reporter) {
10 SkDynamicMemoryWStream stream;
11
12 SkAutoTUnref<SkDocument> doc(SkDocument::CreatePDF(&stream));
13
14 doc->close();
15
16 REPORTER_ASSERT(reporter, stream.bytesWritten() == 0);
17 }
18
test_abort(skiatest::Reporter * reporter)19 static void test_abort(skiatest::Reporter* reporter) {
20 SkDynamicMemoryWStream stream;
21 SkAutoTUnref<SkDocument> doc(SkDocument::CreatePDF(&stream));
22
23 SkCanvas* canvas = doc->beginPage(100, 100);
24 canvas->drawColor(SK_ColorRED);
25 doc->endPage();
26
27 doc->abort();
28
29 REPORTER_ASSERT(reporter, stream.bytesWritten() == 0);
30 }
31
test_abortWithFile(skiatest::Reporter * reporter)32 static void test_abortWithFile(skiatest::Reporter* reporter) {
33 SkString tmpDir = skiatest::Test::GetTmpDir();
34
35 if (tmpDir.isEmpty()) {
36 return; // TODO(edisonn): unfortunatelly this pattern is used in other
37 // tests, but if GetTmpDir() starts returning and empty dir
38 // allways, then all these tests will be disabled.
39 }
40
41 SkString path = SkOSPath::SkPathJoin(tmpDir.c_str(), "aborted.pdf");
42
43 // Make sure doc's destructor is called to flush.
44 {
45 SkAutoTUnref<SkDocument> doc(SkDocument::CreatePDF(path.c_str()));
46
47 SkCanvas* canvas = doc->beginPage(100, 100);
48 canvas->drawColor(SK_ColorRED);
49 doc->endPage();
50
51 doc->abort();
52 }
53
54 FILE* file = fopen(path.c_str(), "r");
55 // The created file should be empty.
56 char buffer[100];
57 REPORTER_ASSERT(reporter, fread(buffer, 1, 1, file) == 0);
58 fclose(file);
59 }
60
test_file(skiatest::Reporter * reporter)61 static void test_file(skiatest::Reporter* reporter) {
62 SkString tmpDir = skiatest::Test::GetTmpDir();
63 if (tmpDir.isEmpty()) {
64 return; // TODO(edisonn): unfortunatelly this pattern is used in other
65 // tests, but if GetTmpDir() starts returning and empty dir
66 // allways, then all these tests will be disabled.
67 }
68
69 SkString path = SkOSPath::SkPathJoin(tmpDir.c_str(), "file.pdf");
70
71 SkAutoTUnref<SkDocument> doc(SkDocument::CreatePDF(path.c_str()));
72
73 SkCanvas* canvas = doc->beginPage(100, 100);
74
75 canvas->drawColor(SK_ColorRED);
76 doc->endPage();
77 doc->close();
78
79 FILE* file = fopen(path.c_str(), "r");
80 REPORTER_ASSERT(reporter, file != NULL);
81 char header[100];
82 REPORTER_ASSERT(reporter, fread(header, 4, 1, file) != 0);
83 REPORTER_ASSERT(reporter, strncmp(header, "%PDF", 4) == 0);
84 fclose(file);
85 }
86
test_close(skiatest::Reporter * reporter)87 static void test_close(skiatest::Reporter* reporter) {
88 SkDynamicMemoryWStream stream;
89 SkAutoTUnref<SkDocument> doc(SkDocument::CreatePDF(&stream));
90
91 SkCanvas* canvas = doc->beginPage(100, 100);
92 canvas->drawColor(SK_ColorRED);
93 doc->endPage();
94
95 doc->close();
96
97 REPORTER_ASSERT(reporter, stream.bytesWritten() != 0);
98 }
99
DEF_TEST(document_tests,reporter)100 DEF_TEST(document_tests, reporter) {
101 test_empty(reporter);
102 test_abort(reporter);
103 test_abortWithFile(reporter);
104 test_file(reporter);
105 test_close(reporter);
106 }
107