keyrack 0.3.0.pre → 0.3.0

data/test/unit/test_site.rb

@@ -0,0 +1,181 @@
+require 'helper'
+
+class TestSite < Test::Unit::TestCase
+  def new_site(*args)
+    Keyrack::Site.new(*args)
+  end
+
+  test "creating new site" do
+    site = new_site("Enterprise", "picard", "livingston")
+    assert_equal "Enterprise", site.name
+  end
+
+  test "name setter" do
+    site = new_site("Enterprise", "picard", "livingston")
+    site.name = "NCC1701D"
+    assert_equal "NCC1701D", site.name
+  end
+
+  test "username" do
+    site = new_site("Enterprise", "picard", "livingston")
+    assert_equal "picard", site.username
+  end
+
+  test "username setter" do
+    site = new_site("Enterprise", "picard", "livingston")
+    site.username = "jean_luc"
+    assert_equal "jean_luc", site.username
+  end
+
+  test "password" do
+    site = new_site("Enterprise", "picard", "livingston")
+    assert_equal "livingston", site.password
+  end
+
+  test "password setter" do
+    site = new_site("Enterprise", "picard", "livingston")
+    site.password = "crusher"
+    assert_equal "crusher", site.password
+  end
+
+  test "loading site from hash" do
+    hash = {
+      'name' => 'Enterprise',
+      'username' => 'picard',
+      'password' => 'livingston'
+    }
+    site = new_site(hash)
+    assert_equal 'Enterprise', site.name
+    assert_equal 'picard', site.username
+    assert_equal 'livingston', site.password
+  end
+
+  test "loading site from hash with missing name" do
+    hash = {
+      'username' => 'picard',
+      'password' => 'livingston'
+    }
+    assert_raises(ArgumentError) do
+      site = new_site(hash)
+    end
+  end
+
+  test "loading site from hash with non-string name" do
+    hash = {
+      'name' => [123],
+      'username' => 'picard',
+      'password' => 'livingston'
+    }
+    assert_raises(ArgumentError) do
+      site = new_site(hash)
+    end
+  end
+
+  test "loading site from hash with missing username" do
+    hash = {
+      'name' => 'Enterprise',
+      'password' => 'livingston'
+    }
+    assert_raises(ArgumentError) do
+      site = new_site(hash)
+    end
+  end
+
+  test "loading site from hash with non-string username" do
+    hash = {
+      'name' => 'Enterprise',
+      'username' => [123],
+      'password' => 'livingston'
+    }
+    assert_raises(ArgumentError) do
+      site = new_site(hash)
+    end
+  end
+
+  test "loading site from hash with missing password" do
+    hash = {
+      'name' => 'Enterprise',
+      'username' => 'picard'
+    }
+    assert_raises(ArgumentError) do
+      site = new_site(hash)
+    end
+  end
+
+  test "loading site from hash with non-string password" do
+    hash = {
+      'name' => 'Enterprise',
+      'username' => 'picard',
+      'password' => [123]
+    }
+    assert_raises(ArgumentError) do
+      site = new_site(hash)
+    end
+  end
+
+  test "name change event" do
+    site = new_site("Enterprise", "picard", "livingston")
+
+    called = false
+    site.after_event do |event|
+      called = true
+      assert_equal 'change', event.name
+      assert_same site, event.owner
+      assert_equal 'name', event.attribute_name
+      assert_equal 'Enterprise', event.previous_value
+      assert_equal 'NCC1701D', event.new_value
+    end
+    site.name = "NCC1701D"
+    assert called
+  end
+
+  test "password changed callback" do
+    site = new_site("Enterprise", "picard", "livingston")
+
+    called = false
+    site.after_event do |event|
+      called = true
+      assert_same site, event.owner
+      assert_equal 'change', event.name
+      assert_equal 'password', event.attribute_name
+      assert_equal 'livingston', event.previous_value
+      assert_equal 'crusher', event.new_value
+    end
+    site.password = "crusher"
+    assert called
+  end
+
+  test "username changed callback" do
+    site = new_site("Enterprise", "picard", "livingston")
+
+    called = false
+    site.after_event do |event|
+      called = true
+      assert_same site, event.owner
+      assert_equal 'change', event.name
+      assert_equal 'username', event.attribute_name
+      assert_equal 'picard', event.previous_value
+      assert_equal 'jean_luc', event.new_value
+    end
+    site.username = "jean_luc"
+    assert called
+  end
+
+  test "serializing to yaml" do
+    site = new_site("Enterprise", "picard", "livingston")
+    expected = {
+      'name' => 'Enterprise',
+      'username' => 'picard',
+      'password' => 'livingston'
+    }.to_yaml
+    assert_equal expected, site.to_yaml
+  end
+
+  test "sites with same name and username are equal" do
+    site_1 = new_site("Enterprise", "picard", "livingston")
+    site_2 = new_site("Enterprise", "picard", "crusher")
+    site_3 = new_site("Enterprise", "jean_luc", "crusher")
+    assert_equal site_1, site_2
+    assert_not_equal site_1, site_3
+  end
+end

