# File: friend_list_tests.rb
require 'friend_list'
require 'test/unit'

class TestFriendList < Test::Unit::TestCase
  def test_return
    list = FriendList.new

    assert_equal('Paulo Soreto', list.add('Paulo Soreto'))
    assert_equal('Pedro Josefino', list.add('Pedro Josefino'))
    assert_equal([], list.clear)
  end

  def test_methods
    list = FriendList.new

    list.add('Paulo Soreto')
    assert_equal(['Paulo Soreto'], list)

    list.add('Pedro Josefino')
    assert_equal(['Paulo Soreto', 'Pedro Josefino'], list)

    list.remove('Paulo Soreto')
    assert_equal(['Pedro Josefino'], list)

    list.clear
    assert_equal([], list)
  end
end