Add tests for validate function

Signed-off-by: Keshav Priyadarshi <[email protected]>

Author: keshav-space

src/lib.rs

@@ -7,12 +7,19 @@ use std::path::Path;
 
 static VALIDATOR: Lazy<Set<Mmap>> = Lazy::new(|| {
     let path = Path::new(env!("CARGO_MANIFEST_DIR")).join("purls.fst");
+    load_fst(&path)
+});
+
+fn load_fst(path: &Path) -> Set<Mmap> {
     let file = File::open(path).expect("Failed to open FST file");
     let mmap = unsafe { Mmap::map(&file).expect("Failed to mmap FST file") };
     Set::new(mmap).expect("Failed to load FST from mmap")
-});
+}
 
 pub fn validate(packageurl: &str) -> bool {
     let trimmed_packageurl = packageurl.trim_end_matches("/");
     VALIDATOR.contains(trimmed_packageurl)
 }
+
+#[cfg(test)]
+mod validate_tests;

src/validate_tests.rs

@@ -0,0 +1,11 @@
+use super::*;
+use std::path::Path;
+
+#[test]
+fn test_validate_with_custom_file() {
+    let test_path = Path::new(env!("CARGO_MANIFEST_DIR")).join("tests/data/test_purls.fst");
+    let validator = load_fst(&test_path);
+
+    assert!(validator.contains("pkg:nuget/FluentUtils.EnumExtensions"));
+    assert!(!validator.contains("pkg:example/nonexistent"));
+}