Skip to content

Commit e994aa2

Browse files
committed
refactor: [#1235] WhitelistManager tests
1 parent 2862c77 commit e994aa2

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

packages/tracker-core/src/whitelist/manager.rs

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ mod tests {
8282

8383
struct WhitelistManagerDeps {
8484
pub _database: Arc<Box<dyn Database>>,
85-
pub _database_whitelist: Arc<DatabaseWhitelist>,
85+
pub database_whitelist: Arc<DatabaseWhitelist>,
8686
pub in_memory_whitelist: Arc<InMemoryWhitelist>,
8787
}
8888

@@ -102,7 +102,7 @@ mod tests {
102102
whitelist_manager,
103103
Arc::new(WhitelistManagerDeps {
104104
_database: database,
105-
_database_whitelist: database_whitelist,
105+
database_whitelist,
106106
in_memory_whitelist,
107107
}),
108108
)
@@ -123,6 +123,7 @@ mod tests {
123123
whitelist_manager.add_torrent_to_whitelist(&info_hash).await.unwrap();
124124

125125
assert!(services.in_memory_whitelist.contains(&info_hash).await);
126+
assert!(services.database_whitelist.load_from_database().unwrap().contains(&info_hash));
126127
}
127128

128129
#[tokio::test]
@@ -136,6 +137,7 @@ mod tests {
136137
whitelist_manager.remove_torrent_from_whitelist(&info_hash).await.unwrap();
137138

138139
assert!(!services.in_memory_whitelist.contains(&info_hash).await);
140+
assert!(!services.database_whitelist.load_from_database().unwrap().contains(&info_hash));
139141
}
140142

141143
mod persistence {
@@ -148,11 +150,7 @@ mod tests {
148150

149151
let info_hash = sample_info_hash();
150152

151-
whitelist_manager.add_torrent_to_whitelist(&info_hash).await.unwrap();
152-
153-
services.in_memory_whitelist.remove(&info_hash).await;
154-
155-
assert!(!services.in_memory_whitelist.contains(&info_hash).await);
153+
services.database_whitelist.add(&info_hash).unwrap();
156154

157155
whitelist_manager.load_whitelist_from_database().await.unwrap();
158156

0 commit comments

Comments
 (0)