added ssh key auth test

Author: mariotaku

src-tauri/src/conn_pool/cmd.rs

@@ -103,9 +103,7 @@ mod test {
         ))
         .unwrap();
         let err = DeviceConnection::new(device, None).expect_err("Should have failed");
-        assert!(
-            matches!(err, Error::Authorization { message } if message == "Bad SSH password")
-        );
+        assert!(matches!(err, Error::Authorization { message } if message == "Bad SSH password"));
     }
 
     #[test]
@@ -124,6 +122,23 @@ mod test {
         assert_eq!(b"root\n", output.stdout.as_ref());
     }
 
+    #[test]
+    fn execute_command_whoami_keyauth() {
+        let sshd = SshContainer::new();
+        let port = sshd.wait();
+        let device = serde_json::from_str::<Device>(&format!(
+            "{{\"profile\":\"ose\",\"name\":\"test\",\"host\":\"127.0.0.1\",\
+            \"port\": {port},\"username\": \"root\",\"privateKey\": {{\"openSsh\": \"id_root\"}}}}"
+        ))
+        .unwrap();
+        let conn = DeviceConnection::new(device, Some(&SshContainer::fixture_path("keys", false)))
+            .expect("Failed to create connection");
+        let output = conn
+            .execute_command("whoami", None, Encoding::String)
+            .expect("Failed to execute command");
+        assert_eq!(b"root\n", output.stdout.as_ref());
+    }
+
     #[test]
     fn execute_command_false() {
         let sshd = SshContainer::new();

src-tauri/src/tests/common/test_server/keys/id_root

@@ -0,0 +1,9 @@
+-----BEGIN OPENSSH PRIVATE KEY-----
+b3BlbnNzaC1rZXktdjEAAAAABG5vbmUAAAAEbm9uZQAAAAAAAAABAAAAaAAAABNlY2RzYS
+1zaGEyLW5pc3RwMjU2AAAACG5pc3RwMjU2AAAAQQTqu3dj7L8aVOTlvR/+wFfzn/t7VA4H
+sA9wPRBtLxeOePcxYtXb+a/htuy0I7VKQxyNk6u3yz3AP5rImvc0po+4AAAAsBDlh8UQ5Y
+fFAAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBOq7d2PsvxpU5OW9
+H/7AV/Of+3tUDgewD3A9EG0vF4549zFi1dv5r+G27LQjtUpDHI2Tq7fLPcA/msia9zSmj7
+gAAAAgDz8yxLi2BFo+MQ9b4s6a8YZG2pXElVd/oP3kk3em8U4AAAAXbWFyaW90YWt1QE1B
+UklPVEFLVS1OVUMB
+-----END OPENSSH PRIVATE KEY-----

src-tauri/src/tests/common/test_server/keys/id_root.pub

@@ -0,0 +1 @@
+ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBOq7d2PsvxpU5OW9H/7AV/Of+3tUDgewD3A9EG0vF4549zFi1dv5r+G27LQjtUpDHI2Tq7fLPcA/msia9zSmj7g= mariotaku@MARIOTAKU-NUC

src-tauri/src/tests/common/test_server/mod.rs

@@ -14,10 +14,6 @@ pub struct SshContainer {
 
 impl SshContainer {
     pub fn new() -> Self {
-        let mp = format!(
-            "{}:/etc/entrypoint.d/",
-            Self::entrypoint_d().to_string_lossy()
-        );
         let output = Command::new("docker")
             .args(["run", "--rm", "-d", "-p", "22"])
             .args(["-e", "SSH_ENABLE_ROOT=true"])
@@ -30,7 +26,18 @@ impl SshContainer {
                 "--health-interval",
                 "3s",
             ])
-            .args(["-v", &mp])
+            .args([
+                "-v",
+                &format!(
+                    "{}:/etc/entrypoint.d/",
+                    Self::fixture_path("entrypoint.d", true).to_string_lossy()
+                ),
+                "-v",
+                &format!(
+                    "{}:/root/.ssh/authorized_keys",
+                    Self::fixture_path("keys/id_root.pub", true).to_string_lossy()
+                ),
+            ])
             .arg("public.ecr.aws/panubo/sshd")
             .output()
             .expect("Failed to start sshd container");
@@ -75,9 +82,9 @@ impl SshContainer {
         panic!("Failed to start sshd container");
     }
 
-    fn entrypoint_d() -> PathBuf {
-        let path = abs_file!().parent().unwrap().join("entrypoint.d");
-        if cfg!(target_family = "windows") {
+    pub fn fixture_path(name: &str, for_container: bool) -> PathBuf {
+        let path = abs_file!().parent().unwrap().join(name);
+        if for_container && cfg!(target_family = "windows") {
             let mut components = path.components();
             let drive: Component = components.next().unwrap().into();
             components.next();