domreactor-redglass 0.0.6 → 0.0.7

data/domreactor-redglass.gemspec

@@ -6,7 +6,7 @@ require 'domreactor-redglass/version'
 Gem::Specification.new do |s|
   s.name          = 'domreactor-redglass'
   s.version       = DomReactorRedGlass::VERSION
-  s.date          = '2013-11-14'
+  s.date          = '2014-04-26'
   s.summary       = 'DomReactor plugin for RedGlass.'
   s.description   = 'Send RedGlass page archives to DomReactor for automated layout analysis.'
   s.authors       = ["Frank O'Hara", "Chris Lamb"]

data/lib/domreactor-redglass.rb

@@ -3,6 +3,7 @@ require 'domreactor-redglass/archive'
 require 'domreactor-redglass/archive_location'
 require 'domreactor-redglass/config'
 require 'domreactor-redglass/version'
+require 'domreactor-redglass/report_poller'
 
 module DomReactorRedGlass
 
@@ -21,6 +22,12 @@ module DomReactorRedGlass
     @chain_reaction.post_archives(archive_location)
   end
 
+  def poll_report(opts={})
+    poller = ReportPoller.new(@chain_reaction, opts)
+    poller.poll_completion
+    poller.report
+  end
+
   def parse_json_file(path)
     json_str = File.open(path, 'rb') {|f| f.read}
     JSON.parse(json_str, symbolize_names: true)

data/lib/domreactor-redglass/chain_reaction.rb

@@ -23,7 +23,7 @@ module DomReactorRedGlass
     end
 
     def id
-      id ||= info[:id]
+      @id ||= info[:id]
     end
 
     def info

data/lib/domreactor-redglass/report_poller.rb

@@ -0,0 +1,55 @@
+require 'timeout'
+require 'rest_client'
+
+class ReportPoller
+  attr_reader :time_limit
+
+  def initialize(chain_reaction, opts={})
+    @chain_reaction = chain_reaction
+    @time_limit = opts[:time_limit] || 60
+  end
+
+  def params
+    @params ||= build_params(@chain_reaction)
+  end
+
+  def poll_completion
+    Timeout::timeout(@time_limit) {
+      percent_complete = @chain_reaction.info[:percent_complete]
+      until percent_complete == 100
+        percent_complete = fetch_percent_complete
+        sleep 1 unless percent_complete == 100
+      end
+      percent_complete
+    }
+  end
+
+  def report
+    fetch_report
+  end
+
+  private
+
+  def build_params(chain_reaction)
+    {
+        auth_token: chain_reaction.auth_token,
+        chain_reaction_id: chain_reaction.id,
+        content_type: 'application/json',
+        accept: :json
+    }
+  end
+
+  def fetch_percent_complete
+    fetch_chain_reaction[:percent_complete]
+  end
+
+  def fetch_chain_reaction
+    response = RestClient.get(@chain_reaction.chain_reaction_url, { params: params })
+    JSON.parse(response, symbolize_names: true)[:chain_reaction]
+  end
+
+  def fetch_report
+    response = RestClient.get("#{@chain_reaction.chain_reaction_url}/reports", { params: params })
+    JSON.parse(response, symbolize_names: true)[:reports].first
+  end
+end

data/lib/domreactor-redglass/version.rb

@@ -1,3 +1,3 @@
 module DomReactorRedGlass
-  VERSION = "0.0.6"
+  VERSION = "0.0.7"
 end

data/spec/report_poller_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+require 'timeout'
+
+describe ReportPoller do
+  let(:chain_reaction) { DomReactorRedGlass::ChainReaction.new({}) }
+  before do
+    RestClient.stub(:post) { {chain_reaction: {id: 42, some_info: 'yay'}}.to_json }
+    ChainReaction.any_instance.stub(:baseline_browser) { {id: 1} }
+  end
+
+  it 'time limit defaults to 60 seconds' do
+    poller = ReportPoller.new(chain_reaction)
+    poller.time_limit.should eq 60
+  end
+
+  it 'times out' do
+    poller = ReportPoller.new(chain_reaction, { time_limit: 2 })
+    chain_reaction.should_receive(:info).and_return({ percent_complete: 0 })
+    poller.should_receive(:fetch_percent_complete).at_least(1).times.and_return(1)
+    expect{ poller.poll_completion }.to raise_error(TimeoutError)
+  end
+
+  it 'polls until completion' do
+    poller = ReportPoller.new(chain_reaction, { time_limit: 2 })
+    chain_reaction.should_receive(:info).and_return({ percent_complete: 0 })
+    poller.should_receive(:fetch_percent_complete).and_return(100)
+    poller.poll_completion.should eq 100
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: domreactor-redglass
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
   prerelease: 
 platform: ruby
 authors:
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-11-14 00:00:00.000000000 Z
+date: 2014-04-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rest-client
@@ -161,6 +161,7 @@ files:
 - lib/domreactor-redglass/archive_location.rb
 - lib/domreactor-redglass/chain_reaction.rb
 - lib/domreactor-redglass/config.rb
+- lib/domreactor-redglass/report_poller.rb
 - lib/domreactor-redglass/version.rb
 - spec/archive_location_spec.rb
 - spec/archive_spec.rb
@@ -178,6 +179,7 @@ files:
 - spec/data/valid_archive/firefox_20.0_1369614644/screenshot.png
 - spec/data/valid_archive/firefox_20.0_1369614644/source.html
 - spec/domreactor_spec.rb
+- spec/report_poller_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/bimech/domreactor-redglass
 licenses:
@@ -221,4 +223,5 @@ test_files:
 - spec/data/valid_archive/firefox_20.0_1369614644/screenshot.png
 - spec/data/valid_archive/firefox_20.0_1369614644/source.html
 - spec/domreactor_spec.rb
+- spec/report_poller_spec.rb
 - spec/spec_helper.rb