da5df9bdf781250e2ba1e778da2cf8fff7f8bab7
[WebKit-https.git] / Tools / DumpRenderTree / chromium / TestShell.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 "TestShell.h"
33
34 #include "DRTDevToolsAgent.h"
35 #include "DRTDevToolsClient.h"
36 #include "LayoutTestController.h"
37 #include "WebDataSource.h"
38 #include "WebDocument.h"
39 #include "WebElement.h"
40 #include "WebFrame.h"
41 #include "WebHistoryItem.h"
42 #include "WebTestingSupport.h"
43 #include "WebKit.h"
44 #include "WebPermissions.h"
45 #include "WebRuntimeFeatures.h"
46 #include "WebScriptController.h"
47 #include "WebSettings.h"
48 #include "WebSize.h"
49 #include "WebSpeechInputControllerMock.h"
50 #include "WebString.h"
51 #include "WebURLRequest.h"
52 #include "WebURLResponse.h"
53 #include "WebView.h"
54 #include "WebViewHost.h"
55 #include "skia/ext/platform_canvas.h"
56 #include "webkit/support/webkit_support.h"
57 #include "webkit/support/webkit_support_gfx.h"
58 #include <algorithm>
59 #include <cctype>
60 #include <vector>
61 #include <wtf/MD5.h>
62
63 using namespace WebKit;
64 using namespace std;
65
66 // Content area size for newly created windows.
67 static const int testWindowWidth = 800;
68 static const int testWindowHeight = 600;
69
70 // The W3C SVG layout tests use a different size than the other layout tests.
71 static const int SVGTestWindowWidth = 480;
72 static const int SVGTestWindowHeight = 360;
73
74 static const char layoutTestsPattern[] = "/LayoutTests/";
75 static const string::size_type layoutTestsPatternSize = sizeof(layoutTestsPattern) - 1;
76 static const char fileUrlPattern[] = "file:/";
77 static const char fileTestPrefix[] = "(file test):";
78 static const char dataUrlPattern[] = "data:";
79 static const string::size_type dataUrlPatternSize = sizeof(dataUrlPattern) - 1;
80
81 // FIXME: Move this to a common place so that it can be shared with
82 // WebCore::TransparencyWin::makeLayerOpaque().
83 static void makeCanvasOpaque(SkCanvas* canvas)
84 {
85     const SkBitmap& bitmap = canvas->getTopDevice()->accessBitmap(true);
86     ASSERT(bitmap.config() == SkBitmap::kARGB_8888_Config);
87
88     SkAutoLockPixels lock(bitmap);
89     for (int y = 0; y < bitmap.height(); y++) {
90         uint32_t* row = bitmap.getAddr32(0, y);
91         for (int x = 0; x < bitmap.width(); x++)
92             row[x] |= 0xFF000000; // Set alpha bits to 1.
93     }
94 }
95
96 TestShell::TestShell(bool testShellMode)
97     : m_testIsPending(false)
98     , m_testIsPreparing(false)
99     , m_focusedWidget(0)
100     , m_testShellMode(testShellMode)
101     , m_devTools(0)
102     , m_allowExternalPages(false)
103     , m_acceleratedCompositingEnabled(false)
104     , m_compositeToTexture(false)
105     , m_forceCompositingMode(false)
106     , m_accelerated2dCanvasEnabled(false)
107     , m_legacyAccelerated2dCanvasEnabled(false)
108     , m_acceleratedDrawingEnabled(false)
109     , m_stressOpt(false)
110     , m_stressDeopt(false)
111     , m_dumpWhenFinished(true)
112 {
113     WebRuntimeFeatures::enableDataTransferItems(true);
114     WebRuntimeFeatures::enableGeolocation(true);
115     WebRuntimeFeatures::enableIndexedDatabase(true);
116     WebRuntimeFeatures::enableFileSystem(true);
117     WebRuntimeFeatures::enableJavaScriptI18NAPI(true);
118     WebRuntimeFeatures::enableMediaStream(true);
119
120     m_webPermissions = adoptPtr(new WebPermissions());
121     m_accessibilityController = adoptPtr(new AccessibilityController(this));
122     m_layoutTestController = adoptPtr(new LayoutTestController(this));
123     m_eventSender = adoptPtr(new EventSender(this));
124     m_plainTextController = adoptPtr(new PlainTextController());
125     m_textInputController = adoptPtr(new TextInputController(this));
126     m_notificationPresenter = adoptPtr(new NotificationPresenter(this));
127     m_printer = m_testShellMode ? TestEventPrinter::createTestShellPrinter() : TestEventPrinter::createDRTPrinter();
128
129     // 30 second is the same as the value in Mac DRT.
130     // If we use a value smaller than the timeout value of
131     // (new-)run-webkit-tests, (new-)run-webkit-tests misunderstands that a
132     // timed-out DRT process was crashed.
133     m_timeout = 30 * 1000;
134
135     createMainWindow();
136 }
137
138 void TestShell::createMainWindow()
139 {
140     m_drtDevToolsAgent = adoptPtr(new DRTDevToolsAgent);
141     m_webViewHost = adoptPtr(createNewWindow(WebURL(), m_drtDevToolsAgent.get()));
142     m_webView = m_webViewHost->webView();
143     m_drtDevToolsAgent->setWebView(m_webView);
144 }
145
146 TestShell::~TestShell()
147 {
148     // Note: DevTools are closed together with all the other windows in the
149     // windows list.
150
151     // Destroy the WebView before its WebViewHost.
152     m_drtDevToolsAgent->setWebView(0);
153 }
154
155 void TestShell::createDRTDevToolsClient(DRTDevToolsAgent* agent)
156 {
157     m_drtDevToolsClient = adoptPtr(new DRTDevToolsClient(agent, m_devTools->webView()));
158 }
159
160 void TestShell::showDevTools()
161 {
162     if (!m_devTools) {
163         WebURL url = webkit_support::GetDevToolsPathAsURL();
164         if (!url.isValid()) {
165             ASSERT(false);
166             return;
167         }
168         m_devTools = createNewWindow(url);
169         ASSERT(m_devTools);
170         createDRTDevToolsClient(m_drtDevToolsAgent.get());
171     }
172     m_devTools->show(WebKit::WebNavigationPolicyNewWindow);
173 }
174
175 void TestShell::closeDevTools()
176 {
177     if (m_devTools) {
178         m_drtDevToolsAgent->reset();
179         m_drtDevToolsClient.clear();
180         closeWindow(m_devTools);
181         m_devTools = 0;
182     }
183 }
184
185 void TestShell::resetWebSettings(WebView& webView)
186 {
187     m_prefs.reset();
188     m_prefs.acceleratedCompositingEnabled = m_acceleratedCompositingEnabled;
189     m_prefs.compositeToTexture = m_compositeToTexture;
190     m_prefs.forceCompositingMode = m_forceCompositingMode;
191     m_prefs.accelerated2dCanvasEnabled = m_accelerated2dCanvasEnabled;
192     m_prefs.legacyAccelerated2dCanvasEnabled = m_legacyAccelerated2dCanvasEnabled;
193     m_prefs.acceleratedDrawingEnabled = m_acceleratedDrawingEnabled;
194     m_prefs.applyTo(&webView);
195 }
196
197 void TestShell::runFileTest(const TestParams& params)
198 {
199     ASSERT(params.testUrl.isValid());
200     m_testIsPreparing = true;
201     m_params = params;
202     string testUrl = m_params.testUrl.spec();
203
204     if (testUrl.find("loading/") != string::npos
205         || testUrl.find("loading\\") != string::npos)
206         m_layoutTestController->setShouldDumpFrameLoadCallbacks(true);
207
208     if (testUrl.find("/dumpAsText/") != string::npos
209         || testUrl.find("\\dumpAsText\\") != string::npos) {
210         m_layoutTestController->setShouldDumpAsText(true);
211         m_layoutTestController->setShouldGeneratePixelResults(false);
212     }
213
214     if (testUrl.find("/inspector/") != string::npos
215         || testUrl.find("\\inspector\\") != string::npos)
216         showDevTools();
217
218     if (m_params.debugLayerTree)
219         m_layoutTestController->setShowDebugLayerTree(true);
220
221     if (m_dumpWhenFinished)
222         m_printer->handleTestHeader(testUrl.c_str());
223     loadURL(m_params.testUrl);
224
225     m_testIsPreparing = false;
226     waitTestFinished();
227 }
228
229 static inline bool isSVGTestURL(const WebURL& url)
230 {
231     return url.isValid() && string(url.spec()).find("W3C-SVG-1.1") != string::npos;
232 }
233
234 void TestShell::resizeWindowForTest(WebViewHost* window, const WebURL& url)
235 {
236     int width, height;
237     if (isSVGTestURL(url)) {
238         width = SVGTestWindowWidth;
239         height = SVGTestWindowHeight;
240     } else {
241         width = testWindowWidth;
242         height = testWindowHeight;
243     }
244     window->setWindowRect(WebRect(0, 0, width + virtualWindowBorder * 2, height + virtualWindowBorder * 2));
245 }
246
247 void TestShell::resetTestController()
248 {
249     resetWebSettings(*webView());
250     m_webPermissions->reset();
251     m_accessibilityController->reset();
252     m_layoutTestController->reset();
253     m_eventSender->reset();
254     m_webViewHost->reset();
255     m_notificationPresenter->reset();
256     m_drtDevToolsAgent->reset();
257     if (m_drtDevToolsClient)
258         m_drtDevToolsClient->reset();
259     webView()->mainFrame()->clearOpener();
260 }
261
262 void TestShell::loadURL(const WebURL& url)
263 {
264     m_webViewHost->loadURLForFrame(url, WebString());
265 }
266
267 void TestShell::reload()
268 {
269     m_webViewHost->navigationController()->reload();
270 }
271
272 void TestShell::goToOffset(int offset)
273 {
274      m_webViewHost->navigationController()->goToOffset(offset);
275 }
276
277 int TestShell::navigationEntryCount() const
278 {
279     return m_webViewHost->navigationController()->entryCount();
280 }
281
282 void TestShell::callJSGC()
283 {
284     m_webView->mainFrame()->collectGarbage();
285 }
286
287 void TestShell::setFocus(WebWidget* widget, bool enable)
288 {
289     // Simulate the effects of InteractiveSetFocus(), which includes calling
290     // both setFocus() and setIsActive().
291     if (enable) {
292         if (m_focusedWidget != widget) {
293             if (m_focusedWidget)
294                 m_focusedWidget->setFocus(false);
295             webView()->setIsActive(enable);
296             widget->setFocus(enable);
297             m_focusedWidget = widget;
298         }
299     } else {
300         if (m_focusedWidget == widget) {
301             widget->setFocus(enable);
302             webView()->setIsActive(enable);
303             m_focusedWidget = 0;
304         }
305     }
306 }
307
308 void TestShell::testFinished()
309 {
310     if (!m_testIsPending)
311         return;
312     m_testIsPending = false;
313     if (m_dumpWhenFinished)
314         dump();
315     webkit_support::QuitMessageLoop();
316 }
317
318 void TestShell::testTimedOut()
319 {
320     m_printer->handleTimedOut();
321     testFinished();
322 }
323
324 static string dumpDocumentText(WebFrame* frame)
325 {
326     // We use the document element's text instead of the body text here because
327     // not all documents have a body, such as XML documents.
328     WebElement documentElement = frame->document().documentElement();
329     if (documentElement.isNull())
330         return string();
331     return documentElement.innerText().utf8();
332 }
333
334 static string dumpFramesAsText(WebFrame* frame, bool recursive)
335 {
336     string result;
337
338     // Add header for all but the main frame. Skip empty frames.
339     if (frame->parent() && !frame->document().documentElement().isNull()) {
340         result.append("\n--------\nFrame: '");
341         result.append(frame->name().utf8().data());
342         result.append("'\n--------\n");
343     }
344
345     result.append(dumpDocumentText(frame));
346     result.append("\n");
347
348     if (recursive) {
349         for (WebFrame* child = frame->firstChild(); child; child = child->nextSibling())
350             result.append(dumpFramesAsText(child, recursive));
351     }
352
353     return result;
354 }
355
356 static void dumpFrameScrollPosition(WebFrame* frame, bool recursive)
357 {
358     WebSize offset = frame->scrollOffset();
359     if (offset.width > 0 || offset.height > 0) {
360         if (frame->parent())
361             printf("frame '%s' ", frame->name().utf8().data());
362         printf("scrolled to %d,%d\n", offset.width, offset.height);
363     }
364
365     if (!recursive)
366         return;
367     for (WebFrame* child = frame->firstChild(); child; child = child->nextSibling())
368         dumpFrameScrollPosition(child, recursive);
369 }
370
371 struct ToLower {
372     char16 operator()(char16 c) { return tolower(c); }
373 };
374
375 // FIXME: Eliminate std::transform(), std::vector, and std::sort().
376
377 // Returns True if item1 < item2.
378 static bool HistoryItemCompareLess(const WebHistoryItem& item1, const WebHistoryItem& item2)
379 {
380     string16 target1 = item1.target();
381     string16 target2 = item2.target();
382     std::transform(target1.begin(), target1.end(), target1.begin(), ToLower());
383     std::transform(target2.begin(), target2.end(), target2.begin(), ToLower());
384     return target1 < target2;
385 }
386
387 static string dumpHistoryItem(const WebHistoryItem& item, int indent, bool isCurrent)
388 {
389     string result;
390
391     if (isCurrent) {
392         result.append("curr->");
393         result.append(indent - 6, ' '); // 6 == "curr->".length()
394     } else
395         result.append(indent, ' ');
396
397     string url = item.urlString().utf8();
398     size_t pos;
399     if (!url.find(fileUrlPattern) && ((pos = url.find(layoutTestsPattern)) != string::npos)) {
400         // adjust file URLs to match upstream results.
401         url.replace(0, pos + layoutTestsPatternSize, fileTestPrefix);
402     } else if (!url.find(dataUrlPattern)) {
403         // URL-escape data URLs to match results upstream.
404         string path = webkit_support::EscapePath(url.substr(dataUrlPatternSize));
405         url.replace(dataUrlPatternSize, url.length(), path);
406     }
407
408     result.append(url);
409     if (!item.target().isEmpty()) {
410         result.append(" (in frame \"");
411         result.append(item.target().utf8());
412         result.append("\")");
413     }
414     if (item.isTargetItem())
415         result.append("  **nav target**");
416     result.append("\n");
417
418     const WebVector<WebHistoryItem>& children = item.children();
419     if (!children.isEmpty()) {
420         // Must sort to eliminate arbitrary result ordering which defeats
421         // reproducible testing.
422         // FIXME: WebVector should probably just be a std::vector!!
423         std::vector<WebHistoryItem> sortedChildren;
424         for (size_t i = 0; i < children.size(); ++i)
425             sortedChildren.push_back(children[i]);
426         std::sort(sortedChildren.begin(), sortedChildren.end(), HistoryItemCompareLess);
427         for (size_t i = 0; i < sortedChildren.size(); ++i)
428             result += dumpHistoryItem(sortedChildren[i], indent + 4, false);
429     }
430
431     return result;
432 }
433
434 static void dumpBackForwardList(const TestNavigationController& navigationController, string& result)
435 {
436     result.append("\n============== Back Forward List ==============\n");
437     for (int index = 0; index < navigationController.entryCount(); ++index) {
438         int currentIndex = navigationController.lastCommittedEntryIndex();
439         WebHistoryItem historyItem = navigationController.entryAtIndex(index)->contentState();
440         if (historyItem.isNull()) {
441             historyItem.initialize();
442             historyItem.setURLString(navigationController.entryAtIndex(index)->URL().spec().utf16());
443         }
444         result.append(dumpHistoryItem(historyItem, 8, index == currentIndex));
445     }
446     result.append("===============================================\n");
447 }
448
449 string TestShell::dumpAllBackForwardLists()
450 {
451     string result;
452     for (unsigned i = 0; i < m_windowList.size(); ++i)
453         dumpBackForwardList(*m_windowList[i]->navigationController(), result);
454     return result;
455 }
456
457 void TestShell::dump()
458 {
459     WebScriptController::flushConsoleMessages();
460
461     // Dump the requested representation.
462     WebFrame* frame = m_webView->mainFrame();
463     if (!frame)
464         return;
465     bool shouldDumpAsText = m_layoutTestController->shouldDumpAsText();
466     bool shouldDumpAsAudio = m_layoutTestController->shouldDumpAsAudio();
467     bool shouldGeneratePixelResults = m_layoutTestController->shouldGeneratePixelResults();
468     bool dumpedAnything = false;
469
470     if (shouldDumpAsAudio) {
471         m_printer->handleAudioHeader();
472
473         const string& encodedAudioData = m_layoutTestController->encodedAudioData();
474         if (fwrite(encodedAudioData.c_str(), 1, encodedAudioData.size(), stdout) != encodedAudioData.size())
475             FATAL("Short write to stdout, disk full?\n");
476         printf("\n");
477
478         m_printer->handleTestFooter(true);
479         fflush(stdout);
480         fflush(stderr);
481         return;
482     }
483
484     if (m_params.dumpTree) {
485         dumpedAnything = true;
486         m_printer->handleTextHeader();
487         // Text output: the test page can request different types of output
488         // which we handle here.
489         if (!shouldDumpAsText) {
490             // Plain text pages should be dumped as text
491             string mimeType = frame->dataSource()->response().mimeType().utf8();
492             if (mimeType == "text/plain") {
493                 shouldDumpAsText = true;
494                 shouldGeneratePixelResults = false;
495             }
496         }
497         if (shouldDumpAsText) {
498             bool recursive = m_layoutTestController->shouldDumpChildFramesAsText();
499             string dataUtf8 = dumpFramesAsText(frame, recursive);
500             if (fwrite(dataUtf8.c_str(), 1, dataUtf8.size(), stdout) != dataUtf8.size())
501                 FATAL("Short write to stdout, disk full?\n");
502         } else {
503             printf("%s", frame->renderTreeAsText(m_params.debugRenderTree).utf8().data());
504             bool recursive = m_layoutTestController->shouldDumpChildFrameScrollPositions();
505             dumpFrameScrollPosition(frame, recursive);
506         }
507         if (m_layoutTestController->shouldDumpBackForwardList())
508             printf("%s", dumpAllBackForwardLists().c_str());
509     }
510     if (dumpedAnything && m_params.printSeparators)
511         m_printer->handleTextFooter();
512
513     if (m_params.dumpPixels && shouldGeneratePixelResults) {
514         // Image output: we write the image data to the file given on the
515         // command line (for the dump pixels argument), and the MD5 sum to
516         // stdout.
517         dumpedAnything = true;
518         m_webView->layout();
519         if (m_layoutTestController->testRepaint()) {
520             WebSize viewSize = m_webView->size();
521             int width = viewSize.width;
522             int height = viewSize.height;
523             if (m_layoutTestController->sweepHorizontally()) {
524                 for (WebRect column(0, 0, 1, height); column.x < width; column.x++)
525                     m_webViewHost->paintRect(column);
526             } else {
527                 for (WebRect line(0, 0, width, 1); line.y < height; line.y++)
528                     m_webViewHost->paintRect(line);
529             }
530         } else
531             m_webViewHost->paintInvalidatedRegion();
532
533         // See if we need to draw the selection bounds rect. Selection bounds
534         // rect is the rect enclosing the (possibly transformed) selection.
535         // The rect should be drawn after everything is laid out and painted.
536         if (m_layoutTestController->shouldDumpSelectionRect()) {
537             // If there is a selection rect - draw a red 1px border enclosing rect
538             WebRect wr = frame->selectionBoundsRect();
539             if (!wr.isEmpty()) {
540                 // Render a red rectangle bounding selection rect
541                 SkPaint paint;
542                 paint.setColor(0xFFFF0000); // Fully opaque red
543                 paint.setStyle(SkPaint::kStroke_Style);
544                 paint.setFlags(SkPaint::kAntiAlias_Flag);
545                 paint.setStrokeWidth(1.0f);
546                 SkIRect rect; // Bounding rect
547                 rect.set(wr.x, wr.y, wr.x + wr.width, wr.y + wr.height);
548                 m_webViewHost->canvas()->drawIRect(rect, paint);
549             }
550         }
551
552         dumpImage(m_webViewHost->canvas());
553     }
554     m_printer->handleImageFooter();
555     m_printer->handleTestFooter(dumpedAnything);
556     fflush(stdout);
557     fflush(stderr);
558 }
559
560 void TestShell::dumpImage(SkCanvas* canvas) const
561 {
562     // Fix the alpha. The expected PNGs on Mac have an alpha channel, so we want
563     // to keep it. On Windows, the alpha channel is wrong since text/form control
564     // drawing may have erased it in a few places. So on Windows we force it to
565     // opaque and also don't write the alpha channel for the reference. Linux
566     // doesn't have the wrong alpha like Windows, but we match Windows.
567 #if OS(MAC_OS_X)
568     bool discardTransparency = false;
569 #else
570     bool discardTransparency = true;
571     makeCanvasOpaque(canvas);
572 #endif
573
574     const SkBitmap& sourceBitmap = canvas->getTopDevice()->accessBitmap(false);
575     SkAutoLockPixels sourceBitmapLock(sourceBitmap);
576
577     // Compute MD5 sum.
578     MD5 digester;
579     Vector<uint8_t, 16> digestValue;
580     digester.addBytes(reinterpret_cast<const uint8_t*>(sourceBitmap.getPixels()), sourceBitmap.getSize());
581     digester.checksum(digestValue);
582     string md5hash;
583     md5hash.reserve(16 * 2);
584     for (unsigned i = 0; i < 16; ++i) {
585         char hex[3];
586         // Use "x", not "X". The string must be lowercased.
587         sprintf(hex, "%02x", digestValue[i]);
588         md5hash.append(hex);
589     }
590
591     // Only encode and dump the png if the hashes don't match. Encoding the
592     // image is really expensive.
593     if (md5hash.compare(m_params.pixelHash)) {
594         std::vector<unsigned char> png;
595         webkit_support::EncodeBGRAPNGWithChecksum(reinterpret_cast<const unsigned char*>(sourceBitmap.getPixels()), sourceBitmap.width(),
596             sourceBitmap.height(), static_cast<int>(sourceBitmap.rowBytes()), discardTransparency, md5hash, &png);
597
598         m_printer->handleImage(md5hash.c_str(), m_params.pixelHash.c_str(), &png[0], png.size(), m_params.pixelFileName.c_str());
599     } else
600         m_printer->handleImage(md5hash.c_str(), m_params.pixelHash.c_str(), 0, 0, m_params.pixelFileName.c_str());
601 }
602
603 void TestShell::bindJSObjectsToWindow(WebFrame* frame)
604 {
605     WebTestingSupport::injectInternalsObject(frame);
606     m_accessibilityController->bindToJavascript(frame, WebString::fromUTF8("accessibilityController"));
607     m_layoutTestController->bindToJavascript(frame, WebString::fromUTF8("layoutTestController"));
608     m_eventSender->bindToJavascript(frame, WebString::fromUTF8("eventSender"));
609     m_plainTextController->bindToJavascript(frame, WebString::fromUTF8("plainText"));
610     m_textInputController->bindToJavascript(frame, WebString::fromUTF8("textInputController"));
611 }
612
613 WebViewHost* TestShell::createNewWindow(const WebKit::WebURL& url)
614 {
615     return createNewWindow(url, 0);
616 }
617
618 WebViewHost* TestShell::createNewWindow(const WebKit::WebURL& url, DRTDevToolsAgent* devToolsAgent)
619 {
620     WebViewHost* host = new WebViewHost(this);
621     WebView* view = WebView::create(host);
622     view->setPermissionClient(webPermissions());
623     view->setDevToolsAgentClient(devToolsAgent);
624     host->setWebWidget(view);
625     m_prefs.applyTo(view);
626     view->initializeMainFrame(host);
627     m_windowList.append(host);
628     host->loadURLForFrame(url, WebString());
629     return host;
630 }
631
632 void TestShell::closeWindow(WebViewHost* window)
633 {
634     size_t i = m_windowList.find(window);
635     if (i == notFound) {
636         ASSERT_NOT_REACHED();
637         return;
638     }
639     m_windowList.remove(i);
640     WebWidget* focusedWidget = m_focusedWidget;
641     if (window->webWidget() == m_focusedWidget)
642         focusedWidget = 0;
643
644     delete window;
645     // We set the focused widget after deleting the web view host because it
646     // can change the focus.
647     m_focusedWidget = focusedWidget;
648     if (m_focusedWidget) {
649         webView()->setIsActive(true);
650         m_focusedWidget->setFocus(true);
651     }
652 }
653
654 void TestShell::closeRemainingWindows()
655 {
656     // Just close devTools window manually because we have custom deinitialization code for it.
657     closeDevTools();
658
659     // Iterate through the window list and close everything except the main
660     // window. We don't want to delete elements as we're iterating, so we copy
661     // to a temp vector first.
662     Vector<WebViewHost*> windowsToDelete;
663     for (unsigned i = 0; i < m_windowList.size(); ++i) {
664         if (m_windowList[i] != webViewHost())
665             windowsToDelete.append(m_windowList[i]);
666     }
667     ASSERT(windowsToDelete.size() + 1 == m_windowList.size());
668     for (unsigned i = 0; i < windowsToDelete.size(); ++i)
669         closeWindow(windowsToDelete[i]);
670     ASSERT(m_windowList.size() == 1);
671 }
672
673 int TestShell::windowCount()
674 {
675     return m_windowList.size();
676 }