Skip to content
Snippets Groups Projects
Commit 88737f56 authored by Han-Wen Nienhuys's avatar Han-Wen Nienhuys Committed by Brad Fitzpatrick
Browse files

ssh/test: use t.Run for MAC/kex/pubkey tests

Change-Id: Ic57625964b6d40d63acade476a0f99c00d598267
Reviewed-on: https://go-review.googlesource.com/c/crypto/+/171681


Reviewed-by: default avatarBrad Fitzpatrick <bradfitz@golang.org>
parent e7b772b7
No related branches found
No related tags found
No related merge requests found
...@@ -384,17 +384,19 @@ func TestMACs(t *testing.T) { ...@@ -384,17 +384,19 @@ func TestMACs(t *testing.T) {
macOrder := config.MACs macOrder := config.MACs
for _, mac := range macOrder { for _, mac := range macOrder {
server := newServer(t) t.Run(mac, func(t *testing.T) {
defer server.Shutdown() server := newServer(t)
conf := clientConfig() defer server.Shutdown()
conf.MACs = []string{mac} conf := clientConfig()
// Don't fail if sshd doesn't have the MAC. conf.MACs = []string{mac}
conf.MACs = append(conf.MACs, macOrder...) // Don't fail if sshd doesn't have the MAC.
if conn, err := server.TryDial(conf); err == nil { conf.MACs = append(conf.MACs, macOrder...)
conn.Close() if conn, err := server.TryDial(conf); err == nil {
} else { conn.Close()
t.Fatalf("failed for MAC %q", mac) } else {
} t.Fatalf("failed for MAC %q", mac)
}
})
} }
} }
...@@ -403,17 +405,19 @@ func TestKeyExchanges(t *testing.T) { ...@@ -403,17 +405,19 @@ func TestKeyExchanges(t *testing.T) {
config.SetDefaults() config.SetDefaults()
kexOrder := config.KeyExchanges kexOrder := config.KeyExchanges
for _, kex := range kexOrder { for _, kex := range kexOrder {
server := newServer(t) t.Run(kex, func(t *testing.T) {
defer server.Shutdown() server := newServer(t)
conf := clientConfig() defer server.Shutdown()
// Don't fail if sshd doesn't have the kex. conf := clientConfig()
conf.KeyExchanges = append([]string{kex}, kexOrder...) // Don't fail if sshd doesn't have the kex.
conn, err := server.TryDial(conf) conf.KeyExchanges = append([]string{kex}, kexOrder...)
if err == nil { conn, err := server.TryDial(conf)
conn.Close() if err == nil {
} else { conn.Close()
t.Errorf("failed for kex %q", kex) } else {
} t.Errorf("failed for kex %q", kex)
}
})
} }
} }
...@@ -424,20 +428,22 @@ func TestClientAuthAlgorithms(t *testing.T) { ...@@ -424,20 +428,22 @@ func TestClientAuthAlgorithms(t *testing.T) {
"ecdsa", "ecdsa",
"ed25519", "ed25519",
} { } {
server := newServer(t) t.Run(key, func(t *testing.T) {
conf := clientConfig() server := newServer(t)
conf.SetDefaults() conf := clientConfig()
conf.Auth = []ssh.AuthMethod{ conf.SetDefaults()
ssh.PublicKeys(testSigners[key]), conf.Auth = []ssh.AuthMethod{
} ssh.PublicKeys(testSigners[key]),
}
conn, err := server.TryDial(conf)
if err == nil { conn, err := server.TryDial(conf)
conn.Close() if err == nil {
} else { conn.Close()
t.Errorf("failed for key %q", key) } else {
} t.Errorf("failed for key %q", key)
}
server.Shutdown()
server.Shutdown()
})
} }
} }
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment