chore: refactor to support multi search
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
use crate::message::{Message, MessageHandler};
|
||||
use crate::peer::PeerMessage;
|
||||
use crate::types::FileSearchResult;
|
||||
use crate::types::SearchResult;
|
||||
use std::sync::mpsc::Sender;
|
||||
|
||||
pub struct FileSearchResponse;
|
||||
@@ -9,7 +9,7 @@ impl MessageHandler<PeerMessage> for FileSearchResponse {
|
||||
9
|
||||
}
|
||||
fn handle(&self, message: &mut Message, sender: Sender<PeerMessage>) {
|
||||
let file_search = match FileSearchResult::new_from_message(message) {
|
||||
let file_search = match SearchResult::new_from_message(message) {
|
||||
Ok(result) => result,
|
||||
Err(_) => return, // Skip malformed search results
|
||||
};
|
||||
@@ -43,7 +43,7 @@ fn test_new_from_message() {
|
||||
let mut message = Message::new_with_data(data);
|
||||
message.set_pointer(8);
|
||||
|
||||
let file_search = FileSearchResult::new_from_message(&mut message).unwrap();
|
||||
let file_search = SearchResult::new_from_message(&mut message).unwrap();
|
||||
assert_eq!(file_search.token, 882125677);
|
||||
assert_eq!(file_search.files.len(), 2);
|
||||
let file = &file_search.files[0];
|
||||
|
||||
Reference in New Issue
Block a user