graybook 1.0.22

data/test/gmx_importer_test.rb

@@ -0,0 +1,53 @@
+require File.dirname(__FILE__) + '/../test_helper.rb'
+
+context "The GMX importer" do
+
+  setup do
+    @username, @password = GraybookExtensions::TestHelper::Credentials.new(:gmx).get
+
+    @importer = Graybook::Importer::GMX.new
+    @importer.create_agent
+  end
+
+  specify "should match emails" do
+    @importer.=~(nil).should.not.be
+    @importer.=~({}).should.not.be
+    @importer.=~({ 'username' => "john@foo.com" }).should.not.be
+    @importer.=~({ :username => "john@foo.com" }).should.not.be
+    @importer.=~({ :username => "john" }).should.not.be
+    @importer.=~({ :username => "john@gmx.de" }).should.be
+    @importer.=~({ :username => "JOHN@GMX.DE" }).should.be
+  end
+
+  specify "should be able to login with correct credentials" do
+    should.not.raise Exception do
+      login(@username, @password)
+    end
+  end
+
+  specify "should raise BadCredentialsError on login with wrong credentials" do
+    should.raise Graybook::BadCredentialsError do
+      login
+    end
+  end
+
+  specify "should raise BadCredentialsError on login with correct username and wrong password" do
+    should.raise Graybook::BadCredentialsError do
+      login(@username)
+    end
+  end
+
+  specify "should able to get contacts" do
+    login(@username, @password)
+
+    contacts = @importer.scrape_contacts
+    contacts.should.not.be.empty
+    contacts.should.be.all { |contact| contact.is_a? Hash }
+  end
+
+  private
+    def login(username = "foo", password = "bar")
+      @importer.options = { :username => username, :password => password }
+      @importer.login
+    end
+end

data/test/scripts/live_test.rb

@@ -0,0 +1,25 @@
+#!/usr/bin/env ruby
+require File.join( File.dirname(__FILE__), '../../lib/graybook.rb' )
+require 'optparse'
+
+options = {}
+importer = :auto
+
+opts = OptionParser.new do |opts|
+  opts.on("-u USERNAME") do |username|
+    options[:username] = username
+  end
+  opts.on("-p PASSWORD") do |password|
+    options[:password] = password
+  end
+  opts.on("-i [IMPORTER]") do |importer_name|
+    importer = importer_name.to_sym
+  end
+  opts.on("-x [EXPORTER]") do |exporter_name|
+    options[:as] = exporter_name.to_sym
+  end
+end
+
+opts.parse!(ARGV)
+
+puts contacts = Graybook.get( importer, options )

data/test/test_graybook.rb

@@ -0,0 +1,60 @@
+require File.join( File.dirname(__FILE__), '../lib/graybook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestGraybook < Test::Unit::TestCase
+
+  include TestHelper
+
+  def test_class_get
+    Graybook.any_instance.expects(:get)
+    Graybook.get
+  end
+
+  def test_class_instance
+    assert Graybook.instance.is_a?(Graybook)
+  end
+
+  def test_class_register
+    Graybook.register(:base, Graybook::Exporter::Base)
+    assert Graybook.instance.exporters[:base].is_a?(Graybook::Exporter::Base)
+    
+    Graybook.register(:base, Graybook::Importer::Base)
+    assert Graybook.instance.importers[:base].is_a?(Graybook::Importer::Base)
+  end
+
+  def test_class_register_raises_on_bad_register
+    assert_raises(ArgumentError) do
+      Graybook.register(:bad, String)
+    end
+  end
+
+  def test_export
+    importer = Graybook::Importer::Base.new
+    exporter = Graybook::Exporter::Base.new
+    importer.expects(:import)
+    exporter.expects(:export)
+    Graybook.instance.export( importer, exporter, {} )
+  end
+
+  def test_find_importer
+    base = Graybook::Importer::Base.new
+    base.stubs(:=~).returns(true)
+    Graybook.any_instance.stubs(:importers).returns({:basic => base})
+    assert_equal base, Graybook.instance.find_importer(:as => :basic)
+    
+    base.stubs(:=~).returns(false)
+    assert_nil Graybook.instance.find_importer(:as => :basic)
+  end
+
+  def test_get
+    cards = Graybook.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_graybook_exporter_base.rb

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

data/test/test_graybook_exporter_vcf.rb

@@ -0,0 +1,52 @@
+require File.join( File.dirname(__FILE__), '../lib/graybook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestGraybookExporterVcf < Test::Unit::TestCase
+
+  include TestHelper
+
+  def setup
+    @card = Graybook::Exporter::Vcf::Vcard.new(
+             {'first' => 'joe', 'last' => 'user', 
+              'email' => 'joe.user@example.com'})
+    @exporter = Graybook::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_graybook_exporter_xml.rb

@@ -0,0 +1,16 @@
+require File.join( File.dirname(__FILE__), '../lib/graybook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestGraybookExporterXml < Test::Unit::TestCase
+
+  include TestHelper
+
+  def test_export
+    exporter = Graybook::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_graybook_importer_aol.rb

@@ -0,0 +1,108 @@
+require File.join( File.dirname(__FILE__), '../lib/graybook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+require 'mechanize'
+
+
+class TestGraybookImporterAol < Test::Unit::TestCase
+
+  include TestHelper
+
+  def setup
+    @importer = Graybook::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('/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(Graybook::BadCredentialsError) do
+      @importer.login
+    end
+  end
+
+  def test_scrape_contacts_raises_badcredentialerror_when_not_logged_in
+    @importer.agent.expects(:cookies).once.returns([])
+    assert_raises(Graybook::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:graybookuid&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'}
+      assert contacts.detect{|c| c[:name] == 'NoEmail Dude'} == nil
+    end
+  end
+
+end

data/test/test_graybook_importer_base.rb

@@ -0,0 +1,24 @@
+require File.join( File.dirname(__FILE__), '../lib/graybook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestGraybookImporterBase < Test::Unit::TestCase
+
+  include TestHelper
+
+  def test_service_name
+    assert_equal "Base", Graybook::Importer::Base.new.service_name
+  end
+
+  def test_import
+    importer = Graybook::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_graybook_importer_csv.rb

@@ -0,0 +1,60 @@
+require File.join( File.dirname(__FILE__), '../lib/graybook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'tempfile'
+require 'test/unit'
+require 'mocha'
+
+class TestGraybookImporterCsv < Test::Unit::TestCase
+
+  include TestHelper
+
+  def setup
+    @importer = Graybook::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_graybook_importer_gmail.rb

@@ -0,0 +1,116 @@
+require File.join( File.dirname(__FILE__), '../lib/graybook.rb' )
+require File.join( File.dirname(__FILE__), 'test_helper.rb' )
+require 'test/unit'
+require 'mocha'
+
+class TestGraybookImporterGmail < Test::Unit::TestCase
+
+  include TestHelper
+
+  def setup
+    @importer = Graybook::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(Graybook::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
+      @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
+      @importer.login
+    end
+  end
+
+  def test_scrape_contacts_raises_badcredentialerror_when_not_logged_in
+    @importer.agent.expects(:cookies).once.returns([])
+    assert_raises(Graybook::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