@@ -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