ruby-on-rails - 存根或模拟实例方法 rspec

Question

以下函数是一个控制器操作，用于从 AWS S3 存储桶获取要提供的数据。

def show
    s3_response = Zlib::GzipReader.new(ApiBucket.bucket.object(id).get.body).read
  render json: s3_response
end

我正在为这种方法编写规范。我需要存根/模拟这样s3_response实际上并不向 s3 请求。这是我尝试过的。但这不起作用。请帮忙。

describe '#GET show' do
  let!(:resource) { create(:resource) }

  before do
    json_data = Api::V2::Presenter.consume_as_json(resource)
    allow_any_instance_of(Zlib::GzipReader).to receive(:read).and_return(json_data)
  end

  it 'should return the resource in page format' do
    get :show, format: :json, params: { id: resource.uuid }
    response_body_json = JSON.parse(response.body)
    json_data = Api::V2::Presenter.consume_as_json(
        Api::V2::ResourcePresenter.new(resource).page,
        true
      )
    expect(response_body_json).to eql(JSON.parse(json_data))
  end
end

我收到此错误Zlib::GzipFile::Error: not in gzip format

标签： ruby-on-railsrubyamazon-s3rspecrspec-rails

Answer 1

而不是存根Zlib::GzipReader或S3 Bucket. 处理这种情况的一种简单易行的方法是在控制器中创建一个新的私有函数，然后存根控制器函数。

在控制器端：

def show
  render json: s3_response
end

private:

def s3_response
   Zlib::GzipReader.new(ApiBucket.bucket.object(id).get.body).read
end

规格将是：

   describe '#GET show' do
  let!(:resource) { create(:resource) }

  before do
    json_data = Api::V2::Presenter.consume_as_json(resource)
    ResourceController.any_instance.stub(:s3_response).and_return(json_data)
  end

  it 'should return the resource in page format' do
    get :show, format: :json, params: { id: resource.uuid }
    response_body_json = JSON.parse(response.body)
    json_data = Api::V2::Presenter.consume_as_json(
        Api::V2::Presenter.new(resource).page,
        true
      )
    expect(response_body_json).to eql(JSON.parse(json_data))
  end
end