File: test_results.rb

package info (click to toggle)
ruby-memory-profiler 1.1.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 284 kB
  • sloc: ruby: 1,128; makefile: 4
file content (85 lines) | stat: -rw-r--r-- 2,797 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
# frozen_string_literal: true

require_relative 'test_helper'

class TestResults < Minitest::Test
  def test_pretty_print_works_with_no_args
    assert_output(/^Total allocated/, '') { MemoryProfiler::Results.new.pretty_print }
  end

  def test_pretty_print_works_with_io_arg
    io = StringIO.new
    assert_silent { MemoryProfiler::Results.new.pretty_print(io) }
    assert_match(/^Total allocated/, io.string)
  end

  def test_no_conflict_with_pretty_print
    require 'pp'
    assert_output(/#<MemoryProfiler::Results:.*/) { pp(MemoryProfiler::Results.new) }
  end

  def scale_bytes_result
    MemoryProfiler.report { 1000.times { Array['a'..'z'][Array(0..25).sample] } }
  end

  def verify_unscaled_result(result, io)
    total_size = result.total_allocated_memsize
    array_size = result.allocated_memory_by_class.detect { |h| h[:data] == 'Array' }[:count]
    assert_match(/^Total allocated: #{total_size} bytes/, io.string, 'The total allocated memsize is unscaled.')
    assert_match(/^ +#{array_size}  Array$/, io.string, 'The allocated memsize for Array is unscaled.')
  end

  def test_scale_bytes_default
    result = scale_bytes_result
    io = StringIO.new
    result.pretty_print(io)
    verify_unscaled_result result, io
  end

  def test_scale_bytes_off
    result = scale_bytes_result
    io = StringIO.new
    result.pretty_print(io, scale_bytes: false)
    verify_unscaled_result result, io
  end

  def test_scale_bytes_true
    result = scale_bytes_result
    total_size = result.total_allocated_memsize / 1000.0
    array_size = result.allocated_memory_by_class.detect { |h| h[:data] == 'Array' }[:count] / 1000.0
    io = StringIO.new
    result.pretty_print(io, scale_bytes: true)

    assert_match(/^Total allocated: #{"%.2f" % total_size} kB/, io.string, 'The total allocated memsize is scaled.')
    assert_match(/^ +#{"%.2f" % array_size} kB  Array$/, io.string, 'The allocated memsize for Array is scaled.')
  end

  def normalized_paths_report
    MemoryProfiler.report do
      require_fixture_gem 'longhorn'
      Longhorn.run
    end
  end

  def test_normalize_paths_default
    report = normalized_paths_report
    io = StringIO.new
    report.pretty_print(io)
    assert_match(%r!fixtures/gems/longhorn-0.1.0/lib/longhorn.rb!, io.string)
  end

  def test_normalize_paths_false
    report = normalized_paths_report
    io = StringIO.new
    report.pretty_print(io, normalize_paths: false)
    assert_match(%r!fixtures/gems/longhorn-0.1.0/lib/longhorn.rb!, io.string)
  end

  def test_normalize_paths_true
    report = normalized_paths_report
    io = StringIO.new
    report.pretty_print(io, normalize_paths: true)
    assert_match(%r!\d+\s{2}longhorn-0.1.0/lib/longhorn.rb:\d+!, io.string)
    assert_match(%r!ruby/lib/\S*set.rb!, io.string)
  end
end