Added DiagnosticBuilder::init

commit-id:bf94039d
starkware-libs · Feb 12, 2024 · 98092cb · 98092cb
1 parent b8f0eef
commit 98092cb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/crates/cairo-lang-diagnostics/src/diagnostics.rs b/crates/cairo-lang-diagnostics/src/diagnostics.rs
@@ -160,6 +160,12 @@ pub struct DiagnosticsBuilder<TEntry: DiagnosticEntry> {
     pub subtrees: Vec<Diagnostics<TEntry>>,
 }
 impl<TEntry: DiagnosticEntry> DiagnosticsBuilder<TEntry> {
+    /// Creates a new builder with the given diagnostics.
+    pub fn init(diagnostics: Diagnostics<TEntry>) -> Self {
+        let mut new_self = Self::default();
+        new_self.extend(diagnostics);
+        new_self
+    }
     pub fn add(&mut self, diagnostic: TEntry) -> DiagnosticAdded {
         if diagnostic.severity() == Severity::Error {
             self.error_count += 1;

diff --git a/crates/cairo-lang-semantic/src/test_utils.rs b/crates/cairo-lang-semantic/src/test_utils.rs
@@ -322,8 +322,8 @@ fn get_recursive_module_semantic_diagnostics(
     db: &dyn SemanticGroup,
     module_id: ModuleId,
 ) -> Diagnostics<SemanticDiagnostic> {
-    let mut diagnostics = DiagnosticsBuilder::default();
-    diagnostics.extend(db.module_semantic_diagnostics(module_id).unwrap());
+    let mut diagnostics =
+        DiagnosticsBuilder::init(db.module_semantic_diagnostics(module_id).unwrap());
     for submodule_id in db.module_submodules_ids(module_id).unwrap().iter() {
         if is_submodule_inline(db, *submodule_id) {
             diagnostics.extend(get_recursive_module_semantic_diagnostics(