module ActiveSupport::Testing::MethodCallAssertions

def assert_called_with(object, method_name, args = [], returns: nil)
  mock = Minitest::Mock.new
  if args.all? { |arg| arg.is_a?(Array) }
    args.each { |arg| mock.expect(:call, returns, arg) }
  else
    mock.expect(:call, returns, args)
  end
  object.stub(method_name, mock) { yield }
  mock.verify
end