Add TextSpan::from_str

starkware-libs · Jan 10, 2025 · 3e41ca6 · 3e41ca6
1 parent fa395d4
commit 3e41ca6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/crates/cairo-lang-filesystem/src/span.rs b/crates/cairo-lang-filesystem/src/span.rs
@@ -111,6 +111,11 @@ pub struct TextSpan {
     pub end: TextOffset,
 }
 impl TextSpan {
+    /// Creates a `TextSpan` for the entirety of a given string.
+    #[allow(clippy::should_implement_trait)]
+    pub fn from_str(content: &str) -> Self {
+        Self { start: TextOffset::START, end: TextOffset::from_str(content) }
+    }
     pub fn width(self) -> TextWidth {
         self.end - self.start
     }

diff --git a/crates/cairo-lang-parser/src/db_test.rs b/crates/cairo-lang-parser/src/db_test.rs
@@ -95,7 +95,7 @@ fn test_token_stream_expr_parser() {
     let token_stream = MockTokenStream {
         tokens: vec![MockToken {
             content: expr_code.to_string(),
-            span: TextSpan { start: TextOffset::default(), end: TextOffset::from_str(expr_code) },
+            span: TextSpan::from_str(expr_code),
         }],
         content_string: expr_code.to_string(),
     };