Adds Highest and Current Team League

Author: cadwallion

lib/bnet_scraper/starcraft2/profile_scraper.rb

@@ -22,7 +22,8 @@ module Starcraft2
     #   }
     class ProfileScraper < BaseScraper
       attr_reader :achievement_points, :career_games, :race, :leagues, :most_played,
-        :games_this_season, :highest_solo_league, :current_solo_league
+        :games_this_season, :highest_solo_league, :current_solo_league, :highest_team_league,
+        :current_team_league
 
       def initialize options = {}
         super
@@ -49,12 +50,20 @@ def get_profile_data
           @most_played = html.css(".stat-block:nth-child(2) h2").inner_html()
           @games_this_season = html.css(".stat-block:nth-child(1) h2").inner_html()
 
-          if solo_league_finishes = html.css("#best-finish-SOLO div")[0]
+          if html.css("#best-finish-SOLO div")[0]
             @highest_solo_league = html.css("#best-finish-SOLO div")[0].children[2].inner_text.strip
             @current_solo_league = html.css("#best-finish-SOLO div")[0].children[8].inner_text.strip
           else
-            @highest_solo_league = "Unranked"
-            @current_solo_league = "Unranked"
+            @highest_solo_league = "Not Yet Ranked"
+            @current_solo_league = "Not Yet Ranked"
+          end
+
+          if html.css("#best-finish-TEAM div")[0] 
+            @highest_team_league = html.css("#best-finish-TEAM div")[0].children[2].inner_text.strip
+            @current_team_league = html.css("#best-finish-TEAM div")[0].children[8].inner_text.strip
+          else
+            @highest_team_league = "Not Yet Ranked"
+            @current_team_league = "Not Yet Ranked"
           end
 
         else
@@ -88,6 +97,8 @@ def output
           race: @race,
           current_solo_league: @current_solo_league,
           highest_solo_league: @highest_solo_league,
+          current_team_league: @current_team_league,
+          highest_team_league: @highest_team_league,
           career_games: @career_games,
           games_this_season: @games_this_season,
           most_played: @most_played,

spec/starcraft2/profile_scraper_spec.rb

@@ -9,23 +9,27 @@
   subject { BnetScraper::Starcraft2::ProfileScraper.new(bnet_id: '2377239', account: 'Demon') }
 
   describe '#get_profile_data' do
-    it 'should set the race, wins, and achievements attributes' do
-      subject.race.should be_nil
-      subject.achievement_points.should be_nil
-
+    before do
       subject.get_profile_data
-      
-      subject.race.should == 'Protoss'
-      subject.achievement_points.should == '3660'
     end
+
+    its(:race) { should == 'Protoss' }
+    its(:achievement_points) { should == '3660' }
+    its(:current_solo_league) { should == 'Not Yet Ranked' }
+    its(:highest_solo_league) { should == 'Platinum' }
+    its(:current_team_league) { should == 'Not Yet Ranked' }
+    its(:highest_team_league) { should == 'Diamond' }
+    its(:career_games) { should == '1568' }
+    its(:games_this_season) { should == '0' }
+    its(:most_played) { should == '4v4' }
   end
 
   describe 'get_league_list' do
     it 'should set an array of leagues' do
       subject.should have(0).leagues
       subject.get_league_list
 
-      subject.instance_variable_get(:@leagues).should have(12).leagues
+      subject.should have(12).leagues
     end
   end
 
@@ -75,6 +79,8 @@
         games_this_season: '0',
         highest_solo_league: 'Platinum',
         current_solo_league: 'Not Yet Ranked',
+        highest_team_league: 'Diamond',
+        current_team_league: 'Not Yet Ranked',
         most_played: '4v4',
         achievement_points: '3660',
         leagues: [
@@ -151,6 +157,8 @@
         most_played: nil,
         highest_solo_league: nil, 
         current_solo_league: nil, 
+        highest_team_league: nil,
+        current_team_league: nil,
         achievement_points: nil, 
         leagues: [] 
       }

spec/starcraft2_spec.rb

@@ -13,6 +13,8 @@
         :most_played => '4v4',
         :current_solo_league => 'Not Yet Ranked',
         :highest_solo_league => 'Platinum',
+        :current_team_league => 'Not Yet Ranked',
+        :highest_team_league => 'Diamond',
         :achievement_points=>"3660", 
         :leagues=>[
           {:season=>"6", :size=>"4v4", :name=>"Aleksander Pepper", :division=>"Diamond", :random=>false, :bnet_id=>"2377239", :account=>"Demon"},