unmemftp 0.1.0

A quick and somewhat dirty library for testing read-only ftp clients, based on libunftp
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use async_ftp::FtpStream;
use std::result::Result;
use unmemftp::{ MemStorage, MemFile, serve };

fn somefiles() -> MemStorage {
	MemStorage::new(vec![
		("/a", MemFile::from_str("A")),
		("/b/c", MemFile::from_str("BC")), // Containing folders are automatically created - no support for empty
	])
}

mod serve {
	use super::*;

	#[tokio::test(threaded_scheduler)]
	pub async fn accessdirect() {
		let addr = serve(Box::new(somefiles)).await;

		let ftp = &mut default_ftp_client(addr).await;
		retr_test(ftp, "/a", "A").await;
	}

	#[tokio::test(threaded_scheduler)]
	pub async fn accesspath_no_threaded_scheduler() {
		let addr = serve(Box::new(somefiles)).await;

		let ftp = &mut default_ftp_client(addr).await;
		retr_test(ftp, "/b/c", "BC").await;
	}

	#[tokio::test(threaded_scheduler)]
	pub async fn cd_and_get() {
		let addr = serve(Box::new(somefiles)).await;

		let ftp = &mut default_ftp_client(addr).await;
		ftp.cwd("b").await.unwrap();
		retr_test(ftp, "c", "BC").await;
	}
	
	#[tokio::test(threaded_scheduler)]
	pub async fn cannot_cd_to_nowhere() {
		let addr = serve(Box::new(somefiles)).await;

		let ftp = &mut default_ftp_client(addr).await;
		assert!(ftp.cwd("d").await.is_err());
	}
	
	#[tokio::test(threaded_scheduler)]
	pub async fn cannot_get_nowhere() {
		let addr = serve(Box::new(somefiles)).await;

		let ftp = &mut default_ftp_client(addr).await;
		assert!(ftp.simple_retr("d").await.is_err());
	}

	async fn retr_test(ftp: &mut FtpStream, path: &str, expected: &str) {
        use async_ftp::{DataStream, FtpError};
        use tokio::io::{AsyncReadExt, BufReader};
		async fn receive(mut reader: BufReader<DataStream>) -> Result<String, FtpError> {
            let mut buffer = Vec::new();
            reader
                .read_to_end(&mut buffer)
                .await
                .map_err(FtpError::ConnectionError)?;
		    return Ok(std::str::from_utf8(&buffer).unwrap().to_owned());
        }
        assert_eq!(ftp.retr(path, receive).await.unwrap(), expected);
	}
}

mod acro {
	use super::*;
	use std::sync::Arc;
	use libunftp::auth::{ Authenticator, DefaultUser };
	use std::error::Error;

	struct BobAuthenticator {}
	#[async_trait::async_trait]
	impl Authenticator<DefaultUser> for BobAuthenticator {
		async fn authenticate(&self, username: &str, password: &str) -> Result<DefaultUser, Box<dyn Error + Send + Sync>> {
			if username == "bob" && password == "bob" {
				Ok(DefaultUser {})
			} else {
				Err("bye!".into())
			}
		}
	}

	// For advanced configurations, start unftp manually
	#[tokio::test(threaded_scheduler)]
	#[should_panic]
	pub async fn youre_not_bob() {
		let addr = "127.0.0.1:46858";
		let server = libunftp::Server::new(Box::new(somefiles))
			.greeting("irrelevant")
			.authenticator(Arc::new(BobAuthenticator {}))
			.passive_ports(50000..65535);
		tokio::spawn(server.listen(addr));
		unmemftp::connectable(addr.parse().unwrap()).await;

		default_ftp_client(addr.parse().unwrap()).await;
	}
	
	#[tokio::test(threaded_scheduler)]
	pub async fn you_are_bob() {
		let addr = "127.0.0.1:46859";
		let server = libunftp::Server::new(Box::new(somefiles))
			.greeting("irrelevant")
			.authenticator(Arc::new(BobAuthenticator {}))
			.passive_ports(50000..65535);
		tokio::spawn(server.listen(addr));
		unmemftp::connectable(addr.parse().unwrap()).await;

		let mut ftp = FtpStream::connect(addr).await .unwrap();
		ftp.login("bob", "bob").await.unwrap();
		ftp.transfer_type(async_ftp::types::FileType::Binary).await.unwrap();
		assert!(ftp.simple_retr("a").await.is_ok());
	}
}
	
async fn default_ftp_client(addr: std::net::SocketAddr) -> FtpStream {
	let mut ftp = FtpStream::connect(addr).await.unwrap();
	ftp.login("anonymous", "onymous").await.unwrap();
	ftp.transfer_type(async_ftp::types::FileType::Binary).await.unwrap();
	return ftp;
}