From f55c5d84f2aab7c9ab585d7116adc35a577f939b Mon Sep 17 00:00:00 2001
From: ChristianVisintin <christian.visintin1997@gmail.com>
Date: Sun, 15 Nov 2020 19:06:45 +0100
Subject: [PATCH] Writing tests for termscp

---
 src/filetransfer/sftp_transfer.rs | 65 +++++++++++++++++++++++++++++--
 1 file changed, 62 insertions(+), 3 deletions(-)

diff --git a/src/filetransfer/sftp_transfer.rs b/src/filetransfer/sftp_transfer.rs
index 6a21220..eae6f65 100644
--- a/src/filetransfer/sftp_transfer.rs
+++ b/src/filetransfer/sftp_transfer.rs
@@ -302,7 +302,7 @@ impl FileTransfer for SftpFileTransfer {
             Some(sftp) => {
                 // Make directory
                 let path: PathBuf = self.get_abs_path(PathBuf::from(dir).as_path());
-                match sftp.mkdir(path.as_path(), 0o755) {
+                match sftp.mkdir(path.as_path(), 0o775) {
                     Ok(_) => Ok(()),
                     Err(_) => Err(FileTransferError::FileCreateDenied),
                 }
@@ -471,11 +471,70 @@ mod tests {
 
     use super::*;
 
-    // TODO: use test.rebex.net with user 'demo' and password 'password'
-
     #[test]
     fn test_filetransfer_sftp_new() {
+        let client: SftpFileTransfer = SftpFileTransfer::new();
+        assert!(client.session.is_none());
+        assert!(client.sftp.is_none());
+        assert_eq!(client.wrkdir, PathBuf::from("~"));
+    }
 
+    #[test]
+    fn test_filetransfer_sftp_connect() {
+        let mut client: SftpFileTransfer = SftpFileTransfer::new();
+        assert!(client.connect(String::from("test.rebex.net"), 22, Some(String::from("demo")), Some(String::from("password"))).is_ok());
+        // Check session and sftp
+        assert!(client.session.is_some());
+        assert!(client.sftp.is_some());
+        assert_eq!(client.wrkdir, PathBuf::from("/"));
+        // Disconnect
+        assert!(client.disconnect().is_ok());
+    }
+
+    #[test]
+    fn test_filetransfer_sftp_bad_auth() {
+        let mut client: SftpFileTransfer = SftpFileTransfer::new();
+        assert!(client.connect(String::from("test.rebex.net"), 22, Some(String::from("demo")), Some(String::from("badpassword"))).is_err());
+    }
+
+    #[test]
+    fn test_filetransfer_sftp_no_credentials() {
+        let mut client: SftpFileTransfer = SftpFileTransfer::new();
+        assert!(client.connect(String::from("test.rebex.net"), 22, None, None).is_err());
+    }
+
+    #[test]
+    fn test_filetransfer_sftp_bad_server() {
+        let mut client: SftpFileTransfer = SftpFileTransfer::new();
+        assert!(client.connect(String::from("mybadserver.veryverybad.awful"), 22, None, None).is_err());
+    }
+
+    #[test]
+    fn test_filetransfer_sftp_pwd() {
+        let mut client: SftpFileTransfer = SftpFileTransfer::new();
+        assert!(client.connect(String::from("test.rebex.net"), 22, Some(String::from("demo")), Some(String::from("password"))).is_ok());
+        // Check session and sftp
+        assert!(client.session.is_some());
+        assert!(client.sftp.is_some());
+        assert_eq!(client.wrkdir, PathBuf::from("/"));
+        // Pwd
+        assert_eq!(client.wrkdir, client.pwd().ok().unwrap());
+        // Disconnect
+        assert!(client.disconnect().is_ok());
+    }
+
+    #[test]
+    fn test_filetransfer_sftp_mkdir() {
+        let mut client: SftpFileTransfer = SftpFileTransfer::new();
+        assert!(client.connect(String::from("test.rebex.net"), 22, Some(String::from("demo")), Some(String::from("password"))).is_ok());
+        let dir: String = String::from("foo");
+        // Mkdir
+        assert!(client.mkdir(dir).is_ok());
+        // cwd
+        assert!(client.change_dir(PathBuf::from("foo/").as_path()).is_ok());
+        assert_eq!(client.wrkdir, PathBuf::from("/foo"));
+        // Disconnect
+        assert!(client.disconnect().is_ok());
     }
 
 }