mirror of
https://github.com/1f349/mjwt.git
synced 2024-11-09 22:22:48 +00:00
63 lines
2.0 KiB
Go
63 lines
2.0 KiB
Go
|
package mjwt
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/rand"
|
||
|
|
"crypto/rsa"
|
||
|
|
"crypto/x509"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestNewMJwtSigner(t *testing.T) {
|
||
|
|
key, err := rsa.GenerateKey(rand.Reader, 2048)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
NewMJwtSigner("Test", key)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNewMJwtSignerFromFile(t *testing.T) {
|
||
|
|
tempKey, err := os.CreateTemp("", "key-test-*.pem")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
key, err := rsa.GenerateKey(rand.Reader, 2048)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
_, err = tempKey.Write(x509.MarshalPKCS1PrivateKey(key))
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.NoError(t, tempKey.Close())
|
||
|
|
signer, err := NewMJwtSignerFromFile("Test", tempKey.Name())
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.NoError(t, os.Remove(tempKey.Name()))
|
||
|
|
_, err = NewMJwtSignerFromFile("Test", tempKey.Name())
|
||
|
|
assert.Error(t, err)
|
||
|
|
assert.True(t, os.IsNotExist(err))
|
||
|
|
assert.True(t, signer.(*defaultMJwtSigner).key.Equal(key))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNewMJwtSignerFromFileOrCreate(t *testing.T) {
|
||
|
|
tempKey, err := os.CreateTemp("", "key-test-*.pem")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.NoError(t, tempKey.Close())
|
||
|
|
assert.NoError(t, os.Remove(tempKey.Name()))
|
||
|
|
signer, err := NewMJwtSignerFromFileOrCreate("Test", tempKey.Name(), rand.Reader, 2048)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
signer2, err := NewMJwtSignerFromFileOrCreate("Test", tempKey.Name(), rand.Reader, 2048)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.True(t, signer.(*defaultMJwtSigner).key.Equal(signer2.(*defaultMJwtSigner).key))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestReadOrCreatePrivateKey(t *testing.T) {
|
||
|
|
tempKey, err := os.CreateTemp("", "key-test-*.pem")
|
||
|
|
assert.NoError(t, err)
|
||
|
|
key, err := rsa.GenerateKey(rand.Reader, 2048)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
_, err = tempKey.Write(x509.MarshalPKCS1PrivateKey(key))
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.NoError(t, tempKey.Close())
|
||
|
|
key2, err := readOrCreatePrivateKey(tempKey.Name(), rand.Reader, 2048)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.True(t, key.Equal(key2))
|
||
|
|
assert.NoError(t, os.Remove(tempKey.Name()))
|
||
|
|
key3, err := readOrCreatePrivateKey(tempKey.Name(), rand.Reader, 2048)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.NoError(t, key3.Validate())
|
||
|
|
}
|