@@ -14,20 +14,33 @@ import (
14
14
)
15
15
16
16
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 ))
19
30
20
- file , err := os .Open (path )
21
- require .NoError (t , err )
31
+ file , err := os .Open (path )
32
+ require .NoError (t , err )
22
33
23
- bytes , err := io .ReadAll (file )
24
- require .NoError (t , err )
34
+ bytes , err := io .ReadAll (file )
35
+ require .NoError (t , err )
25
36
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 )
29
40
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
+ }
33
46
}
0 commit comments