Add association function to urls

Signed-off-by: trivernis <trivernis@protonmail.com>

src/models/url.rs

@@ -58,4 +58,18 @@ impl Url {
     pub fn import(&mut self) -> UrlImportBuilder {
         UrlImportBuilder::new(self.client.clone(), &self.url)
     }
+
+    /// Associates the url with a list of file hashes
+    pub async fn associate(&mut self, hashes: Vec<String>) -> Result<()> {
+        self.client
+            .associate_urls(vec![self.url.clone()], hashes)
+            .await
+    }
+
+    /// Disassociates the url with a list of file hashes
+    pub async fn disassociate(&mut self, hashes: Vec<String>) -> Result<()> {
+        self.client
+            .disassociate_urls(vec![self.url.clone()], hashes)
+            .await
+    }
 }

tests/wrapper/test_url.rs

@@ -23,3 +23,25 @@ async fn it_imports() {
         .await
         .unwrap();
 }
+
+#[tokio::test]
+async fn it_associates() {
+    let mut url = get_url().await;
+
+    url.associate(vec![
+        "0000000000000000000000000000000000000000000000000000000000000000".to_string(),
+    ])
+    .await
+    .unwrap();
+}
+
+#[tokio::test]
+async fn it_disassociates() {
+    let mut url = get_url().await;
+
+    url.disassociate(vec![
+        "0000000000000000000000000000000000000000000000000000000000000000".to_string(),
+    ])
+    .await
+    .unwrap();
+}