aesgcm_test.go raw

   1  package config
   2  
   3  import (
   4  	"encoding/hex"
   5  	"testing"
   6  
   7  	"github.com/stretchr/testify/assert"
   8  	"github.com/tyler-smith/go-bip32"
   9  	"github.com/tyler-smith/go-bip39"
  10  )
  11  
  12  func TestDecryptExistingCiphertextWithPassword(t *testing.T) {
  13  	value, err := AesGcmDecryptWithPassword("323f41394d3175b72454ccae9c0081f94df5fb4c2fb0b9283a87e5aafba81839-c335b9eeea75c28a6f823354-5055b90dadbdd01c52fbdbb7efb80609e4410357481651e89ceb1501c8e1dea1f33a8e3322a1cef4f641773667423bca5154dfeccac390cfcd719b36965adc3e6ae56fd5d6c82819596e9ef4ff07193ae345eb291fa412a1ce6066864b", "123")
  14  	assert.NoError(t, err)
  15  	assert.Equal(t, "connect maximum march lava ignore resist visa kind kiwi kidney develop animal", value)
  16  }
  17  
  18  func TestEncryptDecryptWithPassword(t *testing.T) {
  19  	mnemonic := "connect maximum march lava ignore resist visa kind kiwi kidney develop animal"
  20  	encrypted, err := AesGcmEncryptWithPassword(mnemonic, "123")
  21  	assert.NoError(t, err)
  22  	value, err := AesGcmDecryptWithPassword(encrypted, "123")
  23  	assert.NoError(t, err)
  24  	assert.Equal(t, mnemonic, value)
  25  }
  26  
  27  func TestDecryptExistingCiphertextWithKey(t *testing.T) {
  28  	mnemonic := "connect maximum march lava ignore resist visa kind kiwi kidney develop animal"
  29  	masterKey, err := bip32.NewMasterKey(bip39.NewSeed(mnemonic, ""))
  30  	assert.NoError(t, err)
  31  	value, err := AesGcmDecryptWithKey("22ad485dea4f49696594c7c4-afe35ce65fc5a45249bf1b9078472fb28395fc88c30a79c76c7d8d37cf", masterKey.Key)
  32  	assert.NoError(t, err)
  33  	assert.Equal(t, "Hello, world!", value)
  34  }
  35  
  36  func TestEncryptDecryptWithKey(t *testing.T) {
  37  	plaintext := "Hello, world!"
  38  	mnemonic := "connect maximum march lava ignore resist visa kind kiwi kidney develop animal"
  39  	masterKey, err := bip32.NewMasterKey(bip39.NewSeed(mnemonic, ""))
  40  	assert.NoError(t, err)
  41  
  42  	assert.Equal(t, "409e902eafba273b21dff921f0eb4bec6cbb0b657fdce8d245ca78d2920f8b73", hex.EncodeToString(masterKey.Key))
  43  
  44  	encrypted, err := AesGcmEncryptWithKey(plaintext, masterKey.Key)
  45  	assert.NoError(t, err)
  46  	value, err := AesGcmDecryptWithKey(encrypted, masterKey.Key)
  47  	assert.NoError(t, err)
  48  	assert.Equal(t, plaintext, value)
  49  }
  50