From 506b64e2cdcc2e0a7762bd6e4985cd4ab98b1e3b Mon Sep 17 00:00:00 2001
From: Tobias Schmitz <tobiasschmitz2001@gmail.com>
Date: Tue, 20 May 2025 15:01:33 +0200
Subject: [PATCH] test: also use TestWorld::lookup in text_for_range

---
 tests/src/run.rs | 18 +++++-------------
 1 file changed, 5 insertions(+), 13 deletions(-)

diff --git a/tests/src/run.rs b/tests/src/run.rs
index f2e83e067..a34e38db5 100644
--- a/tests/src/run.rs
+++ b/tests/src/run.rs
@@ -8,7 +8,7 @@ use typst::diag::{SourceDiagnostic, Warned};
 use typst::html::HtmlDocument;
 use typst::layout::{Abs, Frame, FrameItem, PagedDocument, Transform};
 use typst::visualize::Color;
-use typst::{Document, World, WorldExt};
+use typst::{Document, WorldExt};
 use typst_pdf::PdfOptions;
 use typst_syntax::FileId;
 
@@ -289,19 +289,11 @@ impl<'a> Runner<'a> {
     fn text_for_range(&self, file: FileId, range: &Option<Range<usize>>) -> String {
         let Some(range) = range else { return "No text".into() };
         if range.is_empty() {
-            "(empty)".into()
-        } else if file == self.test.source.id() {
-            format!(
-                "`{}`",
-                self.test.source.text()[range.clone()]
-                    .replace('\n', "\\n")
-                    .replace('\r', "\\r")
-            )
-        } else {
-            let bytes = self.world.file(file).unwrap();
-            let text = std::str::from_utf8(&bytes).unwrap();
-            format!("`{}`", text[range.clone()].replace('\n', "\\n").replace('\r', "\\r"))
+            return "(empty)".into();
         }
+
+        let lines = self.world.lookup(file);
+        lines.text()[range.clone()].replace('\n', "\\n").replace('\r', "\\r")
     }
 
     /// Display a byte range as a line:column range.