Skip to content

Commit e123054

Browse files
committed
prepare test for more keys
1 parent ac013f0 commit e123054

File tree

1 file changed

+25
-12
lines changed

1 file changed

+25
-12
lines changed

modules/ssh/ssh_test.go

Lines changed: 25 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -14,20 +14,33 @@ import (
1414
)
1515

1616
func TestGenKeyPair(t *testing.T) {
17-
path := t.TempDir() + "/gitea.rsa"
18-
require.NoError(t, ssh.GenKeyPair(path, ssh.RSA))
17+
testCases := []struct {
18+
keyType ssh.KeyType
19+
expectedType any
20+
}{
21+
{
22+
keyType: ssh.RSA,
23+
expectedType: &rsa.PrivateKey{},
24+
},
25+
}
26+
for _, tC := range testCases {
27+
t.Run("Generate"+string(tC.keyType), func(t *testing.T) {
28+
path := t.TempDir() + "/gitea." + string(tC.keyType)
29+
require.NoError(t, ssh.GenKeyPair(path, tC.keyType))
1930

20-
file, err := os.Open(path)
21-
require.NoError(t, err)
31+
file, err := os.Open(path)
32+
require.NoError(t, err)
2233

23-
bytes, err := io.ReadAll(file)
24-
require.NoError(t, err)
34+
bytes, err := io.ReadAll(file)
35+
require.NoError(t, err)
2536

26-
block, _ := pem.Decode(bytes)
27-
require.NotNil(t, block)
28-
assert.Equal(t, "PRIVATE KEY", block.Type)
37+
block, _ := pem.Decode(bytes)
38+
require.NotNil(t, block)
39+
assert.Equal(t, "PRIVATE KEY", block.Type)
2940

30-
privateKey, err := x509.ParsePKCS8PrivateKey(block.Bytes)
31-
require.NoError(t, err)
32-
assert.IsType(t, &rsa.PrivateKey{}, privateKey)
41+
privateKey, err := x509.ParsePKCS8PrivateKey(block.Bytes)
42+
require.NoError(t, err)
43+
assert.IsType(t, tC.expectedType, privateKey)
44+
})
45+
}
3346
}

0 commit comments

Comments
 (0)