继续重构。
我们先来写User类的单元测试,定义3个测试方法:
[code]
require File.dirname(__FILE__) + '/../test_helper'
class UserTest < Test::Unit::TestCase
fixtures :users
def test_full_name_without_middle_initial
user = User.new(:first_name => 'John', :last_name => 'Doe')
assert_equal 'John Doe', user.full_name
end
def test_full_name_with_middle_initial
user = User.new(:first_name => 'John', :middle_initial => 'H', :last_name => 'Doe')
assert_equal 'John H. Doe', user.full_name
end
def test_full_name_with_blank_middle_initial
user = User.new(:first_name => 'John', :middle_initial => '', :last_name => 'Doe')
assert_equal 'John Doe', user.full_name
end
end
[/code]
我们再看看我们的full_name方法,显然是无法满足所有的测试的:
[code]
class User < ActiveRecord::Base
def full_name
name = first_name
name += "#{middle_initial}." unless middle_initial.nil?
name += last_name
name
end
end
[/code]
而且middle_initial后面跟一个"."也让我们的代码"风水"不好,比如我们突然有一天把"."改成"-"呢?好,我们继续重构吧:
[code]
class User < ActiveRecord::Base
def full_name
[first_name, middle_initial_with_period, last_name].compact.join(' ')
end
def middle_initial_with_period
"#{middle_initial}." unless middle_initial.blank?
end
end
[/code]
运行测试,ok,通过了!
我们先来写User类的单元测试,定义3个测试方法:
[code]
require File.dirname(__FILE__) + '/../test_helper'
class UserTest < Test::Unit::TestCase
fixtures :users
def test_full_name_without_middle_initial
user = User.new(:first_name => 'John', :last_name => 'Doe')
assert_equal 'John Doe', user.full_name
end
def test_full_name_with_middle_initial
user = User.new(:first_name => 'John', :middle_initial => 'H', :last_name => 'Doe')
assert_equal 'John H. Doe', user.full_name
end
def test_full_name_with_blank_middle_initial
user = User.new(:first_name => 'John', :middle_initial => '', :last_name => 'Doe')
assert_equal 'John Doe', user.full_name
end
end
[/code]
我们再看看我们的full_name方法,显然是无法满足所有的测试的:
[code]
class User < ActiveRecord::Base
def full_name
name = first_name
name += "#{middle_initial}." unless middle_initial.nil?
name += last_name
name
end
end
[/code]
而且middle_initial后面跟一个"."也让我们的代码"风水"不好,比如我们突然有一天把"."改成"-"呢?好,我们继续重构吧:
[code]
class User < ActiveRecord::Base
def full_name
[first_name, middle_initial_with_period, last_name].compact.join(' ')
end
def middle_initial_with_period
"#{middle_initial}." unless middle_initial.blank?
end
end
[/code]
运行测试,ok,通过了!