diff --git a/lib/parallel_tests/test/runtime_logger.rb b/lib/parallel_tests/test/runtime_logger.rb index 095208d2..c68ba074 100644 --- a/lib/parallel_tests/test/runtime_logger.rb +++ b/lib/parallel_tests/test/runtime_logger.rb @@ -62,6 +62,9 @@ def log(test, time) def message(test, delta) return unless method = test.public_instance_methods(true).detect { |m| m =~ /^test_/ } filename = test.instance_method(method).source_location.first.sub("#{Dir.pwd}/", "") + unless filename.include?('_test.rb') + filename = Module.const_source_location(test.to_s).first.sub("#{Dir.pwd}/", "") + end "#{filename}:#{delta}" end