data/test/unit/test_store.rb

@@ -0,0 +1,13 @@
+require 'helper'
+
+class TestStore < Test::Unit::TestCase
+  def test_get_filesystem
+    assert_equal Keyrack::Store::Filesystem, Keyrack::Store[:filesystem]
+    assert_equal Keyrack::Store::Filesystem, Keyrack::Store['filesystem']
+  end
+
+  def test_get_ssh
+    assert_equal Keyrack::Store::SSH, Keyrack::Store[:ssh]
+    assert_equal Keyrack::Store::SSH, Keyrack::Store['ssh']
+  end
+end

data/test/unit/test_utils.rb

@@ -0,0 +1,8 @@
+require 'helper'
+
+class TestUtils < Test::Unit::TestCase
+  def test_generate_password
+    result = Keyrack::Utils.generate_password
+    assert_match /^[!-~]{8}$/, result
+  end
+end

data/test/unit/ui/test_console.rb

@@ -0,0 +1,495 @@
+require 'helper'
+
+class TestConsole < Test::Unit::TestCase
+  def setup
+    @twitter = stub('Twitter', {
+      :name => 'Twitter',
+      :username => 'tweeb',
+      :password => 'secret'
+    })
+    @google_1 = stub('Google 1', {
+      :name => 'Google',
+      :username => 'catfan',
+      :password => 'kitty'
+    })
+    @google_2 = stub('Google 2', {
+      :name => 'Google',
+      :username => 'dogfan',
+      :password => 'puppy'
+    })
+    @top_group = stub('top group', {
+      :sites => [@twitter, @google_1, @google_2],
+      :group_names => []
+    })
+    @top_group.stubs(:site).with(0).returns(@twitter)
+    @top_group.stubs(:site).with(1).returns(@google_1)
+    @top_group.stubs(:site).with(2).returns(@google_2)
+
+    @highline = stub('highline')
+    @highline.stubs(:color).with("Keyrack Main Menu", instance_of(Symbol)).
+      returns("Keyrack Main Menu")
+    #@highline.stubs(:say)
+    HighLine.expects(:new).returns(@highline)
+    @console = Keyrack::UI::Console.new
+
+    HighLine::SystemExtensions.stubs(:terminal_size).returns([20, 20])
+  end
+
+  test "select login from menu" do
+    seq = sequence('say')
+    @highline.expects(:say).with("=== Keyrack Main Menu ===")
+    @highline.expects(:say).with("1. Twitter [tweeb]")
+    @highline.expects(:say).with("2. Google [catfan]")
+    @highline.expects(:say).with("3. Google [dogfan]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [m]ode [q]uit")
+
+    question = mock('question')
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 2 3 e g}) }).returns('1')
+    Clipboard.expects(:copy).with('secret')
+    @highline.expects(:say).with("The password has been copied to your clipboard.")
+    assert_nil @console.menu(:group => @top_group, :at_top => true)
+  end
+
+  test "select entry from menu in print mode" do
+    seq = sequence('say')
+    @console.mode = :print
+    @highline.expects(:say).with("=== Keyrack Main Menu ===")
+    @highline.expects(:say).with("1. Twitter [tweeb]")
+    @highline.expects(:say).with("2. Google [catfan]")
+    @highline.expects(:say).with("3. Google [dogfan]")
+    @highline.expects(:say).with("Mode: print")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [m]ode [q]uit")
+
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 2 3 e g}) }).returns('1')
+    @highline.expects(:color).with('secret', instance_of(Symbol)).
+      returns('secret').in_sequence(seq)
+    question = mock do
+      expects(:echo=).with(false)
+      if HighLine::SystemExtensions::CHARACTER_MODE != 'stty'
+        expects(:character=).with(true)
+        expects(:overwrite=).with(true)
+      end
+    end
+    @highline.expects(:ask).
+      with('Here you go: secret. Done? ').
+      yields(question).
+      returns('')
+
+    assert_nil @console.menu(:group => @top_group, :at_top => true)
+  end
+
+  test "select new from menu" do
+    @highline.expects(:say).with("=== Keyrack Main Menu ===")
+    @highline.expects(:say).with("1. Twitter [tweeb]")
+    @highline.expects(:say).with("2. Google [catfan]")
+    @highline.expects(:say).with("3. Google [dogfan]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [m]ode [q]uit")
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 2 3 e g}) }).returns('n')
+    assert_equal :new, @console.menu(:group => @top_group, :at_top => true)
+  end
+
+  test "select edit from menu" do
+    @highline.expects(:say).with("=== Keyrack Main Menu ===")
+    @highline.expects(:say).with("1. Twitter [tweeb]")
+    @highline.expects(:say).with("2. Google [catfan]")
+    @highline.expects(:say).with("3. Google [dogfan]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [m]ode [q]uit")
+    question = mock('question')
+    question.expects(:in=).with(%w{n q m 1 2 3 e g})
+    @highline.expects(:ask).yields(question).returns('e')
+    assert_equal :edit, @console.menu(:group => @top_group, :at_top => true)
+  end
+
+  test "select quit from menu" do
+    @highline.expects(:say).with("=== Keyrack Main Menu ===")
+    @highline.expects(:say).with("1. Twitter [tweeb]")
+    @highline.expects(:say).with("2. Google [catfan]")
+    @highline.expects(:say).with("3. Google [dogfan]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [m]ode [q]uit")
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 2 3 e g}) }).returns('q')
+    assert_equal :quit, @console.menu(:group => @top_group, :at_top => true)
+  end
+
+  test "select quit from menu when database is dirty" do
+    @highline.expects(:say).with("=== Keyrack Main Menu ===")
+    @highline.expects(:say).with("1. Twitter [tweeb]")
+    @highline.expects(:say).with("2. Google [catfan]")
+    @highline.expects(:say).with("3. Google [dogfan]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [s]ave [m]ode [q]uit")
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 2 3 e g s}) }).returns('q')
+    @highline.expects(:agree).with("Really quit?  You have unsaved changes! [yn] ").returns(false)
+    assert_equal nil, @console.menu(:group => @top_group, :at_top => true, :dirty => true)
+  end
+
+  test "select save from menu" do
+    @highline.expects(:say).with("=== Keyrack Main Menu ===")
+    @highline.expects(:say).with("1. Twitter [tweeb]")
+    @highline.expects(:say).with("2. Google [catfan]")
+    @highline.expects(:say).with("3. Google [dogfan]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [s]ave [m]ode [q]uit")
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 2 3 e g s}) }).returns('s')
+    assert_equal :save, @console.menu(:group => @top_group, :at_top => true, :dirty => true)
+  end
+
+  test "select group from menu" do
+    @top_group.stubs(:group_names).returns(["Blargh"])
+    blargh = stub('Blargh group')
+    @top_group.stubs(:group).with('Blargh').returns(blargh)
+
+    @highline.stubs(:color).with('Blargh', instance_of(Symbol)).
+      returns('Blargh')
+    @highline.expects(:say).with("=== Keyrack Main Menu ===")
+    @highline.expects(:say).with("1. Blargh")
+    @highline.expects(:say).with("2. Twitter [tweeb]")
+    @highline.expects(:say).with("3. Google [catfan]")
+    @highline.expects(:say).with("4. Google [dogfan]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [m]ode [q]uit")
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 2 3 4 e g}) }).returns('1')
+    assert_equal({:group => blargh}, @console.menu(:group => @top_group, :at_top => true))
+  end
+
+  test "select entry from group menu" do
+    facebook = stub('Facebook site', {
+      :name => 'Facebook',
+      :username => 'username',
+      :password => 'password'
+    })
+    foo_group = stub('Foo group', {
+      :name => 'Foo',
+      :sites => [facebook],
+      :group_names => []
+    })
+
+    @highline.expects(:color).with("Foo", instance_of(Symbol)).
+      returns("Foo")
+    @highline.expects(:say).with("======== Foo =========")
+    @highline.expects(:say).with("1. Facebook [username]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [t]op [m]ode [q]uit")
+
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 e g t}) }).returns('1')
+    Clipboard.expects(:copy).with('password')
+    @highline.expects(:say).with("The password has been copied to your clipboard.")
+    assert_nil @console.menu(:group => foo_group, :at_top => false)
+  end
+
+  test "get password" do
+    question = mock('question')
+    question.expects(:echo=).with(false)
+    @highline.expects(:ask).with("Keyrack password: ").yields(question).returns("foobar")
+    assert_equal "foobar", @console.get_password
+  end
+
+  test "get new entry with manual password" do
+    seq = sequence("new entry")
+    @highline.expects(:ask).with("Label: ").returns("Foo").in_sequence(seq)
+    @highline.expects(:ask).with("Username: ").returns("bar").in_sequence(seq)
+    @highline.expects(:ask).with("Generate password? [ync] ").returns("n").in_sequence(seq)
+    @highline.expects(:ask).with("Password: ").yields(mock { expects(:echo=).with(false) }).returns("baz").in_sequence(seq)
+    @highline.expects(:ask).with("Password (again): ").yields(mock { expects(:echo=).with(false) }).returns("bar").in_sequence(seq)
+    @highline.expects(:say).with("Passwords didn't match. Try again!").in_sequence(seq)
+    @highline.expects(:ask).with("Password: ").yields(mock { expects(:echo=).with(false) }).returns("baz").in_sequence(seq)
+    @highline.expects(:ask).with("Password (again): ").yields(mock { expects(:echo=).with(false) }).returns("baz").in_sequence(seq)
+    assert_equal({:site => "Foo", :username => "bar", :password => "baz"}, @console.get_new_entry)
+  end
+
+  test "get new entry with generated password" do
+    seq = sequence("new entry")
+    @highline.expects(:ask).with("Label: ").returns("Foo").in_sequence(seq)
+    @highline.expects(:ask).with("Username: ").returns("bar").in_sequence(seq)
+    @highline.expects(:ask).with("Generate password? [ync] ").returns("y").in_sequence(seq)
+    Keyrack::Utils.expects(:generate_password).returns('foobar').in_sequence(seq)
+    @highline.expects(:color).with('foobar', instance_of(Symbol)).
+      returns('foobar').in_sequence(seq)
+    @highline.expects(:ask).with("Generated foobar.  Sound good? [ync] ").returns("n").in_sequence(seq)
+    Keyrack::Utils.expects(:generate_password).returns('foobar').in_sequence(seq)
+    @highline.expects(:color).with('foobar', instance_of(Symbol)).
+      returns('foobar').in_sequence(seq)
+    @highline.expects(:ask).with("Generated foobar.  Sound good? [ync] ").returns("y").in_sequence(seq)
+    assert_equal({:site => "Foo", :username => "bar", :password => "foobar"}, @console.get_new_entry)
+  end
+
+  test "get new entry with cancel" do
+    seq = sequence("new entry")
+    @highline.expects(:ask).with("Label: ").returns("Foo").in_sequence(seq)
+    @highline.expects(:ask).with("Username: ").returns("bar").in_sequence(seq)
+
+    question = mock('question')
+    question.expects(:in=).with(%w{y n c})
+    @highline.expects(:ask).with("Generate password? [ync] ").yields(question).returns('c').in_sequence(seq)
+    assert_nil @console.get_new_entry
+  end
+
+  test "get new entry with generated password cancel" do
+    seq = sequence("new entry")
+    @highline.expects(:ask).with("Label: ").returns("Foo").in_sequence(seq)
+    @highline.expects(:ask).with("Username: ").returns("bar").in_sequence(seq)
+    @highline.expects(:ask).with("Generate password? [ync] ").returns("y").in_sequence(seq)
+    Keyrack::Utils.expects(:generate_password).returns('foobar').in_sequence(seq)
+    @highline.expects(:color).with('foobar', instance_of(Symbol)).
+      returns('foobar').in_sequence(seq)
+    @highline.expects(:ask).with("Generated foobar.  Sound good? [ync] ").returns('c').in_sequence(seq)
+    @highline.expects(:ask).with("Password: ").yields(mock { expects(:echo=).with(false) }).returns("baz").in_sequence(seq)
+    @highline.expects(:ask).with("Password (again): ").yields(mock { expects(:echo=).with(false) }).returns("baz").in_sequence(seq)
+    assert_equal({:site => "Foo", :username => "bar", :password => "baz"}, @console.get_new_entry)
+  end
+
+  test "display first time notice" do
+    @highline.expects(:say).with("This looks like your first time using Keyrack.  I'll need to ask you a few questions first.")
+    @console.display_first_time_notice
+  end
+
+  test "password setup" do
+    seq = sequence("password setup")
+    @highline.expects(:ask).with("New passphrase: ").yields(mock{expects(:echo=).with(false)}).returns('huge').in_sequence(seq)
+    @highline.expects(:ask).with("Confirm passphrase: ").yields(mock{expects(:echo=).with(false)}).returns('small').in_sequence(seq)
+    @highline.expects(:say).with("Passphrases didn't match.").in_sequence(seq)
+    @highline.expects(:ask).with("New passphrase: ").yields(mock{expects(:echo=).with(false)}).returns('huge').in_sequence(seq)
+    @highline.expects(:ask).with("Confirm passphrase: ").yields(mock{expects(:echo=).with(false)}).returns('huge').in_sequence(seq)
+    assert_equal 'huge', @console.password_setup
+  end
+
+  test "store setup for filesystem" do
+    @highline.expects(:choose).yields(mock {
+      expects(:header=).with("Choose storage type")
+      expects(:choices).with("filesystem", "ssh")
+    }).returns("filesystem")
+
+    expected = {'type' => 'filesystem', 'path' => 'database'}
+    assert_equal expected, @console.store_setup
+  end
+
+  test "store setup for ssh" do
+    seq = sequence("store setup")
+    @highline.expects(:choose).yields(mock {
+      expects(:header=).with("Choose storage type")
+      expects(:choices).with("filesystem", "ssh")
+    }).returns("ssh").in_sequence(seq)
+    @highline.expects(:ask).with("Host: ").returns("example.com").in_sequence(seq)
+    @highline.expects(:ask).with("User: ").returns("dudeguy").in_sequence(seq)
+    @highline.expects(:ask).with("Remote path: ").returns(".keyrack/database").in_sequence(seq)
+
+    expected = {'type' => 'ssh', 'host' => 'example.com', 'user' => 'dudeguy', 'path' => '.keyrack/database'}
+    assert_equal expected, @console.store_setup
+  end
+
+  test "get new group" do
+    @highline.expects(:ask).with("Group: ").yields(mock {
+      expects(:validate=).with(/^\w[\w\s]*$/)
+    }).returns("Foo")
+    assert_equal('Foo', @console.get_new_group)
+  end
+
+  test "choose entry to edit" do
+    seq = SequenceHelper.new("editing")
+    @highline.expects(:color).with("Choose entry", instance_of(Symbol)).returns("Choose entry")
+    seq << @highline.expects(:say).with('=== Choose entry ===')
+    seq << @highline.expects(:say).with("1. Twitter [tweeb]")
+    seq << @highline.expects(:say).with("2. Google [catfan]")
+    seq << @highline.expects(:say).with("3. Google [dogfan]")
+    seq << @highline.expects(:say).with("c. Cancel")
+    seq << @highline.expects(:ask).yields(mock { expects(:in=).with(%w{c 1 2 3}) }).returns('1')
+    assert_equal({:site => @twitter}, @console.choose_entry_to_edit(@top_group))
+  end
+
+  test "choose group to edit" do
+    seq = SequenceHelper.new("editing")
+    @top_group.expects(:group_names).returns(%w{Foo})
+    @highline.expects(:color).with("Choose entry", instance_of(Symbol)).
+      returns("Choose entry")
+    @highline.expects(:color).with("Foo", instance_of(Symbol)).
+      returns("Foo")
+    seq << @highline.expects(:say).with('=== Choose entry ===')
+    seq << @highline.expects(:say).with("1. Foo")
+    seq << @highline.expects(:say).with("2. Twitter [tweeb]")
+    seq << @highline.expects(:say).with("3. Google [catfan]")
+    seq << @highline.expects(:say).with("4. Google [dogfan]")
+    seq << @highline.expects(:say).with("c. Cancel")
+    seq << @highline.expects(:ask).yields(mock { expects(:in=).with(%w{c 1 2 3 4}) }).returns('1')
+    assert_equal({:group => "Foo"}, @console.choose_entry_to_edit(@top_group))
+  end
+
+  test "choose entry to edit with cancel" do
+    seq = SequenceHelper.new("editing")
+    @highline.expects(:color).with("Choose entry", instance_of(Symbol)).returns("Choose entry")
+    seq << @highline.expects(:say).with('=== Choose entry ===')
+    seq << @highline.expects(:say).with("1. Twitter [tweeb]")
+    seq << @highline.expects(:say).with("2. Google [catfan]")
+    seq << @highline.expects(:say).with("3. Google [dogfan]")
+    seq << @highline.expects(:say).with("c. Cancel")
+    seq << @highline.expects(:ask).yields(mock { expects(:in=).with(%w{c 1 2 3}) }).returns('c')
+    assert_nil @console.choose_entry_to_edit(@top_group)
+  end
+
+  test "selecting username from edit menu" do
+    seq = sequence("editing")
+
+    @highline.expects(:color).with("Twitter [tweeb]", instance_of(Symbol)).
+      returns("Twitter [tweeb]")
+    @highline.expects(:say).with("Editing entry: Twitter [tweeb]").in_sequence(seq)
+    @highline.expects(:say).with("u. Change username").in_sequence(seq)
+    @highline.expects(:say).with("p. Change password").in_sequence(seq)
+    @highline.expects(:say).with("d. Delete").in_sequence(seq)
+    @highline.expects(:say).with("c. Cancel").in_sequence(seq)
+    @highline.expects(:ask).yields(mock {
+      expects(:in=).with(%w{u p d c})
+    }).returns('u').in_sequence(seq)
+
+    actual = @console.edit_entry(@twitter)
+    assert_equal :change_username, actual
+  end
+
+  test "selecting password from edit menu" do
+    seq = sequence("editing")
+
+    @highline.expects(:color).with("Twitter [tweeb]", instance_of(Symbol)).
+      returns("Twitter [tweeb]")
+    @highline.expects(:say).with("Editing entry: Twitter [tweeb]").in_sequence(seq)
+    @highline.expects(:say).with("u. Change username").in_sequence(seq)
+    @highline.expects(:say).with("p. Change password").in_sequence(seq)
+    @highline.expects(:say).with("d. Delete").in_sequence(seq)
+    @highline.expects(:say).with("c. Cancel").in_sequence(seq)
+    @highline.expects(:ask).yields(mock {
+      expects(:in=).with(%w{u p d c})
+    }).returns('p').in_sequence(seq)
+
+    actual = @console.edit_entry(@twitter)
+    assert_equal :change_password, actual
+  end
+
+  test "selecting delete from edit menu" do
+    seq = sequence("editing")
+
+    @highline.expects(:color).with("Twitter [tweeb]", instance_of(Symbol)).
+      returns("Twitter [tweeb]")
+    @highline.expects(:say).with("Editing entry: Twitter [tweeb]").in_sequence(seq)
+    @highline.expects(:say).with("u. Change username").in_sequence(seq)
+    @highline.expects(:say).with("p. Change password").in_sequence(seq)
+    @highline.expects(:say).with("d. Delete").in_sequence(seq)
+    @highline.expects(:say).with("c. Cancel").in_sequence(seq)
+    @highline.expects(:ask).yields(mock {
+      expects(:in=).with(%w{u p d c})
+    }).returns('d').in_sequence(seq)
+
+    actual = @console.edit_entry(@twitter)
+    assert_equal :delete, actual
+  end
+
+  test "cancel edit selection" do
+    seq = sequence("editing")
+
+    @highline.expects(:color).with("Twitter [tweeb]", instance_of(Symbol)).
+      returns("Twitter [tweeb]")
+    @highline.expects(:say).with("Editing entry: Twitter [tweeb]").in_sequence(seq)
+    @highline.expects(:say).with("u. Change username").in_sequence(seq)
+    @highline.expects(:say).with("p. Change password").in_sequence(seq)
+    @highline.expects(:say).with("d. Delete").in_sequence(seq)
+    @highline.expects(:say).with("c. Cancel").in_sequence(seq)
+    @highline.expects(:ask).yields(mock {
+      expects(:in=).with(%w{u p d c})
+    }).returns('c').in_sequence(seq)
+
+    assert_nil @console.edit_entry(@twitter)
+  end
+
+  test "changing username" do
+    seq = SequenceHelper.new("changing username")
+    @highline.expects(:color).with("username", instance_of(Symbol)).returns("username")
+    seq << @highline.expects(:say).with("Current username: username")
+    seq << @highline.expects(:ask).with("New username (blank to cancel): ").returns("foo")
+    assert_equal "foo", @console.change_username('username')
+  end
+
+  test "switching mode from menu" do
+    @console.mode = :copy
+
+    @highline.expects(:say).with("=== Keyrack Main Menu ===")
+    @highline.expects(:say).with("1. Twitter [tweeb]")
+    @highline.expects(:say).with("2. Google [catfan]")
+    @highline.expects(:say).with("3. Google [dogfan]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [m]ode [q]uit")
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 2 3 e g}) }).returns('m')
+    assert_nil @console.menu(:group => @top_group, :at_top => true)
+    assert_equal :print, @console.mode
+  end
+
+  test "confirm overwrite entry" do
+    @highline.expects(:color).with("Twitter [tweeb]", instance_of(Symbol)).
+      returns("Twitter [tweeb]")
+    @highline.expects(:agree).with("There's already an entry for: Twitter [tweeb]. Do you want to overwrite it? [yn] ").returns(true)
+    assert_equal true, @console.confirm_overwrite_entry(@twitter)
+  end
+
+  test "confirm delete entry" do
+    @highline.expects(:color).with("Twitter [tweeb]", instance_of(Symbol)).
+      returns("Twitter [tweeb]")
+    @highline.expects(:agree).with("You're about to delete Twitter [tweeb]. Are you sure? [yn] ").returns(true)
+    assert_equal true, @console.confirm_delete_entry(@twitter)
+  end
+
+  test "top command" do
+    foo_group = stub('Foo group', :name => 'Foo', :sites => [], :group_names => [])
+    @highline.stubs(:color).with('Foo', instance_of(Symbol)).returns('Foo')
+    @highline.expects(:say).with("=== Foo ===")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [g]roup [t]op [m]ode [q]uit")
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m g t}) }).returns('t')
+    assert_equal :top, @console.menu(:group => foo_group, :at_top => false)
+  end
+
+  test "up command" do
+    foo_group = stub('Foo group', :name => 'Foo', :sites => [], :group_names => [])
+    @highline.stubs(:color).with('Foo', instance_of(Symbol)).returns('Foo')
+    @highline.expects(:say).with("=== Foo ===")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [g]roup [u]p [t]op [m]ode [q]uit")
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m g u t}) }).returns('u')
+    assert_equal :up, @console.menu(:group => foo_group, :at_top => false, :enable_up => true)
+  end
+
+  test "invalid password" do
+    @highline.expects(:say).with("Invalid password.")
+    @console.display_invalid_password_notice
+  end
+
+  test "menu prints two columns" do
+    # ========== Keyrack Main Menu =========
+    # 1. Twitter [tweeb]  2. Google [catfan]
+    # 3. Google [dogfan]
+
+    HighLine::SystemExtensions.expects(:terminal_size).returns([40, 32])
+    seq = sequence('say')
+    @highline.expects(:say).with('========= Keyrack Main Menu =========')
+    @highline.expects(:say).with("1. Twitter [tweeb] ")
+    @highline.expects(:say).with("2. Google [catfan]")
+    @highline.expects(:say).with("3. Google [dogfan]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [m]ode [q]uit")
+
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 2 3 e g}) }).returns('q')
+    assert_equal :quit, @console.menu(:group => @top_group, :at_top => true)
+  end
+
+  test "menu prints three columns" do
+    HighLine::SystemExtensions.expects(:terminal_size).returns([80, 32])
+    @twitter.stubs(:usernames).returns(%w{foo})
+    seq = sequence('say')
+    @highline.expects(:say).with('============================ Keyrack Main Menu ============================')
+    @highline.expects(:say).with("1. Twitter [tweeb] ")
+    @highline.expects(:say).with("2. Google [catfan] ")
+    @highline.expects(:say).with("3. Google [dogfan]")
+    @highline.expects(:say).with("Mode: copy")
+    @highline.expects(:say).with("Commands: [n]ew [e]dit [g]roup [m]ode [q]uit")
+
+    @highline.expects(:ask).yields(mock { expects(:in=).with(%w{n q m 1 2 3 e g}) }).returns('q')
+    assert_equal :quit, @console.menu(:group => @top_group, :at_top => true)
+  end
+end