thoughtbot · niklasb · Sep 24, 2011 · Sep 24, 2011 · Oct 6, 2011 · niklasb
diff --git a/lib/capybara/driver/webkit/browser.rb b/lib/capybara/driver/webkit/browser.rb
@@ -86,6 +86,14 @@ def render(path, width, height)
       command "Render", path, width, height
     end
 
+    def set_html(html, url=nil)
+      if url
+        command("SetHtml", html, url)
+      else
+        command("SetHtml", html)
+      end
+    end
+
     private
 
     def start_server

diff --git a/spec/driver_spec.rb b/spec/driver_spec.rb
@@ -884,6 +884,77 @@ def make_the_server_go_away
     end
   end
 
+  context "custom HTML" do
+    before(:all) do
+      @requested = []
+      @app = lambda do |env|
+        params = ::Rack::Utils.parse_query(env['QUERY_STRING'])
+
+        type = params["get"]
+        @requested << type
+        if type == "javascript"
+          js = <<-JS
+            onload_handler = function() {
+              document.getElementById("url").innerHTML = document.location;
+            }
+          JS
+          [200,
+            { 'Content-Type'   => 'text/javascript; charset=UTF-8',
+              'Content-Length' => js.size.to_s,
+            }, [js]]
+        else
+          [200,
+            { 'Content-Type'   => 'text/%s; charset=UTF-8' % params["get"],
+              'Content-Length' => "0",
+            }, [""]]
+        end
+      end
+    end
+
+    before do
+      @requested.clear
+      @root_url = subject.send(:url, "/")
+    end
+
+    def set_html(url = @root_url)
+      subject.browser.set_html <<-HTML, url
+        <html>
+          <head>
+            <title>Test</title>
+            <script type="text/javascript" src="?get=javascript"></script>
+            <link rel="stylesheet" type="text/css" href="?get=css" />
+          </head>
+          <body onload="onload_handler();">
+            <p id="welcome">Hello</p>
+            <p id="url">n/a</p>
+          </body>
+        </html>
+      HTML
+      subject.find("id('welcome')").first.text.should == "Hello"
+    end
+
+    it "allows setting custom HTML without URL" do
+      set_html nil
+    end
+
+    it "allows setting custom HTML with URL" do
+      set_html
+      subject.browser.url.should == @root_url
+    end
+
+    it "loads resources from right location" do
+      set_html
+      @requested.size.should == 2
+      @requested.should include "javascript"
+      @requested.should include "css"
+    end
+
+    it "makes fake URL accessible through document.location" do
+      set_html
+      subject.find("id('url')").first.text.should == @root_url
+    end
+  end
+
   context "with socket debugger" do
     let(:socket_debugger_class){ Capybara::Driver::Webkit::SocketDebugger }
     let(:browser_with_debugger){

diff --git a/src/Connection.cpp b/src/Connection.cpp
@@ -16,6 +16,7 @@
 #include "Body.h"
 #include "Status.h"
 #include "Headers.h"
+#include "SetHtml.h"
 
 #include <QTcpSocket>
 #include <iostream>

diff --git a/src/SetHtml.cpp b/src/SetHtml.cpp
@@ -0,0 +1,25 @@
+#include "SetHtml.h"
+#include "WebPage.h"
+#include <QUrl>
+
+SetHtml::SetHtml(WebPage *page, QObject *parent)
+  : Command(page, parent)
+{ }
+
+void SetHtml::start(QStringList &arguments) {
+  if (arguments.size() > 1)
+    page()->currentFrame()->setHtml(arguments[0], QUrl(arguments[1]));
+  else
+    page()->currentFrame()->setHtml(arguments[0]);
+  emit finished(new Response(true));
+}
+
+void SetHtml::loadFinished(bool success) {
+  QString message;
+  if (!success)
+    message = page()->failureString();
+
+  disconnect(page(), SIGNAL(pageFinished(bool)), this,
+             SLOT(loadFinished(bool)));
+  emit finished(new Response(success, message));
+}
diff --git a/src/SetHtml.h b/src/SetHtml.h
@@ -0,0 +1,14 @@
+#include "Command.h"
+
+class WebPage;
+
+class SetHtml : public Command {
+  Q_OBJECT;
+
+ public:
+  SetHtml(WebPage *page, QObject *parent = 0);
+  virtual void start(QStringList &arguments);
+
+ private slots:
+  void loadFinished(bool success);
+};
diff --git a/src/find_command.h b/src/find_command.h
@@ -16,4 +16,5 @@ CHECK_COMMAND(Header)
 CHECK_COMMAND(Render)
 CHECK_COMMAND(Body)
 CHECK_COMMAND(Status)
-CHECK_COMMAND(Headers)
+CHECK_COMMAND(Headers)
+CHECK_COMMAND(SetHtml)
diff --git a/src/webkit_server.pro b/src/webkit_server.pro
@@ -1,8 +1,57 @@
 TEMPLATE = app
 TARGET = webkit_server
 DESTDIR = .
-HEADERS = WebPage.h Server.h Connection.h Command.h Visit.h Find.h Reset.h Node.h JavascriptInvocation.h Url.h Source.h Evaluate.h Execute.h FrameFocus.h Response.h NetworkAccessManager.h Header.h Render.h body.h Status.h Headers.h UnsupportedContentHandler.h
-SOURCES = main.cpp WebPage.cpp Server.cpp Connection.cpp Command.cpp Visit.cpp Find.cpp Reset.cpp Node.cpp JavascriptInvocation.cpp Url.cpp Source.cpp Evaluate.cpp Execute.cpp FrameFocus.cpp Response.cpp NetworkAccessManager.cpp Header.cpp Render.cpp body.cpp Status.cpp Headers.cpp UnsupportedContentHandler.cpp
+HEADERS = \
+  WebPage.h \
+  Server.h \
+  Connection.h \
+  Command.h \
+  Visit.h \
+  Find.h \
+  Reset.h \
+  Node.h \
+  JavascriptInvocation.h \
+  Url.h \
+  Source.h \
+  Evaluate.h \
+  Execute.h \
+  FrameFocus.h \
+  Response.h \
+  NetworkAccessManager.h \
+  Header.h \
+  Render.h \
+  body.h \
+  Status.h \
+  Headers.h \
+  UnsupportedContentHandler.h \
+  SetHtml.h \
+
+SOURCES = \
+  main.cpp \
+  WebPage.cpp \
+  Server.cpp \
+  Connection.cpp \
+  Command.cpp \
+  Visit.cpp \
+  Find.cpp \
+  Reset.cpp \
+  Node.cpp \
+  JavascriptInvocation.cpp \
+  Url.cpp \
+  Source.cpp \
+  Evaluate.cpp \
+  Execute.cpp \
+  FrameFocus.cpp \
+  Response.cpp \
+  NetworkAccessManager.cpp \
+  Header.cpp \
+  Render.cpp \
+  body.cpp \
+  Status.cpp \
+  Headers.cpp \
+  UnsupportedContentHandler.cpp \
+  SetHtml.cpp \
+
 RESOURCES = webkit_server.qrc
 QT += network webkit
 CONFIG += console