Merge pull request #9 from oscar-project/dev-edit-blocklist

add mut getter on domains

src/blocklist_multi.rs

@@ -269,6 +269,10 @@ impl Blocklist {
             Some(detections)
         }
     }
+
+    pub fn domains_mut(&mut self) -> &mut HashMap<String, Vec<String>> {
+        &mut self.domains
+    }
 }
 
 #[cfg(test)]
@@ -278,8 +282,6 @@ mod tests {
         path::Path,
     };
 
-
-
     use super::Blocklist;
 
     // long test
@@ -293,6 +295,26 @@ mod tests {
         );
     }
 
+    #[test]
+    fn test_edit() {
+        let domains = vec![("blogspot.com".to_string(), vec!["blog".to_string()])]
+            .into_iter()
+            .collect();
+
+        let mut b = Blocklist::new(domains, HashMap::new());
+        assert_eq!(
+            b.detect("https://not_here.com/foo"),
+            None // Some(vec![&"blog".to_string()].into_iter().collect())
+        );
+
+        b.domains_mut()
+            .insert("not_here.com".to_string(), vec!["blog".to_string()]);
+        assert_eq!(
+            b.detect("https://not_here.com/foo"),
+            Some(vec![&"blog".to_string()].into_iter().collect())
+        );
+    }
+
     #[test]
     fn subdomains() {
         let domains = vec![