forked from TrueCloudLab/frostfs-crypto
86 lines
1.7 KiB
Go
86 lines
1.7 KiB
Go
package crypto
|
|
|
|
import (
|
|
"crypto/x509"
|
|
"encoding/hex"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/TrueCloudLab/frostfs-crypto/test"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_LoadPrivateKey_FromWIF(t *testing.T) {
|
|
for i := 0; i < 10; i++ {
|
|
expected := test.DecodeKey(i)
|
|
|
|
wif, err := WIFEncode(expected)
|
|
require.NoError(t, err)
|
|
|
|
actual, err := LoadPrivateKey(wif)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, expected, actual)
|
|
}
|
|
}
|
|
|
|
func Test_LoadPrivateKey_FromHexString(t *testing.T) {
|
|
for i := 0; i < 10; i++ {
|
|
expected := test.DecodeKey(i)
|
|
|
|
hs := hex.EncodeToString(expected.D.Bytes())
|
|
|
|
actual, err := LoadPrivateKey(hs)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, expected, actual)
|
|
}
|
|
}
|
|
|
|
func Test_LoadPrivateKey_FromFile(t *testing.T) {
|
|
for i := 0; i < 10; i++ {
|
|
expected := test.DecodeKey(i)
|
|
|
|
file, err := ioutil.TempFile("", "_marshaled.key")
|
|
require.NoError(t, err)
|
|
|
|
defer func() {
|
|
require.NoError(t, file.Close())
|
|
require.NoError(t, os.Remove(file.Name()))
|
|
}()
|
|
|
|
data, err := x509.MarshalECPrivateKey(expected)
|
|
require.NoError(t, err)
|
|
|
|
_, err = file.Write(data)
|
|
require.NoError(t, err)
|
|
|
|
actual, err := LoadPrivateKey(file.Name())
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, expected, actual)
|
|
}
|
|
}
|
|
|
|
func Test_LoadPrivateKey_FromCompressedFormatFile(t *testing.T) {
|
|
for i := 0; i < 10; i++ {
|
|
expected := test.DecodeKey(i)
|
|
|
|
file, err := ioutil.TempFile("", "_compressed.key")
|
|
require.NoError(t, err)
|
|
|
|
defer func() {
|
|
require.NoError(t, file.Close())
|
|
require.NoError(t, os.Remove(file.Name()))
|
|
}()
|
|
|
|
_, err = file.Write(expected.D.Bytes())
|
|
require.NoError(t, err)
|
|
|
|
actual, err := LoadPrivateKey(file.Name())
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, expected, actual)
|
|
}
|
|
}
|