github.com/unicornultrafoundation/go-u2u@v1.0.0-rc1.0.20240205080301-e74a83d3fadc/valkeystore/mem_test.go (about)

     1  package valkeystore
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/require"
     7  )
     8  
     9  func TestMemKeystoreAdd(t *testing.T) {
    10  	require := require.New(t)
    11  	keystore := NewMemKeystore()
    12  
    13  	key, err := keystore.Get(pubkey1, "auth1")
    14  	require.EqualError(err, ErrNotFound.Error())
    15  	require.Nil(key)
    16  
    17  	err = keystore.Add(pubkey1, key1, "auth1")
    18  	require.NoError(err)
    19  
    20  	testGet(t, keystore, pubkey1, key1, "auth1")
    21  
    22  	err = keystore.Add(pubkey2, key2, "auth2")
    23  	require.NoError(err)
    24  
    25  	testGet(t, keystore, pubkey1, key1, "auth1")
    26  	testGet(t, keystore, pubkey2, key2, "auth2")
    27  
    28  	err = keystore.Add(pubkey2, key2, "auth1")
    29  	require.Error(err, ErrAlreadyExists.Error())
    30  
    31  	testGet(t, keystore, pubkey2, key2, "auth2")
    32  }