如何在Ruby中存根请求/响应

时间:2020-04-10 03:05:09

标签: ruby-on-rails ruby

使用RSpec的模拟/存根,如何为find_by_id方法编写单元测试?

我想使用RSpec,而不是WebMock或VCR。如何为请求/响应创建存根?

class RapidApiClient
  HOST_URL = 'https://brianiswu-open-brewery-db-v1.p.rapidapi.com/breweries'
  API_KEY  = 'private_api_key'

   def request_api(url)
    Excon.get(
      url,
      headers: {
        'X-RapidAPI-Host' => HOST_URL,
        'X-RapidAPI-Key' => 'API_KEY'
      }
    )
  end

 def find_by_id(id)
    response = request_api("#{HOST_URL}/#{id}")
    return nil if response.status != 200
    JSON.parse(response.body)
 end  
end

响应为:

[
{"id":4 , "name":"Ban Brewing Company" , "brewery_type":"micro", "city":"Tulsa" , "state":"OK"}
{"id":44,"name":"Tab Brewing" "brewery_type":"micro", "city":"Birmingham", "state":"MO"}
]

1 个答案:

答案 0 :(得分:1)

您可以对request_api方法本身进行存根,这样就不必使用类似以下的方式发出HTTP请求:

expect_any_instance_of(RapidApiClient)
    .to receive(:request_api)
    .and_return([
                  {"id":4 , "name":"Ban Brewing Company" , "brewery_type":"micro", "city":"Tulsa" , "state":"OK"}
                  {"id":44,"name":"Tab Brewing" "brewery_type":"micro", "city":"Birmingham", "state":"MO"}
                ])

如果要对实际请求/响应进行存根处理。

您可以像这样子化get库的Excon方法:

expect(Excon)
.to receive(:get)
.and_return(Excon::Response.new(
     :status => 200,
     :body => '[{"id":4 , "name":"Ban Brewing Company","brewery_type":"micro", "city":"Tulsa" , "state":"OK"},{"id":44,"name":"Tab Brewing" "brewery_type":"micro", "city":"Birmingham", "state":"MO"}]'
    ))