如何使用Ruby模拟类?

时间:2012-03-06 14:17:56

标签: ruby class testing mocking minitest

我正在使用minitest / mock并且想要模拟一个类。我不是试图测试模型类本身,而是试图测试服务(SomeService)与模型(SomeModel)的交互。

我想出了这个(Hack :: ClassDelegate),但我不相信这是个好主意:

require 'minitest/autorun'
require 'minitest/mock'

module Hack
  class ClassDelegate
    def self.set_delegate(delegate); @@delegate = delegate; end
    def self.method_missing(sym, *args, &block)
      @@delegate.method_missing(sym, *args, &block)
    end 
  end
end

class TestClassDelegation < MiniTest::Unit::TestCase

  class SomeModel < Hack::ClassDelegate ; end

  class SomeService
    def delete(id)
      SomeModel.delete(id)
    end
  end

  def test_delegation
    id = '123456789'
    mock = MiniTest::Mock.new
    mock.expect(:delete, nil, [id])

    SomeModel.set_delegate(mock)
    service = SomeService.new
    service.delete(id)

    assert mock.verify
  end
end

我很确定无论如何嘲笑一个类都不是一个好主意,但我有一个遗留系统,我需要编写一些测试,我不想更改系统,直到我包装了一些测试在它周围。

2 个答案:

答案 0 :(得分:5)

我认为这有点复杂。那怎么样:

mock = MiniTest::Mock.new
SomeService.send(:const_set, :SomeModel, mock)
mock.expect(:delete, nil, [1])
service = SomeService.new
service.delete(1)
mock.verify
SomeService.send(:remove_const, :SomeModel)

答案 1 :(得分:2)

在遇到同样的问题并思考了很长一段时间之后,我发现暂时改变类常数可能是最好的方法(就像Elliot在他的回答中所说的那样)。

但是,我发现了一种更好的方法:https://github.com/adammck/minitest-stub-const

使用这个gem,你可以这样编写你的测试:

def test_delegation
  id = '123456789'
  mock = MiniTest::Mock.new
  mock.expect(:delete, nil, [id])

  SomeService.stub_const 'SomeModel', mock do
    service = SomeService.new
    service.delete(id)
  end

  assert mock.verify
end