Use passed-in AuthMethods in rigtest

Signed-off-by: Kimmo Lehto <[email protected]>

cmd/rigtest/rigtest.go

@@ -156,10 +156,9 @@ func main() {
 			port = p
 		}
 
-		var h *Host
 		switch *proto {
 		case "ssh":
-			h = &Host{
+			hosts = append(hosts, &Host{
 				Connection: rig.Connection{
 					SSH: &rig.SSH{
 						Address:          address,
@@ -169,9 +168,27 @@ func main() {
 						PasswordCallback: passfunc,
 					},
 				},
+			})
+			// use a key from memory
+			key, err := goos.ReadFile(*kp)
+			if err != nil {
+				panic(err)
 			}
+			authMethods, err := rig.ParseSSHPrivateKey(key, rig.DefaultPasswordCallback)
+			hosts = append(hosts, &Host{
+				Connection: rig.Connection{
+					SSH: &rig.SSH{
+						Address:          address,
+						Port:             port,
+						User:             *usr,
+						KeyPath:          nil,
+						PasswordCallback: passfunc,
+						AuthMethods:      authMethods,
+					},
+				},
+			})
 		case "winrm":
-			h = &Host{
+			hosts = append(hosts, &Host{
 				Connection: rig.Connection{
 					WinRM: &rig.WinRM{
 						Address:  *dh,
@@ -182,19 +199,18 @@ func main() {
 						Password: *pwd,
 					},
 				},
-			}
+			})
 		case "localhost":
-			h = &Host{
+			hosts = append(hosts, &Host{
 				Connection: rig.Connection{
 					Localhost: &rig.Localhost{
 						Enabled: true,
 					},
 				},
-			}
+			})
 		default:
 			panic("unknown protocol " + *proto)
 		}
-		hosts = append(hosts, h)
 	}
 
 	t := testRunner{}