#!/usr/bin/env ruby # encoding: UTF-8 require File.expand_path('../test_helper', __FILE__) require 'fileutils' require 'stringio' require 'tmpdir' require_relative 'prime' # -- Tests ---- class PrintersTest < TestCase def setup # WALL_TIME so we can use sleep in our test and get same measurements on linux and windows RubyProf::measure_mode = RubyProf::WALL_TIME @result = RubyProf.profile do run_primes(1000, 5000) end end def test_printers output = ENV['SHOW_RUBY_PROF_PRINTER_OUTPUT'] == "1" ? STDOUT : StringIO.new('') printer = RubyProf::CallStackPrinter.new(@result) printer.print(output) printer = RubyProf::CallTreePrinter.new(@result) printer.print() printer = RubyProf::FlatPrinter.new(@result) printer.print(output) printer = RubyProf::GraphHtmlPrinter.new(@result) printer.print(output) printer = RubyProf::GraphPrinter.new(@result) printer.print(output) end def test_print_to_files output_dir = 'tmp/examples2' if ENV['SAVE_NEW_PRINTER_EXAMPLES'] output_dir = 'tmp/examples' end FileUtils.mkdir_p output_dir printer = RubyProf::DotPrinter.new(@result) File.open("#{output_dir}/graph.dot", "w") {|f| printer.print(f)} printer = RubyProf::CallStackPrinter.new(@result) File.open("#{output_dir}/stack.html", "w") {|f| printer.print(f, :application => "primes")} # printer = RubyProf::MultiPrinter.new(@result) # printer.print(:path => "#{output_dir}", :profile => "multi", :application => "primes") # for file in ['graph.dot', 'multi.flat.txt', 'multi.graph.html', "multi.callgrind.out.#{$$}", 'multi.stack.html', 'stack.html'] # existant_file = output_dir + '/' + file # assert File.size(existant_file) > 0 # end end def test_refuses_io_objects p = RubyProf::MultiPrinter.new(@result) begin p.print(STDOUT) flunk "should have raised an ArgumentError" rescue ArgumentError => e assert_match(/IO/, e.to_s) end end def test_refuses_non_hashes p = RubyProf::MultiPrinter.new (@result) begin p.print([]) flunk "should have raised an ArgumentError" rescue ArgumentError => e assert_match(/hash/, e.to_s) end end def test_flat_string output = helper_test_flat_string(RubyProf::FlatPrinter) assert_match(/prime.rb/, output) end def helper_test_flat_string(klass) output = '' printer = klass.new(@result) printer.print(output) assert_match(/Thread ID: -?\d+/i, output) assert_match(/Fiber ID: -?\d+/i, output) assert_match(/Total: \d+\.\d+/i, output) assert_match(/Object#run_primes/i, output) output end def test_graph_html_string output = '' printer = RubyProf::GraphHtmlPrinter.new(@result) printer.print(output) assert_match(//i, output) assert_match( %r{Total}i, output) assert_match(/Object#run_primes/i, output) end def test_graph_string output = '' printer = RubyProf::GraphPrinter.new(@result) printer.print(output) assert_match(/Thread ID: -?\d+/i, output) assert_match(/Fiber ID: -?\d+/i, output) assert_match(/Total Time: \d+\.\d+/i, output) assert_match(/Object#run_primes/i, output) end def do_nothing start = Time.now while(Time.now == start) end end def test_all_with_small_percentiles result = RubyProf.profile do sleep 2 do_nothing end # RubyProf::CallTreePrinter doesn't "do" a min_percent # RubyProf::FlatPrinter only outputs if self time > percent... for klass in [RubyProf::GraphPrinter, RubyProf::GraphHtmlPrinter] printer = klass.new(result) out = '' printer.print(out, :min_percent => 0.00000001) assert_match(/do_nothing/, out) end end end