blackbook 1.0.0

data/test/test_blackbook.rb

@@ -0,0 +1,60 @@
+require File.join( File.dirname(__FILE__), '../lib/blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestBlackbook < Test::Unit::TestCase
+
+  include TestHelper
+
+  def test_class_get
+    Blackbook.any_instance.expects(:get)
+    Blackbook.get
+  end
+
+  def test_class_instance
+    assert Blackbook.instance.is_a?(Blackbook)
+  end
+
+  def test_class_register
+    Blackbook.register(:base, Blackbook::Exporter::Base)
+    assert Blackbook.instance.exporters[:base].is_a?(Blackbook::Exporter::Base)
+    
+    Blackbook.register(:base, Blackbook::Importer::Base)
+    assert Blackbook.instance.importers[:base].is_a?(Blackbook::Importer::Base)
+  end
+
+  def test_class_register_raises_on_bad_register
+    assert_raises(ArgumentError) do
+      Blackbook.register(:bad, String)
+    end
+  end
+
+  def test_export
+    importer = Blackbook::Importer::Base.new
+    exporter = Blackbook::Exporter::Base.new
+    importer.expects(:import)
+    exporter.expects(:export)
+    Blackbook.instance.export( importer, exporter, {} )
+  end
+
+  def test_find_importer
+    base = Blackbook::Importer::Base.new
+    base.stubs(:=~).returns(true)
+    Blackbook.any_instance.stubs(:importers).returns({:basic => base})
+    assert_equal base, Blackbook.instance.find_importer(:as => :basic)
+    
+    base.stubs(:=~).returns(false)
+    assert_nil Blackbook.instance.find_importer(:as => :basic)
+  end
+
+  def test_get
+    cards = Blackbook.instance.get( :csv, :file => fixture_file('gmail.csv') )
+    assert_equal 2, cards.size
+    assert cards.detect{|card| card[:name] == "Joe User"}
+    assert cards.detect{|card| card[:name] == "Some Guy"}
+    assert cards.detect{|card| card[:email] == "joeuser@example.com"}
+    assert cards.detect{|card| card[:email] == "someguy@example.com"}
+  end
+  
+end

data/test/test_blackbook_exporter_base.rb

@@ -0,0 +1,16 @@
+require File.join( File.dirname(__FILE__), '../lib/blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestBlackbookExporterBase < Test::Unit::TestCase
+
+  include TestHelper
+
+  def test_export
+    exporter = Blackbook::Exporter::Base.new
+    contacts = [{ :name => 'Test', :email => 'test@domain.com' }]
+    assert_equal contacts, exporter.export(contacts)
+  end
+
+end

data/test/test_blackbook_exporter_vcf.rb

@@ -0,0 +1,52 @@
+require File.join( File.dirname(__FILE__), '../lib/blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestBlackbookExporterVcf < Test::Unit::TestCase
+
+  include TestHelper
+
+  def setup
+    @card = Blackbook::Exporter::Vcf::Vcard.new(
+             {'first' => 'joe', 'last' => 'user', 
+              'email' => 'joe.user@example.com'})
+    @exporter = Blackbook::Exporter::Vcf.new
+  end
+
+  def test_vcard
+    assert_equal 'joe', @card.first
+    assert_equal 'user', @card.last
+    assert_equal 'joe.user@example.com', @card.email
+  end
+
+  def test_vcard_to_s
+    card = %q!BEGIN:VCARD
+N:user;joe
+EMAIL:joe.user@example.com
+END:VCARD
+!
+    assert_equal card, @card.to_s
+    assert_equal 'user', @card.last
+    assert_equal 'joe.user@example.com', @card.email
+  end
+
+  def test_export
+    contacts = [{ :name => 'Test', :email => 'test@example.com' }]
+    expected = <<VCF
+BEGIN:VCARD
+N:;Test
+EMAIL:test@example.com
+END:VCARD
+VCF
+    exported = @exporter.export(contacts)
+    assert_equal 1, exported.size
+    assert_equal expected, exported.first.to_s
+  end
+
+  def test_export_when_contacts_blank
+    assert_nil @exporter.export()
+    assert_nil @exporter.export([])
+  end
+
+end

data/test/test_blackbook_exporter_xml.rb

@@ -0,0 +1,16 @@
+require File.join( File.dirname(__FILE__), '../lib/blackbook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestBlackbookExporterXml < Test::Unit::TestCase
+
+  include TestHelper
+
+  def test_export
+    exporter = Blackbook::Exporter::Xml.new
+    contacts = [{ :name => 'Test', :email => 'test@domain.com' }]
+    assert_equal "<?xml version='1.0'?><contacts><contact><name>Test</name><email>test@domain.com</email></contact></contacts>", exporter.export(contacts)
+  end
+
+end

data/test/test_blackbook_importer_aol.rb

@@ -0,0 +1,107 @@
+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('http://webmail.aol.com/31361/aol/en-us/Suite.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_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/aim/en-us/Lite/addresslist-print.aspx?command=all&sort=FirstLastNick&sortDir=Ascending&nameFormat=FirstLastNick&user=blackbookuid'
+    ).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,96 @@
+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/').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/').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_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,143 @@
+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
+      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
+
+  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