如何使用 gem `memory_profiler` 编写单元测试？

Question

例如，我们有类实现：

class Memory
  def call
   2 * 2
  end
end

我们可以通过 memory_profiler 的使用情况得到报告：

require 'memory_profiler'
MemoryProfiler.report{ Memory.new.call  }.pretty_print

当我们增加内存或内存泄漏并更改为Memory#call时，如何实现单元测试失败？

例如，如果我们要这样更改Memory#call：

- 2 * 2
+ loop { 2 * 2 }

Answer 1

我不认为你应该把它作为单元测试，因为测试是一个有点不同的环境它不应该知道你的代码的时间成本或内存成本，它不应该验证指标，这是其他工具的工作.

对于您的情况，我建议您编写一些模块，您可以在其中断言分配的内存对您来说是可以的，例如您可以做一些事情

module MyMemsizeNotifier
  extend self
  ALLOWED_MEMSIZE = 0..4640

  # Or you can receive block, lambdas, procs
  # I will leave it for you implementation
  def call(klass) 
    report = MemoryProfiler.report
      klass.new.call
    end

    exceeds_limit?(report.total_allocated_memsize)
    report
  end

  def exceeds_limit?(total_usage)
    return if ALLOWED_MEMSIZE.include?(total_usage)

    # notify in rollbar, write a log to stdout or to file or any other way to notify
  end
end

## Usage

class MyController
  def index
    ...
    MyMemsizeNotifier.call(Memory)
    ...
  end
end