Adds BnetScraper::InvalidProfileError to ProfileScraper

Author: cadwallion

lib/bnet_scraper.rb

@@ -3,4 +3,5 @@
 require 'nokogiri'
 
 module BnetScraper
+  class InvalidProfileError < Exception ; end
 end

lib/bnet_scraper/starcraft2/profile_scraper.rb

@@ -38,6 +38,8 @@ def get_profile_data
           @race = html.css("#season-snapshot .module-footer a").first().inner_html()
           @wins = html.css("#career-stats h2").inner_html()
           @achievement_points = html.css("#profile-header h3").inner_html()
+        else
+          raise BnetScraper::InvalidProfileError
         end
       end
 
@@ -54,6 +56,8 @@ def get_league_list
               href: "#{profile_url}ladder/#{league.attr('href')}"
             }
           end
+        else
+          raise BnetScraper::InvalidProfileError
         end
       end
 

spec/starcraft2/profile_scraper_spec.rb

@@ -45,6 +45,11 @@
       subject.should_receive(:output)
       subject.scrape
     end
+
+    it 'should return InvalidProfileError if response is 404' do
+      scraper = BnetScraper::Starcraft2::ProfileScraper.new url: 'http://us.battle.net/sc2/en/profile/2377239/1/SomeDude/'
+      expect { scraper.scrape }.to raise_error(BnetScraper::InvalidProfileError)
+    end
   end
 
   describe '#output' do