alexjp-blackbook_csv 1.0.5

data/test/test_blackbook_importer_aol.rb

@@ -0,0 +1,113 @@
+require File.join( File.dirname(__FILE__), '../lib/blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+require 'mechanize'
+
+
+class TestBlackbookImporterAol < Test::Unit::TestCase
+
+  include TestHelper
+
+  def setup
+    @importer = Blackbook::Importer::Aol.new
+    @importer.options = {:username => 'user@aol.com', :password => 'password'}
+    @importer.create_agent
+  end
+
+  def test_prepare
+    @importer.expects(:login).once
+    @importer.prepare
+  end
+
+  def test_importer_match
+    assert_equal false, @importer =~ nil
+    assert_equal false, @importer =~ {}
+    assert_equal false, @importer =~ {'username' => 'joe@example.com'}
+    assert_equal false, @importer =~ {:username => 'joe@example.com'}
+    assert_equal false, @importer =~ {:username => 'joe'}
+    assert_equal true, @importer =~ {:username => 'joe@aol.com'}
+    assert_equal true, @importer =~ {:username => 'joe@aim.com'}
+    assert_equal true, @importer =~ {:username => 'JOE@AOL.COM'}
+    assert_equal true, @importer =~ {:username => 'JOE@AIM.COM'}
+  end
+
+  def test_login
+    response = {'content-type' => 'text/html'}
+
+    body = load_fixture('aol_login_response_stage_2.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://webmail.aol.com/').once.returns(page)
+
+    body = load_fixture('aol_login_response_stage_3.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:submit).once.returns(page)
+
+    body = load_fixture('aol_login_response_stage_4.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with(
+      'http://webmail.aol.com/_cqr/LoginSuccess.aspx?sitedomain=sns.webmail.aol.com&authLev=2&siteState=ver%3A2%7Cac%3AWS%7Cat%3ASNS%7Cld%3Awebmail.aol.com%7Cuv%3AAOL%7Clc%3Aen-us&lang=en&locale=us&uitype=std&mcAuth=%2Fblackbookauthtest%3D%3D'
+    ).once.returns(page)
+
+    body = load_fixture('aol_login_response_stage_5.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('/31361/aim/en-us/Lite/Today.aspx').once.returns(page)
+
+    assert_nothing_raised do
+      assert @importer.login
+    end
+  end
+
+  def test_login_bad_credentials
+    response = {'content-type' => 'text/html'}
+
+    body = load_fixture('aol_login_response_stage_2.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://webmail.aol.com/').once.returns(page)
+
+    body = load_fixture('aol_bad_login_response_stage_3.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:submit).once.returns(page)
+
+    assert_raises(Blackbook::BadCredentialsError) do
+      @importer.login
+    end
+  end
+
+  def test_scrape_contacts_raises_badcredentialerror_when_not_logged_in
+    @importer.agent.expects(:cookies).once.returns([])
+    assert_raises(Blackbook::BadCredentialsError) do
+      @importer.scrape_contacts
+    end
+  end
+
+  def test_scrape_contacts
+    cookie = WWW::Mechanize::Cookie.new('Auth',
+      'ver:7&uas:user%2cuser%2caim.com%2c0%2c0%2c0%2cAIM%2cen-us%2c2%2c633308135438559961%2c0%2c1%3b&un:user&at:SNS&sn:user&wim:djEgMVM%253D-Qg0R3W9DSOpFNfKRog8SxBigVbPg%252BDTtGsaBctTczECZruX82XONmIMGYsY%253D&sty:0&ud:aim.com&uid:blackbookuid&ss:633308135438559961&sst:1195234657&la:633308135438559961&lrid:25156&act:M&br:0&mbt:T&uv:AIM&lc:en-us&acd:315532800&pix:0&prmc:135925&relm:im&mah:&sah:&snh:ZDdhMTQ5YjlmZjQ5NTE5ZTFkYWI5OTU0ZDU1NWNlYTM%3d&miu:True&sit:sns.webmail.aol.com&ckd:.webmail.aol.com&ckp:%2f&ha:ABrytqN65h7Czwu0%2bDXlQGuc%2fQY%3d&')
+    cookie.domain = 'localhost'
+    @importer.agent.expects(:cookies).once.returns([cookie])
+
+    response = {'content-type' => 'text/html'}
+    body = load_fixture('aol_application_page.html').join
+    uri = URI.parse('http://webmail.aol.com/34865/aim/en-us/Lite/Today.aspx')
+    page = WWW::Mechanize::Page.new(uri, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:current_page).times(2).returns(page)
+
+    response = {'content-type' => 'text/html'}
+    body = load_fixture('aol_contacts.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with(
+      'http://webmail.aol.com/34865/aim/en-us/Lite/addresslist-print.aspx?command=all&sort=FirstLastNick&sortDir=Ascending&nameFormat=FirstLastNick&user=ULKLrUelQ2'
+    ).once.returns(page)
+
+    assert_nothing_raised do
+      contacts = @importer.scrape_contacts
+      assert_equal 2, contacts.size
+      assert contacts.detect{|c| c[:email] == 'joe.user@example.com'}
+      assert contacts.detect{|c| c[:name] == 'Joe User'}
+      assert contacts.detect{|c| c[:email] == 'jane.user@example.com'}
+      assert contacts.detect{|c| c[:name] == 'Jane User'}
+    end
+  end
+
+end

data/test/test_blackbook_importer_base.rb

@@ -0,0 +1,24 @@
+require File.join( File.dirname(__FILE__), '../lib/blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestBlackbookImporterBase < Test::Unit::TestCase
+
+  include TestHelper
+
+  def test_service_name
+    assert_equal "Base", Blackbook::Importer::Base.new.service_name
+  end
+
+  def test_import
+    importer = Blackbook::Importer::Base.new
+    importer.expects(:fetch_contacts!)
+    options = {:foo => 'bar'}
+    assert_nothing_raised do
+      importer.import(options)
+    end
+    assert_equal options, importer.options
+  end
+
+end

data/test/test_blackbook_importer_csv.rb

@@ -0,0 +1,60 @@
+require File.join( File.dirname(__FILE__), '../lib/blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'tempfile'
+require 'test/unit'
+require 'mocha'
+
+class TestBlackbookImporterCsv < Test::Unit::TestCase
+
+  include TestHelper
+
+  def setup
+    @importer = Blackbook::Importer::Csv.new
+  end
+
+  def test_importer_match
+    csv = Tempfile.new('test.csv')
+    assert_equal false, @importer =~ nil
+    assert_equal false, @importer =~ {}
+    assert_equal false, @importer =~ {:file => nil}
+    assert_equal true, @importer =~ {:file => csv}
+  end
+
+  def test_to_hash
+    cols = [:name, :email, :misc]
+    assert_equal({:name => 'joe', :email => 'joe@example.com', :misc => ''}, 
+      @importer.to_hash(cols,['joe', 'joe@example.com']))
+    assert_equal({:name => 'joe', :email => 'joe@example.com', :misc => 'foo'}, 
+      @importer.to_hash(cols,['joe', 'joe@example.com', 'foo']))
+    assert_equal({:name => 'joe', :email => 'joe@example.com', :misc => 'foo,bar'}, 
+      @importer.to_hash(cols,['joe', 'joe@example.com', 'foo', 'bar']))
+  end
+
+  def test_to_columns
+    assert_equal [:name,:email,:misc], @importer.to_columns('name,email,misc')
+    assert_equal [:name,:email,:misc], @importer.to_columns('Name,email,misc')
+    assert_equal [:name,:email,:misc], @importer.to_columns('Name,Email,misc')
+    assert_equal [:name,:email,:misc], @importer.to_columns('Name,E-mail,misc')
+  end
+
+  def test_fetch_contacts_with_no_column_names
+    file = mock(:path => '/tmp/test.csv')
+    options = {:file => file}
+    @importer.instance_variable_set(:@options, options)
+    IO.expects(:readlines).with('/tmp/test.csv').once.returns(['joe,joe@example.com','fred,fred@example.com'])
+    expected = [{:name => 'joe', :email => 'joe@example.com', :misc => ''},
+                {:name => 'fred', :email => 'fred@example.com', :misc => ''}]
+    assert_equal expected, @importer.fetch_contacts!
+  end
+
+  def test_fetch_contacts_with_column_names
+    file = mock(:path => '/tmp/test.csv')
+    options = {:file => file}
+    @importer.instance_variable_set(:@options, options)
+    IO.expects(:readlines).with('/tmp/test.csv').once.returns(['name,email,misc','joe,joe@example.com','fred,fred@example.com'])
+    expected = [{:name => 'joe', :email => 'joe@example.com', :misc => ''},
+                {:name => 'fred', :email => 'fred@example.com', :misc => ''}]
+    assert_equal expected, @importer.fetch_contacts!
+  end
+
+end

data/test/test_blackbook_importer_gmail.rb

@@ -0,0 +1,117 @@
+require File.join( File.dirname(__FILE__), '../lib/blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestBlackbookImporterGmail < Test::Unit::TestCase
+
+  include TestHelper
+
+  def setup
+    @importer = Blackbook::Importer::Gmail.new
+    @importer.options = {:username => 'user@gmail.com', :password => 'password'}
+    @importer.create_agent
+  end
+
+  def test_prepare
+    @importer.expects(:login).once
+    @importer.prepare
+  end
+
+  def test_importer_match
+    assert_equal false, @importer =~ nil
+    assert_equal false, @importer =~ {}
+    assert_equal false, @importer =~ {'username' => 'joe@example.com'}
+    assert_equal false, @importer =~ {:username => 'joe@example.com'}
+    assert_equal false, @importer =~ {:username => 'joe'}
+    assert_equal true, @importer =~ {:username => 'joe@gmail.com'}
+    assert_equal true, @importer =~ {:username => 'JOE@GMAIL.COM'}
+  end
+
+  def test_bad_login
+    response = {'content-type' => 'text/html'}
+
+    body = load_fixture('gmail_login_response_stage_1.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://mail.google.com/mail/').once.returns(page)
+
+    body = load_fixture('gmail_bad_login_response_stage_2.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:submit).once.returns(page)
+
+    assert_raises(Blackbook::BadCredentialsError) do
+      @importer.login
+    end
+  end
+
+  def test_login
+    response = {'content-type' => 'text/html'}
+
+    body = load_fixture('gmail_login_response_stage_1.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://mail.google.com/mail/').once.returns(page)
+
+    body = load_fixture('gmail_login_response_stage_2.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:submit).once.returns(page)
+
+    body = load_fixture('gmail_login_response_stage_3.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://mail.google.com/mail/?ui=html&zy=l'
+    ).once.returns(page)
+
+    assert_nothing_raised do
+      assert @importer.login
+    end
+  end
+
+  def test_login_alternative_uri
+    response = {'content-type' => 'text/html'}
+
+    body = load_fixture('gmail_login_response_stage_1.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://mail.google.com/mail/').once.returns(page)
+
+    body = load_fixture('gmail_login_response_stage_2a.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:submit).once.returns(page)
+
+    body = load_fixture('gmail_login_response_stage_3.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://mail.google.com/mail/?ui=html&zy=l'
+    ).once.returns(page)
+
+    assert_nothing_raised do
+      assert @importer.login
+    end
+  end
+
+  def test_scrape_contacts_raises_badcredentialerror_when_not_logged_in
+    @importer.agent.expects(:cookies).once.returns([])
+    assert_raises(Blackbook::BadCredentialsError) do
+      @importer.scrape_contacts
+    end
+  end
+
+  def test_scrape_contacts
+    cookie = WWW::Mechanize::Cookie.new('GAUSR', 'mail:user@gmail.com')
+    @importer.agent.expects(:cookies).once.returns([cookie])
+
+    response = {'content-type' => 'text/html'}
+    body = load_fixture('gmail_contacts.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with(
+      'http://mail.google.com/mail/h/?v=cl&pnl=a'
+    ).once.returns(page)
+
+    assert_nothing_raised do
+      contacts = @importer.scrape_contacts
+      assert_equal 2, contacts.size
+      assert contacts.detect{|c| c[:email] == 'joe.user@example.com'}
+      assert contacts.detect{|c| c[:name] == 'Joe User'}
+      assert contacts.detect{|c| c[:email] == 'jane.user@example.com'}
+      assert contacts.detect{|c| c[:name] == 'Jane User'}
+    end
+  end
+
+end

data/test/test_blackbook_importer_hotmail.rb

@@ -0,0 +1,147 @@
+require File.join( File.dirname(__FILE__), '../lib/blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+require 'mechanize'
+
+class TestBlackbookImporterHotmail < Test::Unit::TestCase
+
+  include TestHelper
+
+  
+  def setup
+    @importer = Blackbook::Importer::Hotmail.new
+    @importer.options = {:username => 'user@hotmail.com', :password => 'password'}
+    @importer.create_agent
+  end
+
+  def test_bad_login
+    response = {'content-type' => 'text/html'}
+
+    body = load_fixture('hotmail_login_response_stage_1.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://login.live.com/login.srf?id=2'
+      ).once.returns(page)
+
+    body = load_fixture('hotmail_bad_login_response_stage_2.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:submit).once.returns(page)
+
+    assert_raises(Blackbook::BadCredentialsError) do
+      @importer.login
+    end
+
+  end
+
+  def test_login
+    response = {'content-type' => 'text/html'}
+
+    body = load_fixture('hotmail_login_response_stage_1.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://login.live.com/login.srf?id=2'
+      ).once.returns(page)
+
+    body = load_fixture('hotmail_login_response_stage_2.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:submit).once.returns(page)
+
+    body = load_fixture('hotmail_login_response_stage_3.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with(
+    'http://www.hotmail.msn.com/cgi-bin/sbox?t=9m9!oYK!qAKigHv8qDw1X1CMCqcbK10gIMWFQjyEK5dofUz!Aneq2zVs58p30sL*Vn8LAwVDD7952o!s1iny8uLHnZxso8YLLoQo4Z3CxhDx11tpPojydwUvwNF9zLFas7&p=9ZiSOOa!3CzsA5AbkqScGrB6w0zuXf!S*UaqQfk0fTM54toklVx51mUYcS3O8JAxjQmsQiwdEBIueipsTADcMs!2RRit547E5MtOMhA6NdiUNkoRvaGsSXK*j1iKAm7Z5jpdKpvW8oSv5yNpN5mWoULuD6lUEqC7i0KHR22e3NJ95C8uGbIGfgxkCjluo7Ye0eJcGraNIAbG0$&lc=1033&id=2'
+      ).once.returns(page)
+
+    assert_nothing_raised do
+      @importer.login
+    end
+  end
+
+  def test_scrape_contacts_not_logged_in
+    @importer.agent.expects(:cookies).once.returns([])
+    assert_raises(Blackbook::BadCredentialsError) do
+      @importer.scrape_contacts
+    end
+  end
+
+  def test_scrape_contacts
+    cookie = WWW::Mechanize::Cookie.new('MSPPre', 'user@hotmail.com')
+    cookie.domain = 'localhost'
+    @importer.agent.expects(:cookies).once.returns([cookie])
+
+    response = {'content-type' => 'text/html'}
+    body = load_fixture('hotmail_contacts.html').join
+    uri = URI.parse('http://by135w.bay135.mail.live.com/mail/')
+    page = WWW::Mechanize::Page.new(uri, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('PrintShell.aspx?type=contact').once.returns(page)
+
+    assert_nothing_raised do
+      contacts = @importer.scrape_contacts
+      assert_equal 2, contacts.size
+
+      # Joe doesn't have a name, so we use his display name (which hotmail sets to his email)
+      assert contacts.detect{|c| c[:email] == 'joe.user@example.com'}
+      assert contacts.detect{|c| c[:name] == 'joe.user@example.com'}
+
+      # Jane has both a name and an email
+      assert contacts.detect{|c| c[:email] == 'jane.user@example.com'}
+      assert contacts.detect{|c| c[:name] == 'Jane User'}
+    end
+  end
+
+  def test_prepare
+    @importer.expects(:login).once
+    @importer.prepare
+  end
+
+  def test_importer_match
+    assert_equal false, @importer =~ nil
+    assert_equal false, @importer =~ {}
+    assert_equal false, @importer =~ {'username' => 'joe@example.com'}
+    assert_equal false, @importer =~ {:username => 'joe@example.com'}
+    assert_equal false, @importer =~ {:username => 'joe'}
+
+    domains = ["compaq.net",
+               "hotmail.co.jp",
+               "hotmail.co.uk",
+               "hotmail.com",
+               "hotmail.de",
+               "hotmail.fr",
+               "hotmail.it",
+               "messengeruser.com",
+               "msn.com",
+               "passport.com",
+               "webtv.net"]
+    domains.each do |domain|
+      assert_equal true, @importer =~ {:username => "joe@#{domain}"}
+      assert_equal true, @importer =~ {:username => "joe@#{domain}".upcase}
+    end
+  end
+
+
+  def test_login_url
+    assert_equal "https://login.live.com/ppsecure/post.srf", @importer.login_url
+    @importer.options = {:username => 'user@msn.com', :password => 'password'}
+    assert_equal "https://msnia.login.live.com/ppsecure/post.srf", @importer.login_url
+  end
+
+  def test_current_host
+    response = {'content-type' => 'text/html'}
+    page = WWW::Mechanize::Page.new(URI.parse('http://localhost/'), response, '<html></html>', code=nil, mech=nil)
+
+    importer = Blackbook::Importer::Hotmail.new
+    assert_nil importer.current_host
+    importer.create_agent
+    assert_nil importer.current_host
+    importer.agent.expects(:current_page).times(2).returns(page)
+    assert_equal 'http://localhost', importer.current_host
+  end
+
+  def test_username_domain(username = nil)
+    importer = Blackbook::Importer::Hotmail.new
+    assert_nil importer.username_domain
+    assert_equal 'example.com', importer.username_domain('user@example.com')
+    assert_equal 'example.com', @importer.username_domain('user@example.com')
+    assert_equal 'hotmail.com', @importer.username_domain
+  end
+
+end

data/test/test_blackbook_importer_page_scraper.rb

@@ -0,0 +1,51 @@
+require File.join( File.dirname(__FILE__), '..', 'lib', 'blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestBlackbookImporterPageScraper < Test::Unit::TestCase
+
+  include TestHelper
+
+  def setup
+    @scraper = Blackbook::Importer::PageScraper.new
+  end
+
+  def test_mechanize_patch_to_absolute_uri
+    @scraper.create_agent
+    response = {'content-type' => 'text/html'}
+    page = WWW::Mechanize::Page.new(uri=nil, response, '<html></html>', code=nil, mech=nil)
+    url = "http://localhost/?arg=val&jank=AAA%3D"
+    assert_equal URI.parse(url), @scraper.agent.send(:to_absolute_uri, url)
+    # pattern of odd URL created by javascript validator in AOL webmail login
+    url = "http://localhost/?arg=val&jank=AAA%3D%3D"
+    assert_equal URI.parse(url), @scraper.agent.send(:to_absolute_uri, url)
+  end
+
+  def test_create_agent
+    agent = nil
+    assert_nothing_raised do
+      agent = @scraper.create_agent
+    end
+    assert_equal agent, @scraper.agent
+    assert_equal true, agent.is_a?(WWW::Mechanize)
+    assert_equal "Mozilla/4.0 (compatible; Blackbook #{Blackbook::VERSION})", agent.user_agent
+    assert_equal false, agent.keep_alive
+  end
+
+  def test_strip_html
+    assert_equal '', @scraper.strip_html('<foo></foo>')
+    assert_equal 'bar', @scraper.strip_html('<foo>bar</foo>')
+    assert_equal 'bar', @scraper.strip_html('bar</foo>')
+    assert_equal 'bar', @scraper.strip_html('<foo>bar')
+  end
+
+  def test_fetch_contacts!
+    @scraper.expects(:create_agent).once
+    @scraper.expects(:prepare).once
+    @scraper.expects(:scrape_contacts).once
+    assert_nothing_raised do
+      agent = @scraper.fetch_contacts!
+    end
+  end
+end

data/test/test_blackbook_importer_yahoo.rb

@@ -0,0 +1,97 @@
+require File.join( File.dirname(__FILE__), '../lib/blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestBlackbookImporterYahoo < Test::Unit::TestCase
+
+  include TestHelper
+
+  def setup
+    @importer = Blackbook::Importer::Yahoo.new
+    @importer.options = {:username => 'user@yahoo.com', :password => 'password'}
+    @importer.create_agent
+  end
+
+  def test_prepare
+    @importer.expects(:login).once
+    @importer.prepare
+  end
+
+  def test_importer_match
+    assert_equal false, @importer =~ nil
+    assert_equal false, @importer =~ {}
+    assert_equal false, @importer =~ {'username' => 'joe@example.com'}
+    assert_equal false, @importer =~ {:username => 'joe@example.com'}
+    assert_equal false, @importer =~ {:username => 'joe'}
+    assert_equal true, @importer =~ {:username => 'joe@yahoo.com'}
+    assert_equal true, @importer =~ {:username => 'JOE@YAHOO.COM'}
+  end
+
+  def test_login
+    response = {'content-type' => 'text/html'}
+
+    body = load_fixture('yahoo_login_response_stage_1.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('https://login.yahoo.com/config/login_verify2?').once.returns(page)
+
+    body = load_fixture('yahoo_login_response_stage_2.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:submit).once.returns(page)
+
+    assert_nothing_raised do
+      assert @importer.login
+    end
+  end
+
+  def test_bad_login
+    response = {'content-type' => 'text/html'}
+
+    body = load_fixture('yahoo_login_response_stage_1.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('https://login.yahoo.com/config/login_verify2?').once.returns(page)
+
+    body = load_fixture('yahoo_bad_login_response_stage_2.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:submit).once.returns(page)
+
+    assert_raises(Blackbook::BadCredentialsError) do
+      @importer.login
+    end
+  end
+
+  def test_scrape_contacts_raises_badcredentialerror_when_not_logged_in
+    response = {'content-type' => 'text/html'}
+    body = load_fixture('yahoo_contacts_not_logged_in.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://address.yahoo.com/?1=&VPC=import_export'
+      ).once.returns(page)
+
+    assert_raises(Blackbook::BadCredentialsError) do
+      @importer.scrape_contacts
+    end
+  end
+
+  def test_scrape_contacts
+    response = {'content-type' => 'text/html'}
+    body = load_fixture('yahoo_contacts_stage_1.html').join
+    page = WWW::Mechanize::Page.new(uri=nil, response, body, code=nil, mech=nil)
+    @importer.agent.expects(:get).with('http://address.yahoo.com/?1=&VPC=import_export'
+      ).once.returns(page)
+
+    response = {'content-type' => 'text/csv; charset=UTF-8'}
+    body = load_fixture('yahoo_contacts.csv').join
+    page = WWW::Mechanize::File.new(uri=nil, response, body, code=nil)
+    @importer.agent.expects(:submit).once.returns(page)
+
+    assert_nothing_raised do
+      contacts = @importer.scrape_contacts
+      assert_equal 2, contacts.size
+      assert contacts.detect{|c| c[:email] == 'joe.user@example.com'}
+      assert contacts.detect{|c| c[:name] == 'Joe User'}
+      assert contacts.detect{|c| c[:email] == 'jane.user@example.com'}
+      assert contacts.detect{|c| c[:name] == 'Jane User'}
+    end
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,69 @@
+module TestHelper
+
+  def fixture_path(name)
+    "#{File.dirname(__FILE__)}/fixtures/#{name}"
+  end
+
+  def load_fixture(file)
+    IO.readlines(fixture_path(file))
+  end
+
+  def fixture_file(name)
+    File.new fixture_path(name)
+  end
+
+end
+
+module Net #:nodoc:
+  class HTTP #:nodoc:
+
+    extend TestHelper
+
+    RESPONSES = {}
+
+    def self.responses=(r)
+      RESPONSES.clear
+      r.each{|k,v| RESPONSES[k] = v}
+    end
+
+
+    alias :old_net_http_request :request
+
+    def request(req, body = nil, &block)
+      prot = use_ssl ? "https" : "http"
+      uri_cls = use_ssl ? URI::HTTPS : URI::HTTP
+      query = req.path.split('?',2)
+      opts = {:host => self.address,
+             :port => self.port, :path => query[0]}
+      opts[:query] = query[1] if query[1]
+      uri = uri_cls.build(opts)
+      raise ArgumentError.new("#{req.method} method to #{uri} not being handled in testing")
+    end
+
+    def connect
+      raise ArgumentError.new("connect not being handled in testing")
+    end
+  end
+end
+
+plugin_root = "#{ File.dirname(__FILE__) }/.."
+
+require 'rubygems'
+require 'mocha'
+
+module BlackbookExtensions
+  module TestHelper
+    class Credentials
+      attr_accessor :network, :config
+
+      def initialize(network)
+        self.network = network.to_s
+        self.config = YAML.load_file( File.join(File.dirname(__FILE__), "config/credentials.yml") )
+      end
+
+      def get
+        [ self.config[self.network]["username"], self.config[self.network]["password"] ]
+      end
+    end
+  end
+end