add tests

Author: vshakitskiy

compiler-cli/src/dependencies/dependency_manager.rs

@@ -23,6 +23,22 @@ use super::{
     remove_extra_packages, unlock_packages,
 };
 
+/// Verifies that all specified packages exist in the manifest.
+pub fn ensure_packages_exist_locally(manifest: &Manifest, packages: &[EcoString]) -> Result<()> {
+    let missing_packages: Vec<EcoString> = packages
+        .iter()
+        .filter(|package_name| !manifest.packages.iter().any(|p| &p.name == *package_name))
+        .cloned()
+        .collect();
+
+    if !missing_packages.is_empty() {
+        return Err(Error::PackagesToUpdateNotExist {
+            packages: missing_packages,
+        });
+    }
+    Ok(())
+}
+
 pub struct DependencyManagerConfig {
     // If `Yes` we read the manifest from disc. If not set then we ignore any
     // manifest which will result in the latest versions of the dependency
@@ -81,12 +97,12 @@ where
         if !paths.manifest().exists() {
             tracing::debug!("manifest_not_present");
             let manifest = self.perform_version_resolution(paths, config, None, Vec::new())?;
-            self.ensure_packages_exist_locally(&manifest, packages_to_update.clone())?;
+            ensure_packages_exist_locally(&manifest, &packages_to_update)?;
             return Ok(Resolved::all_added(manifest));
         }
 
         let existing_manifest = read_manifest_from_disc(paths)?;
-        self.ensure_packages_exist_locally(&existing_manifest, packages_to_update.clone())?;
+        ensure_packages_exist_locally(&existing_manifest, &packages_to_update)?;
 
         // If we have been asked not to use the manifest then
         let (requirements_changed, manifest_for_resolver) = match self.use_manifest {

compiler-cli/src/dependencies/tests.rs

@@ -1374,3 +1374,48 @@ fn test_pretty_print_version_updates() {
 
     insta::assert_snapshot!(output);
 }
+
+#[test]
+fn test_ensure_packages_exist_locally_all_present() {
+    let manifest = Manifest {
+        requirements: HashMap::new(),
+        packages: vec![
+            manifest_package("package_a", "1.0.0", vec![]),
+            manifest_package("package_b", "2.0.0", vec![]),
+            manifest_package("package_c", "3.0.0", vec![]),
+        ],
+    };
+
+    let packages_to_check = vec!["package_a".into(), "package_b".into()];
+    let result = dependency_manager::ensure_packages_exist_locally(&manifest, &packages_to_check);
+
+    assert!(result.is_ok(), "All packages exist, should return Ok");
+}
+
+#[test]
+fn test_ensure_packages_exist_locally_some_missing() {
+    let manifest = Manifest {
+        requirements: HashMap::new(),
+        packages: vec![
+            manifest_package("package_a", "1.0.0", vec![]),
+            manifest_package("package_b", "2.0.0", vec![]),
+        ],
+    };
+
+    let packages_to_check = vec![
+        "package_a".into(),
+        "package_b".into(),
+        "missing_package".into(),
+        "another_missing".into(),
+    ];
+    let result = dependency_manager::ensure_packages_exist_locally(&manifest, &packages_to_check);
+
+    match result {
+        Err(Error::PackagesToUpdateNotExist { packages }) => {
+            assert_eq!(packages.len(), 2);
+            assert!(packages.contains(&"missing_package".into()));
+            assert!(packages.contains(&"another_missing".into()));
+        }
+        _ => panic!("Expected PackagesToUpdateNotExist error"),
+    }
+}

compiler-core/src/error.rs

@@ -206,7 +206,7 @@ file_names.iter().map(|x| x.as_str()).join(", "))]
     RemovedPackagesNotExist { packages: Vec<String> },
 
     #[error("Packages to update not exist: {}", packages.iter().join(", "))]
-    PackagesToUpdateNotExist { packages: Vec<String> },
+    PackagesToUpdateNotExist { packages: Vec<EcoString> },
 
     #[error("unable to find project root")]
     UnableToFindProjectRoot { path: String },