masque 0.0.2 → 0.1.0

data/lib/masque.rb

@@ -25,10 +25,6 @@ class Masque
         @session = session
       end
 
-      def page
-        session
-      end
-
       def method_missing(*args)
         session.__send__(*args)
       end

data/lib/masque/dsl.rb

@@ -6,5 +6,53 @@ class Masque
       page.driver.render(path)
     end
     alias :render :save_screenshot
+
+    def page
+      session
+    end
+
+    def driver
+      session.driver
+    end
+
+    def driver_name
+      case driver
+      when ::Capybara::Driver::Webkit
+        :webkit
+      when ::Capybara::Poltergeist::Driver
+        :poltergeist
+      else
+        :unknown
+      end
+    end
+
+    def set_headers(headers = {})
+      case driver_name
+      when :webkit
+        headers.each_pair {|k,v| driver.header(k, v)}
+      when :poltergeist
+        driver.headers = headers
+      else
+        raise "for unknown driver"
+      end
+    end
+    alias :set_request_headers :set_headers
+
+    def cookies
+      driver.cookies # driver specific format
+    end
+
+    def resize(x, y)
+      case driver_name
+      when :webkit
+        driver.resize_window(x, y)
+      when :poltergeist
+        driver.resize(x, y)
+      else
+        raise "for unknown driver"
+      end
+    end
+    alias :resize_window :resize
+
   end
 end

data/lib/masque/version.rb

@@ -1,3 +1,3 @@
 class Masque
-  VERSION = "0.0.2"
+  VERSION = "0.1.0"
 end

data/masque.gemspec

@@ -22,4 +22,5 @@ Gem::Specification.new do |gem|
   gem.add_development_dependency "rspec", "~> 2.11"
   gem.add_development_dependency "sinatra"
   gem.add_development_dependency "simplecov"
+  gem.add_development_dependency "multi_json"
 end

data/spec/masque_spec.rb

@@ -11,11 +11,6 @@ describe Masque do
         m.agent.session = Capybara::Session.new(driver, Dummy)
         m
       end
-      after do
-        masque.instance_eval do
-          @agent = nil
-        end
-      end
 
       it ".new" do
         opts = {:display => 50, :driver => driver}
@@ -47,6 +42,32 @@ describe Masque do
           File.unlink(path)
         end
 
+        it "#cookies" do
+          cookie = masque.run do
+            visit "/"
+            cookies["rack.session"]
+          end
+          cookie.should_not be_nil
+        end
+
+        it "#set_headers" do
+          masque.run do
+            set_headers({"User-Agent" => "rspec"})
+            visit "/"
+            find('#ua').text.should == "rspec"
+          end
+        end
+
+        it "#resize" do
+          masque.run do
+            resize(400, 300)
+            visit "/"
+            size = MultiJson.load(find('#js').text)
+            size["w"].should == 400
+            size["h"].should == 300
+          end
+        end
+
         it "return last evaluated value" do
           body = masque.run do
             visit "/"

data/spec/support/app.rb

@@ -18,7 +18,15 @@ class Dummy < Sinatra::Base
     </style>
     <title>hi #{params[:arg]}</title>
     <h1 id="visits">#{session[:visit]}</h1>
+    <div id="ua">#{env["HTTP_USER_AGENT"]}</div>
     <p>Hello, World!</p>
+    <div id="js"></div>
+    <script>
+      document.getElementById("js").innerHTML = JSON.stringify({
+        h: window.innerHeight,
+        w: window.innerWidth
+      });
+    </script>
     HTML
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: masque
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -107,6 +107,22 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: multi_json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: JavaScript enabled web crawler kit
 email:
 - k@uu59.org