prepare test for more keys

Author: TheFox0x7

modules/ssh/ssh_test.go

@@ -14,20 +14,33 @@ import (
 )
 
 func TestGenKeyPair(t *testing.T) {
-	path := t.TempDir() + "/gitea.rsa"
-	require.NoError(t, ssh.GenKeyPair(path, ssh.RSA))
+	testCases := []struct {
+		keyType      ssh.KeyType
+		expectedType any
+	}{
+		{
+			keyType:      ssh.RSA,
+			expectedType: &rsa.PrivateKey{},
+		},
+	}
+	for _, tC := range testCases {
+		t.Run("Generate"+string(tC.keyType), func(t *testing.T) {
+			path := t.TempDir() + "/gitea." + string(tC.keyType)
+			require.NoError(t, ssh.GenKeyPair(path, tC.keyType))
 
-	file, err := os.Open(path)
-	require.NoError(t, err)
+			file, err := os.Open(path)
+			require.NoError(t, err)
 
-	bytes, err := io.ReadAll(file)
-	require.NoError(t, err)
+			bytes, err := io.ReadAll(file)
+			require.NoError(t, err)
 
-	block, _ := pem.Decode(bytes)
-	require.NotNil(t, block)
-	assert.Equal(t, "PRIVATE KEY", block.Type)
+			block, _ := pem.Decode(bytes)
+			require.NotNil(t, block)
+			assert.Equal(t, "PRIVATE KEY", block.Type)
 
-	privateKey, err := x509.ParsePKCS8PrivateKey(block.Bytes)
-	require.NoError(t, err)
-	assert.IsType(t, &rsa.PrivateKey{}, privateKey)
+			privateKey, err := x509.ParsePKCS8PrivateKey(block.Bytes)
+			require.NoError(t, err)
+			assert.IsType(t, tC.expectedType, privateKey)
+		})
+	}
 }