diff --git a/lib/telebugs/middleware/root_directory_filter.rb b/lib/telebugs/middleware/root_directory_filter.rb index 4711a75..b0cc502 100644 --- a/lib/telebugs/middleware/root_directory_filter.rb +++ b/lib/telebugs/middleware/root_directory_filter.rb @@ -14,6 +14,7 @@ def call(report) next unless (file = frame[:file]) next unless file.start_with?(@root_directory) + frame[:root_dir] = true file.sub!(/#{@root_directory}\/?/, "") end end diff --git a/test/middleware/test_root_directory_filter.rb b/test/middleware/test_root_directory_filter.rb index 9533032..c67ae9b 100644 --- a/test/middleware/test_root_directory_filter.rb +++ b/test/middleware/test_root_directory_filter.rb @@ -17,6 +17,8 @@ def test_root_directory_filter report = Telebugs::Report.new(e) Telebugs::Middleware::RootDirectoryFilter.new(root_directory).call(report) - assert_equal "app/models/user.rb", report.data[:errors][0][:backtrace][0][:file] + frame = report.data[:errors][0][:backtrace][0] + assert_equal "app/models/user.rb", frame[:file] + assert frame[:root_dir] end end