用mocha测试rails，mock HTTPResponse

上周六去了shanghai on rails聚会，其中一节是 张元一介绍关于mocha在rails测试中的应用

今天刚好项目测试代码中用到需要mock http response，刚好用到，要感谢张元一兄台

以前也写过信用卡支付网关以及soap服务器通讯相关的测试，用的办法是override掉http的request和response

比如下面这个典型场景，一个信用卡网关，我需要发送一个request请求，包含了请求的xml，网关处理信息后返回结果xml

需要测试的类Demo，包含一个实例方法request_to_services，request到服务器，服务器后执行返回response：

ruby 代码

require 'uri'
require 'net/http'

class Demo
 def request_to_services
 uri = URI.parse("http://www.demo.com/services")
 request = Net::HTTP::Post.new(uri.path)
 request.body = "<xml>request data</xml>"
 http = Net::HTTP.new(uri.host, uri.port)
 response = http.request(request)
 raise unless response.is_a? Net::HTTPSuccess
 return response.body
 end
end

测试方法这样写：

ruby 代码

require 'demo'
require 'test/unit'

class TestDemo < Test::Unit::TestCase
 def test_request_to_services
 assert_equal "<xml>response data</xml>", Demo.new.request_to_services
 end
end

# 这里Override了ruby的HTTP request/response
class Net::HTTP
 def request(request, body = nil, &block)
 Response.new("<xml>response data</xml>")
 end

 class Response < Net::HTTPSuccess
 attr_accessor :body
 def initialize(content)
 @body = content
 end
 end
end

执行测试通过。我一直采用这个办法，非常简单，但是可以工作
这个方法有个缺点，Override的效果是全局的，也就是说在同一个时间只能模拟一种http的行为，缺乏灵活性
直到我碰到了mocha

ruby 代码

require 'demo'
require 'test/unit'
require 'rubygems'
require 'mocha'

class TestDemo < Test::Unit::TestCase
 def test_request_to_services
 http = stub(:request => stub(:is_a? => true, :body => "<xml>response data</xml>"))
 Net::HTTP.expects(:new).times(1).returns(http)

 assert_equal "<xml>response data</xml>", Demo.new.request_to_services
 end

 def test_request_to_services_response_http_exception
 http = stub(:request => stub(:is_a? => false, :body => ""))
 Net::HTTP.expects(:new).times(1).returns(http)

 assert_raise RuntimeError do
 Demo.new.request_to_services
 end
 end
end

用mocha来Mock对象非常灵活，下一步我将把所有的测试转到rspec上
借用hideto的说法：你应该试试 RSpec+ Mocha这道纯正的 墨西哥菜


TO BE CONTINUE!