add tests for rust Repository

Author: rbran

rust/src/repository.rs

@@ -80,7 +80,7 @@ impl RepositoryManager {
     /// Before you can query plugin metadata from a repository, you need to call [RepositoryManager::check_for_updates].
     ///
     /// * `url` - URL to the plugins.json containing the records for this repository
-    /// * `repopath` - path to where the repository will be stored on disk locally
+    /// * `repository_path` - path to where the repository will be stored on disk locally
     ///
     /// Returns true if the repository was successfully added, false otherwise.
     pub fn add_repository<U: BnStrCompatible, P: BnStrCompatible>(
@@ -99,15 +99,15 @@ impl RepositoryManager {
         }
     }
 
-    pub fn repository_by_path<P: BnStrCompatible>(&self, path: P) -> Repository {
+    pub fn repository_by_path<P: BnStrCompatible>(&self, path: P) -> Option<Repository> {
         let path = path.into_bytes_with_nul();
         let result = unsafe {
             BNRepositoryGetRepositoryByPath(
                 self.as_raw(),
                 path.as_ref().as_ptr() as *const ffi::c_char,
             )
         };
-        unsafe { Repository::from_raw(ptr::NonNull::new(result).unwrap()) }
+        ptr::NonNull::new(result).map(|raw| unsafe { Repository::from_raw(raw) })
     }
 
     /// Gets the default Repository

rust/tests/repository.rs

@@ -0,0 +1,30 @@
+use binaryninja::headless::Session;
+use binaryninja::repository::RepositoryManager;
+use rstest::*;
+
+#[fixture]
+fn session() -> Session {
+    Session::new().expect("Failed to initialize session")
+}
+
+#[rstest]
+fn test_list(_session: Session) {
+    let manager = RepositoryManager::default();
+    let repositories = manager.repositories();
+    for repository in &repositories {
+        let repo_path = repository.path();
+        let repository_by_path = manager.repository_by_path(repo_path).unwrap();
+        assert_eq!(repository.url(), repository_by_path.url());
+    }
+
+    let repository = manager.default_repository();
+    let _full_path = repository.full_path();
+    let _path = repository.path();
+    let _url = repository.url();
+    let plugins = repository.plugins();
+    for plugin in &plugins {
+        let plugin_path = plugin.path();
+        let plugin_by_path = repository.plugin_by_path(plugin_path).unwrap();
+        assert_eq!(plugin.package_url(), plugin_by_path.package_url());
+    }
+}