RailsCasts11 Refactoring User Name Part 2 重构User name（二）

在上一篇我们讨论了重构，并演示了一个通过将代码移动到模型中以达到去除冗余的重构实例。不过这段代码依然还有优化的余地。

rubyclass User < ActiveRecord::Base  def full_name    name = first_name + ' '    name += "#{middle_initial}. " unless middle_initial.nil?    name += last_name  endend

User类在上一节结束时的状态。

引入测试

在着手进一步重构User之前，有必要了解一下测试。为了保证重构过程不对代码功能产生破坏，重构往往需要伴随着测试的介入。好的测试用例为重构质量保驾护航。通过rails generate生成模型类的同时，也会生成对应的测试类。与User类对应的是位于/test/unit目录¹的user_test.rb文件。

rubyrequire 'test_helper'class UserTest < ActiveSupport::TestCase  # Replace this with your real tests.  test "the truth" do    assert true  endend

Rails缺省生成的测试类。

由于是自动生成的，代码逻辑知识简单的用true == true做了空实现。要用我们自己编写的测试逻辑代替。创建一个测试用例：新建没有中间名的用户，检查程序的输出是否和我们期待的一样。

rubytest "full name without middle initial" do  user = User.new(:first_name => "John", :last_name => "Smith")  assert_equal 'John Smith', user.full_nameend

用于检测不含中间名用户的测试用例。

可以使用命令行rake test运行用例，但在这里我们准备使用自动测试工具AutoTest²。安装ZenTest gem(命令是sudo gem install ZenTest)。使用自动测试的好处是他会适时的执行测试，以便当程序工作不正常时立刻就能被我们发现。运行autotest查看返回结果

terminalLaa-Laa:ep11 eifion$ autotestloading autotest/rails/usr/local/bin/ruby -I.:lib:test -rtest/unit -e "%w[test/functional/users_controller_test.rb test/unit/user_test.rb].each { |f| require f }" | unit_diff -uLoaded suite -eStarted.Finished in 0.046945 seconds.1 tests, 1 assertions, 0 failures, 0 errors

AutoTest的输出

好消息！测试通过了。我们再往UsetTest类中添加一个用来检测包含中间名这种情况的用例。

rubytest "full name with middle initial" do  user = User.new(:first_name => "Paul", :middle_initial => "P", :last_name => "Hughes")  assert_equal 'Paul P. Hughes', user.full_nameend

当代码保存后，AutoTest会自动执行用例。现在这两个用例都应该是执行成功的。

重构开始

在保证full_name方法工作正常的基础上，我们准备开始重构了。局部变量name其实可以去掉；字符串连接也可以被别的方法代替。取而代之的是将名字的每一部分分别放到数组中，调用数组的join方法将他们用空格连接。

rubyclass User < ActiveRecord::Base  def full_name    [first_name, middle_initial, last_name].join(' ')  endend

对User类的第一次重构。

保存代码后，自动测试的结果显示两个用例全都执行失败了。

terminal1) Failure:  test_full_name_with_middle_initial(UserTest)  [./test/unit/user_test.rb:11:in `test_full_name_with_middle_initial'  /usr/local/lib/ruby/gems/1.8/gems/activesupport-2.2.2/lib/active_support/testing/setup_and_teardown.rb:60:in `run']:  --- /var/folders/yD/yDkhXjIsHAqkCTKsBbUlC++++TI/-Tmp-/expect31666.0    2009-01-04 11:23:35.000000000 +0000  +++ /var/folders/yD/yDkhXjIsHAqkCTKsBbUlC++++TI/-Tmp-/butwas31666.0    2009-01-04 11:23:35.000000000 +0000  @@ -1 +1 @@  -Paul P. Hughes  +Paul P Hughes  2) Failure:  test_full_name_without_middle_initial(UserTest)  [./test/unit/user_test.rb:6:in `test_full_name_without_middle_initial'  /usr/local/lib/ruby/gems/1.8/gems/activesupport-2.2.2/lib/active_support/testing/setup_and_teardown.rb:60:in `run']:  --- /var/folders/yD/yDkhXjIsHAqkCTKsBbUlC++++TI/-Tmp-/expect31666.1    2009-01-04 11:23:35.000000000 +0000  +++ /var/folders/yD/yDkhXjIsHAqkCTKsBbUlC++++TI/-Tmp-/butwas31666.1    2009-01-04 11:23:35.000000000 +0000  @@ -1 +1 @@  -John Smith  +John  Smith  2 tests, 2 assertions, 2 failures, 0 errors

AutoTest输出显示两个用例执行失败。

第一个用例失败的原因是，输出的中间名后面没有带上点号。第二个失败的原因是，当没有中间名时，名和姓之间放入了两个空格。先来改正第二个错误，这个错误产生的原因是无论是否有中间名，都会将各个名字用空格相连。修改这个问题，是在调用join方法之前调用compact方法。(compact方法会从列表中移除掉所有的nil值。)

rubydef full_name  [first_name, middle_initial, last_name].compact.join(' ')end

重新执行用例后，只剩第一个问题了。我们重建一个叫middle_initial_with_full_stop的方法，根据是否有中间名，返回适当的字符串。

rubydef full_name  [first_name, middle_initial_with_full_stop, last_name].compact.join(' ')end  def middle_initial_with_full_stop  "#{middle_initial}." unless middle_initial.blank?end

修改之后，两个用例都能跑过。但仍然有一种情况漏测了。那就是如果输入的中间名是一个空字符串的时候。写一个用例覆盖一下。

rubytest "full name with empty middle initial" do  user = User.new(:first_name => "John", :middle_initial => "", :last_name => "Jones")  assert_equal 'John Jones', user.full_nameend

也能通过，结果正确。原因是blank?方法对于空字符串和nil都会返回true，所以可以在调用compact方法时被移除。

这样一来User类的代码比以前更可读了。不过现在测试用例的代码有点重复了，关于测试用例的优化，下回分解。

重构后的状态

rubyclass User < ActiveRecord::Base  def full_name    [first_name, middle_initial_with_full_stop, last_name].compact.join(' ')  end    def middle_initial_with_full_stop    "#{middle_initial}." unless middle_initial.blank?  endend

重构后的User类代码。

rubyrequire 'test_helper'  class UserTest < ActiveSupport::TestCase    test "full name without middle initial" do      user = User.new(:first_name => "John", :last_name => "Smith")      assert_equal 'John Smith', user.full_name    end    test "full name with middle initial" do      user = User.new(:first_name => "Paul", :middle_initial => "P", :last_name => "Hughes")      assert_equal 'Paul P. Hughes', user.full_name    end    test "full name with empty middle initial" do      user = User.new(:first_name => "John", :middle_initial => "", :last_name => "Jones")      assert_equal 'John Jones', user.full_name    end  end

与User类对应的测试用例代码。

注

1. 视频是基于Rails 1录制的。代码使用Rails 2.2编写。
2. http://rubyforge.org/projects/zentest/

转自：http://railscasts.com/episodes/11-refactoring-user-name-part-2?language=zh&view=asciicast