dearsky/numa
1
0
Fork 0
Code Issues 10 Pull Requests 1 Actions Packages Projects Releases 25 Wiki Activity

style: rustfmt blocklist tests

Browse Source 
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Razvan Dimescu
2026-04-10 21:15:24 +03:00
parent d66a88f467
commit c452f99a45
1 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/blocklist.rs
View File

@@ -95,12 +95,20 @@ impl BlocklistStore {
} }
if let Some(matched) = Self::find_in_set(&domain, &self.allowlist) { if let Some(matched) = Self::find_in_set(&domain, &self.allowlist) {
let reason = if matched == domain { "exact match in allowlist" } else { "parent domain in allowlist" }; let reason = if matched == domain {
"exact match in allowlist"
} else {
"parent domain in allowlist"
};
return BlockCheckResult::allowed(matched, reason); return BlockCheckResult::allowed(matched, reason);
} }
if let Some(matched) = Self::find_in_set(&domain, &self.domains) { if let Some(matched) = Self::find_in_set(&domain, &self.domains) {
let reason = if matched == domain { "exact match in blocklist" } else { "parent domain in blocklist" }; let reason = if matched == domain {
"exact match in blocklist"
} else {
"parent domain in blocklist"
};
return BlockCheckResult::blocked(matched, reason); return BlockCheckResult::blocked(matched, reason);
} }
@@ -229,10 +237,7 @@ mod tests {
fn store_with(domains: &[&str], allowlist: &[&str]) -> BlocklistStore { fn store_with(domains: &[&str], allowlist: &[&str]) -> BlocklistStore {
let mut store = BlocklistStore::new(); let mut store = BlocklistStore::new();
store.swap_domains( store.swap_domains(domains.iter().map(|s| s.to_string()).collect(), vec![]);
domains.iter().map(|s| s.to_string()).collect(),
vec![],
);
for d in allowlist { for d in allowlist {
store.add_to_allowlist(d); store.add_to_allowlist(d);
} }
@@ -262,10 +267,7 @@ mod tests {
#[test] #[test]
fn parent_allowlist_unblocks_subdomain() { fn parent_allowlist_unblocks_subdomain() {
let store = store_with( let store = store_with(&["example.com", "www.example.com"], &["example.com"]);
&["example.com", "www.example.com"],
&["example.com"],
);
assert!(!store.is_blocked("example.com")); assert!(!store.is_blocked("example.com"));
assert!(!store.is_blocked("www.example.com")); assert!(!store.is_blocked("www.example.com"));
assert!(!store.is_blocked("sub.deep.example.com")); assert!(!store.is_blocked("sub.deep.example.com"));