2011-06-13 Chris Rogers <crogers@google.com>
[WebKit-https.git] / Tools / DumpRenderTree / chromium / TestEventPrinter.cpp
1 /*
2  * Copyright (C) 2010 Google Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are
6  * met:
7  *
8  *     * Redistributions of source code must retain the above copyright
9  * notice, this list of conditions and the following disclaimer.
10  *     * Redistributions in binary form must reproduce the above
11  * copyright notice, this list of conditions and the following disclaimer
12  * in the documentation and/or other materials provided with the
13  * distribution.
14  *     * Neither the name of Google Inc. nor the names of its
15  * contributors may be used to endorse or promote products derived from
16  * this software without specific prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
21  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
25  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
26  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
27  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
28  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29  */
30
31 #include "config.h"
32 #include "TestEventPrinter.h"
33
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <wtf/Assertions.h>
37
38 class DRTPrinter : public TestEventPrinter {
39 public:
40     DRTPrinter() { }
41     void handleTestHeader(const char* url) const;
42     void handleTimedOut() const;
43     void handleTextHeader() const;
44     void handleTextFooter() const;
45     void handleAudioHeader() const;
46     void handleImage(const char* actualHash, const char* expectedHash, const unsigned char* imageData, size_t imageSize, const char* fileName) const;
47     void handleImageFooter() const;
48     void handleTestFooter(bool dumpedAnything) const;
49 };
50
51 class TestShellPrinter : public TestEventPrinter {
52 public:
53     TestShellPrinter() { }
54     void handleTestHeader(const char* url) const;
55     void handleTimedOut() const;
56     void handleTextHeader() const;
57     void handleTextFooter() const;
58     void handleAudioHeader() const;
59     void handleImage(const char* actualHash, const char* expectedHash, const unsigned char* imageData, size_t imageSize, const char* fileName) const;
60     void handleImageFooter() const;
61     void handleTestFooter(bool dumpedAnything) const;
62 };
63
64 TestEventPrinter::~TestEventPrinter()
65 {
66 }
67
68 PassOwnPtr<TestEventPrinter> TestEventPrinter::createDRTPrinter()
69 {
70     return adoptPtr(new DRTPrinter);
71 }
72
73 PassOwnPtr<TestEventPrinter> TestEventPrinter::createTestShellPrinter()
74 {
75     return adoptPtr(new TestShellPrinter);
76 }
77
78 // ----------------------------------------------------------------
79
80 void DRTPrinter::handleTestHeader(const char*) const
81 {
82 }
83
84 void DRTPrinter::handleTimedOut() const
85 {
86     fprintf(stderr, "FAIL: Timed out waiting for notifyDone to be called\n");
87     fprintf(stdout, "FAIL: Timed out waiting for notifyDone to be called\n");
88 }
89
90 void DRTPrinter::handleTextHeader() const
91 {
92     printf("Content-Type: text/plain\n");
93 }
94
95 void DRTPrinter::handleTextFooter() const
96 {
97     printf("#EOF\n");
98 }
99
100 void DRTPrinter::handleAudioHeader() const
101 {
102     printf("Content-Type: audio/wav\n");
103     printf("Content-Transfer-Encoding: base64\n");            
104 }
105
106 void DRTPrinter::handleImage(const char* actualHash, const char* expectedHash, const unsigned char* imageData, size_t imageSize, const char*) const
107 {
108     ASSERT(actualHash);
109     printf("\nActualHash: %s\n", actualHash);
110     if (expectedHash && expectedHash[0])
111         printf("\nExpectedHash: %s\n", expectedHash);
112     if (imageData && imageSize) {
113         printf("Content-Type: image/png\n");
114         // Printf formatting for size_t on 32-bit, 64-bit, and on Windows is hard so just cast to an int.
115         printf("Content-Length: %d\n", static_cast<int>(imageSize));
116         if (fwrite(imageData, 1, imageSize, stdout) != imageSize) {
117             fprintf(stderr, "Short write to stdout.\n");
118             exit(1);
119         }
120     }
121 }
122
123 void DRTPrinter::handleImageFooter() const
124 {
125     printf("#EOF\n");
126 }
127
128 void DRTPrinter::handleTestFooter(bool) const
129 {
130 }
131
132 // ----------------------------------------------------------------
133
134 void TestShellPrinter::handleTestHeader(const char* url) const
135 {
136     printf("#URL:%s\n", url);
137 }
138
139 void TestShellPrinter::handleTimedOut() const
140 {
141     puts("#TEST_TIMED_OUT\n");
142 }
143
144 void TestShellPrinter::handleTextHeader() const
145 {
146 }
147
148 void TestShellPrinter::handleTextFooter() const
149 {
150 }
151
152 void TestShellPrinter::handleAudioHeader() const
153 {
154     printf("Content-Type: audio/wav\n");
155     printf("Content-Transfer-Encoding: base64\n");
156 }
157
158 void TestShellPrinter::handleImage(const char* actualHash, const char*, const unsigned char* imageData, size_t imageSize, const char* fileName) const
159 {
160     ASSERT(actualHash);
161     if (imageData && imageSize) {
162         ASSERT(fileName);
163         FILE* fp = fopen(fileName, "wb");
164         if (!fp) {
165             perror(fileName);
166             exit(EXIT_FAILURE);
167         }
168         if (fwrite(imageData, 1, imageSize, fp) != imageSize) {
169             perror(fileName);
170             fclose(fp);
171             exit(EXIT_FAILURE);
172         }
173         fclose(fp);
174     }
175     printf("#MD5:%s\n", actualHash);
176 }
177
178 void TestShellPrinter::handleImageFooter() const
179 {
180 }
181
182 void TestShellPrinter::handleTestFooter(bool dumpedAnything) const
183 {
184     if (dumpedAnything)
185         printf("#EOF\n");
186 }